aboutsummaryrefslogtreecommitdiff
path: root/python/pyspark/sql/readwriter.py
diff options
context:
space:
mode:
authorCheolsoo Park <cheolsoop@netflix.com>2015-06-29 00:13:39 -0700
committerReynold Xin <rxin@databricks.com>2015-06-29 00:13:39 -0700
commitac2e17b01c0843d928a363d2cc4faf57ec8c8b47 (patch)
treea06efb0edf4c0ab4af7abeb1cd0bb3870799deb7 /python/pyspark/sql/readwriter.py
parent0b10662fef11a56f82144b4953d457738e6961ae (diff)
downloadspark-ac2e17b01c0843d928a363d2cc4faf57ec8c8b47.tar.gz
spark-ac2e17b01c0843d928a363d2cc4faf57ec8c8b47.tar.bz2
spark-ac2e17b01c0843d928a363d2cc4faf57ec8c8b47.zip
[SPARK-8355] [SQL] Python DataFrameReader/Writer should mirror Scala
I compared PySpark DataFrameReader/Writer against Scala ones. `Option` function is missing in both reader and writer, but the rest seems to all match. I added `Option` to reader and writer and updated the `pyspark-sql` test. Author: Cheolsoo Park <cheolsoop@netflix.com> Closes #7078 from piaozhexiu/SPARK-8355 and squashes the following commits: c63d419 [Cheolsoo Park] Fix version 524e0aa [Cheolsoo Park] Add option function to df reader and writer
Diffstat (limited to 'python/pyspark/sql/readwriter.py')
-rw-r--r--python/pyspark/sql/readwriter.py14
1 files changed, 14 insertions, 0 deletions
diff --git a/python/pyspark/sql/readwriter.py b/python/pyspark/sql/readwriter.py
index 1b7bc0f9a1..c4cc62e82a 100644
--- a/python/pyspark/sql/readwriter.py
+++ b/python/pyspark/sql/readwriter.py
@@ -73,6 +73,13 @@ class DataFrameReader(object):
self._jreader = self._jreader.schema(jschema)
return self
+ @since(1.5)
+ def option(self, key, value):
+ """Adds an input option for the underlying data source.
+ """
+ self._jreader = self._jreader.option(key, value)
+ return self
+
@since(1.4)
def options(self, **options):
"""Adds input options for the underlying data source.
@@ -235,6 +242,13 @@ class DataFrameWriter(object):
self._jwrite = self._jwrite.format(source)
return self
+ @since(1.5)
+ def option(self, key, value):
+ """Adds an output option for the underlying data source.
+ """
+ self._jwrite = self._jwrite.option(key, value)
+ return self
+
@since(1.4)
def options(self, **options):
"""Adds output options for the underlying data source.