datacontract-cli 0.10.24__py3-none-any.whl → 0.10.26__py3-none-any.whl

datacontract/imports/avro_importer.py

@@ -55,7 +55,7 @@ def import_avro(data_contract_specification: DataContractSpecification, source:
             engine="datacontract",
             original_exception=e,
         )
-  # type record is being used for both the table and the object types in data contract
+    # type record is being used for both the table and the object types in data contract
     # -> CONSTRAINT: one table per .avsc input, all nested records are interpreted as objects
     fields = import_record_fields(avro_schema.fields)
 
@@ -92,19 +92,19 @@ def handle_config_avro_custom_properties(field: avro.schema.Field, imported_fiel
 
 
 LOGICAL_TYPE_MAPPING = {
-     "decimal": "decimal",
-     "date": "date",
-     "time-millis": "time",
-     "time-micros": "time",
-     "timestamp-millis": "timestamp_tz",
-     "timestamp-micros": "timestamp_tz",
-     "local-timestamp-micros": "timestamp_ntz",
-     "local-timestamp-millis": "timestamp_ntz",
-     "duration": "string",
-     "uuid": "string",
- }
-    
-  
+    "decimal": "decimal",
+    "date": "date",
+    "time-millis": "time",
+    "time-micros": "time",
+    "timestamp-millis": "timestamp_tz",
+    "timestamp-micros": "timestamp_tz",
+    "local-timestamp-micros": "timestamp_ntz",
+    "local-timestamp-millis": "timestamp_ntz",
+    "duration": "string",
+    "uuid": "string",
+}
+
+
 def import_record_fields(record_fields: List[avro.schema.Field]) -> Dict[str, Field]:
     """
     Import Avro record fields and convert them to data contract fields.
@@ -150,15 +150,15 @@ def import_record_fields(record_fields: List[avro.schema.Field]) -> Dict[str, Fi
             if not imported_field.config:
                 imported_field.config = {}
             imported_field.config["avroType"] = "enum"
-        else:  
-             logical_type = field.type.get_prop("logicalType")
-             if logical_type in LOGICAL_TYPE_MAPPING:
-                 imported_field.type = LOGICAL_TYPE_MAPPING[logical_type]
-                 if logical_type == "decimal":
-                     imported_field.precision = field.type.precision
-                     imported_field.scale = field.type.scale
-             else:
-                 imported_field.type = map_type_from_avro(field.type.type)
+        else:
+            logical_type = field.type.get_prop("logicalType")
+            if logical_type in LOGICAL_TYPE_MAPPING:
+                imported_field.type = LOGICAL_TYPE_MAPPING[logical_type]
+                if logical_type == "decimal":
+                    imported_field.precision = field.type.precision
+                    imported_field.scale = field.type.scale
+            else:
+                imported_field.type = map_type_from_avro(field.type.type)
         imported_fields[field.name] = imported_field
 
     return imported_fields

datacontract/imports/csv_importer.py

@@ -31,10 +31,10 @@ def import_csv(
     if data_contract_specification.servers is None:
         data_contract_specification.servers = {}
 
-    delimiter = None if dialect is None else dialect['Delimiter'][0]
+    delimiter = None if dialect is None else dialect["Delimiter"][0]
 
     if dialect is not None:
-        dc_types = [map_type_from_duckdb(x["type"]) for x in dialect['Columns'][0]]
+        dc_types = [map_type_from_duckdb(x["type"]) for x in dialect["Columns"][0]]
     else:
         dc_types = [map_type_from_duckdb(str(x)) for x in tbl.dtypes]