sqlframe 3.8.0__tar.gz → 3.8.2__tar.gz

{sqlframe-3.8.0 → sqlframe-3.8.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: sqlframe
-Version: 3.8.0
+Version: 3.8.2
 Summary: Turning PySpark Into a Universal DataFrame API
 Home-page: https://github.com/eakmanrq/sqlframe
 Author: Ryan Eakman

{sqlframe-3.8.0 → sqlframe-3.8.2}/sqlframe/_version.py

@@ -12,5 +12,5 @@ __version__: str
 __version_tuple__: VERSION_TUPLE
 version_tuple: VERSION_TUPLE
 
-__version__ = version = '3.8.0'
-__version_tuple__ = version_tuple = (3, 8, 0)
+__version__ = version = '3.8.2'
+__version_tuple__ = version_tuple = (3, 8, 2)

{sqlframe-3.8.0 → sqlframe-3.8.2}/sqlframe/base/util.py

@@ -6,6 +6,7 @@ import unicodedata
 
 from sqlglot import expressions as exp
 from sqlglot import parse_one, to_table
+from sqlglot.dialects import DuckDB
 from sqlglot.dialects.dialect import Dialect, DialectType
 from sqlglot.optimizer.normalize_identifiers import normalize_identifiers
 from sqlglot.optimizer.qualify_columns import (
@@ -372,6 +373,12 @@ def normalize_string(
 ) -> str:
     from sqlframe.base.session import _BaseSession
 
+    data_type_replacement_mapping = {
+        DuckDB: {
+            "TIMESTAMP_NS": "TIMESTAMP",
+        }
+    }
+
     session: _BaseSession = _BaseSession()
 
     str_to_dialect = {
@@ -397,6 +404,9 @@ def normalize_string(
         elif is_table:
             value_expression = to_table(value_without_star, dialect=from_dialect)
         elif is_datatype:
+            value_without_star = data_type_replacement_mapping.get(from_dialect, {}).get(  # type: ignore
+                value_without_star, value_without_star
+            )
             value_expression = exp.DataType.build(value_without_star, dialect=from_dialect)
         elif is_column:
             value_expression = exp.to_column(value_without_star, dialect=from_dialect)

{sqlframe-3.8.0 → sqlframe-3.8.2}/sqlframe.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: sqlframe
-Version: 3.8.0
+Version: 3.8.2
 Summary: Turning PySpark Into a Universal DataFrame API
 Home-page: https://github.com/eakmanrq/sqlframe
 Author: Ryan Eakman