@@ -253,25 +253,21 @@ def _from_sqlglot_STRUCT(
253253
254254 @classmethod
255255 def _from_sqlglot_TIMESTAMP (
256- cls , scale = None , nullable : bool | None = None
256+ cls , scale : int | None = None , nullable : bool | None = None
257257 ) -> dt .Timestamp :
258258 return dt .Timestamp (
259259 scale = cls .default_temporal_scale if scale is None else int (scale .this .this ),
260260 nullable = nullable ,
261261 )
262262
263263 @classmethod
264- def _from_sqlglot_TIMESTAMPTZ (
264+ def _from_sqlglot_TIMESTAMPNTZ (
265265 cls , scale = None , nullable : bool | None = None
266266 ) -> dt .Timestamp :
267- return dt .Timestamp (
268- timezone = "UTC" ,
269- scale = cls .default_temporal_scale if scale is None else int (scale .this .this ),
270- nullable = nullable ,
271- )
267+ return cls ._from_sqlglot_TIMESTAMP (scale = scale , nullable = nullable )
272268
273269 @classmethod
274- def _from_sqlglot_TIMESTAMPLTZ (
270+ def _from_sqlglot_TIMESTAMPTZ (
275271 cls , scale = None , nullable : bool | None = None
276272 ) -> dt .Timestamp :
277273 return dt .Timestamp (
@@ -281,11 +277,11 @@ def _from_sqlglot_TIMESTAMPLTZ(
281277 )
282278
283279 @classmethod
284- def _from_sqlglot_TIMESTAMPNTZ (
280+ def _from_sqlglot_TIMESTAMPLTZ (
285281 cls , scale = None , nullable : bool | None = None
286282 ) -> dt .Timestamp :
287283 return dt .Timestamp (
288- timezone = None ,
284+ timezone = "UTC" ,
289285 scale = cls .default_temporal_scale if scale is None else int (scale .this .this ),
290286 nullable = nullable ,
291287 )
@@ -614,7 +610,9 @@ def _from_sqlglot_DATETIME(
614610 )
615611
616612 @classmethod
617- def _from_sqlglot_TIMESTAMP (cls , nullable : bool | None = None ) -> dt .Timestamp :
613+ def _from_sqlglot_TIMESTAMP (
614+ cls , scale : int | None = None , nullable : bool | None = None
615+ ) -> dt .Timestamp :
618616 return dt .Timestamp (timezone = "UTC" , nullable = nullable )
619617
620618 @classmethod
@@ -635,7 +633,9 @@ class DuckDBType(SqlglotType):
635633 unknown_type_strings = FrozenDict ({"wkb_blob" : dt .binary })
636634
637635 @classmethod
638- def _from_sqlglot_TIMESTAMP (cls , nullable : bool | None = None ) -> dt .Timestamp :
636+ def _from_sqlglot_TIMESTAMP (
637+ cls , scale : int | None = None , nullable : bool | None = None
638+ ) -> dt .Timestamp :
639639 return dt .Timestamp (scale = 6 , nullable = nullable )
640640
641641 @classmethod
@@ -923,7 +923,9 @@ def _from_sqlglot_DATETIME(cls, nullable: bool | None = None) -> dt.Timestamp:
923923 return dt .Timestamp (timezone = None , nullable = nullable )
924924
925925 @classmethod
926- def _from_sqlglot_TIMESTAMP (cls , nullable : bool | None = None ) -> dt .Timestamp :
926+ def _from_sqlglot_TIMESTAMP (
927+ cls , scale : int | None = None , nullable : bool | None = None
928+ ) -> dt .Timestamp :
927929 return dt .Timestamp (timezone = None , nullable = nullable )
928930
929931 @classmethod
@@ -1147,7 +1149,9 @@ def _from_sqlglot_DATETIME(cls, n=None, nullable: bool | None = None):
11471149 )
11481150
11491151 @classmethod
1150- def _from_sqlglot_TIMESTAMP (cls ):
1152+ def _from_sqlglot_TIMESTAMP (
1153+ cls , scale : int | None = None , nullable : bool | None = None
1154+ ):
11511155 return dt .Binary (nullable = False )
11521156
11531157 @classmethod
0 commit comments