diff options
author | Daniel Baumann <daniel.baumann@progress-linux.org> | 2022-11-11 08:54:35 +0000 |
---|---|---|
committer | Daniel Baumann <daniel.baumann@progress-linux.org> | 2022-11-11 08:54:35 +0000 |
commit | d1f00706bff58b863b0a1c5bf4adf39d36049d4c (patch) | |
tree | 3a8ecc5d1509d655d5df6b1455bc1e309da2c02c /sqlglot/dataframe/sql/readwriter.py | |
parent | Releasing debian version 9.0.6-1. (diff) | |
download | sqlglot-d1f00706bff58b863b0a1c5bf4adf39d36049d4c.tar.xz sqlglot-d1f00706bff58b863b0a1c5bf4adf39d36049d4c.zip |
Merging upstream version 10.0.1.
Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>
Diffstat (limited to 'sqlglot/dataframe/sql/readwriter.py')
-rw-r--r-- | sqlglot/dataframe/sql/readwriter.py | 16 |
1 files changed, 13 insertions, 3 deletions
diff --git a/sqlglot/dataframe/sql/readwriter.py b/sqlglot/dataframe/sql/readwriter.py index 4830035..febc664 100644 --- a/sqlglot/dataframe/sql/readwriter.py +++ b/sqlglot/dataframe/sql/readwriter.py @@ -19,12 +19,19 @@ class DataFrameReader: from sqlglot.dataframe.sql.dataframe import DataFrame sqlglot.schema.add_table(tableName) - return DataFrame(self.spark, exp.Select().from_(tableName).select(*sqlglot.schema.column_names(tableName))) + return DataFrame( + self.spark, + exp.Select().from_(tableName).select(*sqlglot.schema.column_names(tableName)), + ) class DataFrameWriter: def __init__( - self, df: DataFrame, spark: t.Optional[SparkSession] = None, mode: t.Optional[str] = None, by_name: bool = False + self, + df: DataFrame, + spark: t.Optional[SparkSession] = None, + mode: t.Optional[str] = None, + by_name: bool = False, ): self._df = df self._spark = spark or df.spark @@ -33,7 +40,10 @@ class DataFrameWriter: def copy(self, **kwargs) -> DataFrameWriter: return DataFrameWriter( - **{k[1:] if k.startswith("_") else k: v for k, v in object_to_dict(self, **kwargs).items()} + **{ + k[1:] if k.startswith("_") else k: v + for k, v in object_to_dict(self, **kwargs).items() + } ) def sql(self, **kwargs) -> t.List[str]: |