format

Author: chenkovsky

src/expr.rs

@@ -64,10 +64,22 @@ pub mod case;
 pub mod cast;
 pub mod column;
 pub mod conditional_expr;
+pub mod constraints;
+pub mod copy_to;
+pub mod create_catalog;
+pub mod create_catalog_schema;
+pub mod create_external_table;
+pub mod create_function;
+pub mod create_index;
 pub mod create_memory_table;
 pub mod create_view;
+pub mod describe_table;
 pub mod distinct;
+pub mod dml;
+pub mod drop_catalog_schema;
+pub mod drop_function;
 pub mod drop_table;
+pub mod drop_view;
 pub mod empty_relation;
 pub mod exists;
 pub mod explain;
@@ -83,34 +95,22 @@ pub mod literal;
 pub mod logical_node;
 pub mod placeholder;
 pub mod projection;
+pub mod recursive_query;
 pub mod repartition;
 pub mod scalar_subquery;
 pub mod scalar_variable;
 pub mod signature;
 pub mod sort;
 pub mod sort_expr;
+pub mod statement;
 pub mod subquery;
 pub mod subquery_alias;
 pub mod table_scan;
 pub mod union;
 pub mod unnest;
 pub mod unnest_expr;
-pub mod window;
-pub mod statement;
 pub mod values;
-pub mod dml;
-pub mod create_external_table;
-pub mod copy_to;
-pub mod create_catalog_schema;
-pub mod drop_view;
-pub mod create_catalog;
-pub mod drop_catalog_schema;
-pub mod drop_function;
-pub mod create_function;
-pub mod create_index;
-pub mod describe_table;
-pub mod recursive_query;
-pub mod constraints;
+pub mod window;
 
 use sort_expr::{to_sort_expressions, PySortExpr};
 

src/expr/constraints.rs

@@ -26,4 +26,3 @@ impl Display for PyConstraints {
         write!(f, "Constraints: {:?}", self.constraints)
     }
 }
-

src/expr/copy_to.rs

@@ -15,7 +15,11 @@
 // specific language governing permissions and limitations
 // under the License.
 
-use std::{collections::HashMap, fmt::{self, Display, Formatter}, sync::Arc};
+use std::{
+    collections::HashMap,
+    fmt::{self, Display, Formatter},
+    sync::Arc,
+};
 
 use datafusion::{common::file_options::file_type::FileType, logical_expr::dml::CopyTo};
 use pyo3::prelude::*;
@@ -49,7 +53,6 @@ impl Display for PyCopyTo {
 }
 
 impl LogicalNode for PyCopyTo {
-
     fn inputs(&self) -> Vec<PyLogicalPlan> {
         vec![PyLogicalPlan::from((*self.copy.input).clone())]
     }
@@ -67,7 +70,8 @@ impl PyCopyTo {
         output_url: String,
         partition_by: Vec<String>,
         file_type: PyFileType,
-        options: HashMap<String, String>) -> Self {
+        options: HashMap<String, String>,
+    ) -> Self {
         return PyCopyTo {
             copy: CopyTo {
                 input: input.plan(),
@@ -76,7 +80,7 @@ impl PyCopyTo {
                 file_type: file_type.file_type,
                 options,
             },
-        }
+        };
     }
 
     fn input(&self) -> PyLogicalPlan {
@@ -92,7 +96,9 @@ impl PyCopyTo {
     }
 
     fn file_type(&self) -> PyFileType {
-        PyFileType { file_type: self.copy.file_type.clone() }
+        PyFileType {
+            file_type: self.copy.file_type.clone(),
+        }
     }
 
     fn options(&self) -> HashMap<String, String> {

src/expr/create_catalog.rs

@@ -15,7 +15,10 @@
 // specific language governing permissions and limitations
 // under the License.
 
-use std::{fmt::{self, Display, Formatter}, sync::Arc};
+use std::{
+    fmt::{self, Display, Formatter},
+    sync::Arc,
+};
 
 use datafusion::logical_expr::CreateCatalog;
 use pyo3::prelude::*;
@@ -50,15 +53,18 @@ impl Display for PyCreateCatalog {
 
 #[pymethods]
 impl PyCreateCatalog {
-
     #[new]
-    pub fn new(catalog_name: String, if_not_exists: bool, schema: PyDFSchema) -> PyResult<PyCreateCatalog> {
+    pub fn new(
+        catalog_name: String,
+        if_not_exists: bool,
+        schema: PyDFSchema,
+    ) -> PyResult<PyCreateCatalog> {
         Ok(PyCreateCatalog {
             create: CreateCatalog {
                 catalog_name,
                 if_not_exists,
                 schema: Arc::new(schema.into()),
-            }
+            },
         })
     }
 

src/expr/create_catalog_schema.rs

@@ -15,7 +15,10 @@
 // specific language governing permissions and limitations
 // under the License.
 
-use std::{fmt::{self, Display, Formatter}, sync::Arc};
+use std::{
+    fmt::{self, Display, Formatter},
+    sync::Arc,
+};
 
 use datafusion::logical_expr::CreateCatalogSchema;
 use pyo3::prelude::*;
@@ -30,7 +33,6 @@ pub struct PyCreateCatalogSchema {
     create: CreateCatalogSchema,
 }
 
-
 impl From<PyCreateCatalogSchema> for CreateCatalogSchema {
     fn from(create: PyCreateCatalogSchema) -> Self {
         create.create
@@ -49,18 +51,20 @@ impl Display for PyCreateCatalogSchema {
     }
 }
 
-
 #[pymethods]
 impl PyCreateCatalogSchema {
-
     #[new]
-    pub fn new(schema_name: String, if_not_exists: bool, schema: PyDFSchema) -> PyResult<PyCreateCatalogSchema> {
+    pub fn new(
+        schema_name: String,
+        if_not_exists: bool,
+        schema: PyDFSchema,
+    ) -> PyResult<PyCreateCatalogSchema> {
         Ok(PyCreateCatalogSchema {
             create: CreateCatalogSchema {
                 schema_name,
                 if_not_exists,
                 schema: Arc::new(schema.into()),
-            }
+            },
         })
     }
 

src/expr/create_external_table.rs

@@ -16,7 +16,11 @@
 // under the License.
 
 use crate::{expr::PyExpr, sql::logical::PyLogicalPlan};
-use std::{collections::HashMap, fmt::{self, Display, Formatter}, sync::Arc};
+use std::{
+    collections::HashMap,
+    fmt::{self, Display, Formatter},
+    sync::Arc,
+};
 
 use datafusion::logical_expr::CreateExternalTable;
 use pyo3::prelude::*;
@@ -45,7 +49,11 @@ impl From<CreateExternalTable> for PyCreateExternalTable {
 
 impl Display for PyCreateExternalTable {
     fn fmt(&self, f: &mut Formatter) -> fmt::Result {
-        write!(f, "CreateExternalTable: {:?}{}", self.create.name, self.create.constraints)
+        write!(
+            f,
+            "CreateExternalTable: {:?}{}",
+            self.create.name, self.create.constraints
+        )
     }
 }
 
@@ -92,76 +100,66 @@ impl PyCreateExternalTable {
         PyCreateExternalTable { create: create }
     }
 
-
     pub fn schema(&self) -> PyDFSchema {
         (*self.create.schema).clone().into()
     }
 
-
     pub fn name(&self) -> PyResult<String> {
         Ok(self.create.name.to_string())
     }
 
-
     pub fn location(&self) -> String {
         self.create.location.clone()
     }
 
-
     pub fn file_type(&self) -> String {
         self.create.file_type.clone()
     }
 
-
     pub fn table_partition_cols(&self) -> Vec<String> {
         self.create.table_partition_cols.clone()
     }
 
-
     pub fn if_not_exists(&self) -> bool {
         self.create.if_not_exists
     }
 
-
     pub fn temporary(&self) -> bool {
         self.create.temporary
     }
 
-
     pub fn definition(&self) -> Option<String> {
         self.create.definition.clone()
     }
 
-
     pub fn order_exprs(&self) -> Vec<Vec<PySortExpr>> {
-        self.create.order_exprs.iter().map(|vec| vec.iter().map(|s| s.clone().into()).collect()).collect()
+        self.create
+            .order_exprs
+            .iter()
+            .map(|vec| vec.iter().map(|s| s.clone().into()).collect())
+            .collect()
     }
 
-
     pub fn unbounded(&self) -> bool {
         self.create.unbounded
     }
 
-
     pub fn options(&self) -> HashMap<String, String> {
         self.create.options.clone()
     }
 
-
     pub fn constraints(&self) -> PyConstraints {
         PyConstraints {
             constraints: self.create.constraints.clone(),
         }
     }
 
-
     pub fn column_defaults(&self) -> HashMap<String, PyExpr> {
         self.create
             .column_defaults
             .iter()
             .map(|(k, v)| (k.clone(), v.clone().into()))
             .collect()
-
     }
 
     fn __repr__(&self) -> PyResult<String> {
@@ -173,7 +171,6 @@ impl PyCreateExternalTable {
     }
 }
 
-
 impl LogicalNode for PyCreateExternalTable {
     fn inputs(&self) -> Vec<PyLogicalPlan> {
         vec![]
@@ -183,4 +180,3 @@ impl LogicalNode for PyCreateExternalTable {
         Ok(self.clone().into_py(py))
     }
 }
-