@@ -253,25 +253,21 @@ def _from_sqlglot_STRUCT(
253
253
254
254
@classmethod
255
255
def _from_sqlglot_TIMESTAMP (
256
- cls , scale = None , nullable : bool | None = None
256
+ cls , scale : int | None = None , nullable : bool | None = None
257
257
) -> dt .Timestamp :
258
258
return dt .Timestamp (
259
259
scale = cls .default_temporal_scale if scale is None else int (scale .this .this ),
260
260
nullable = nullable ,
261
261
)
262
262
263
263
@classmethod
264
- def _from_sqlglot_TIMESTAMPTZ (
264
+ def _from_sqlglot_TIMESTAMPNTZ (
265
265
cls , scale = None , nullable : bool | None = None
266
266
) -> 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 )
272
268
273
269
@classmethod
274
- def _from_sqlglot_TIMESTAMPLTZ (
270
+ def _from_sqlglot_TIMESTAMPTZ (
275
271
cls , scale = None , nullable : bool | None = None
276
272
) -> dt .Timestamp :
277
273
return dt .Timestamp (
@@ -281,11 +277,11 @@ def _from_sqlglot_TIMESTAMPLTZ(
281
277
)
282
278
283
279
@classmethod
284
- def _from_sqlglot_TIMESTAMPNTZ (
280
+ def _from_sqlglot_TIMESTAMPLTZ (
285
281
cls , scale = None , nullable : bool | None = None
286
282
) -> dt .Timestamp :
287
283
return dt .Timestamp (
288
- timezone = None ,
284
+ timezone = "UTC" ,
289
285
scale = cls .default_temporal_scale if scale is None else int (scale .this .this ),
290
286
nullable = nullable ,
291
287
)
@@ -614,7 +610,9 @@ def _from_sqlglot_DATETIME(
614
610
)
615
611
616
612
@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 :
618
616
return dt .Timestamp (timezone = "UTC" , nullable = nullable )
619
617
620
618
@classmethod
@@ -635,7 +633,9 @@ class DuckDBType(SqlglotType):
635
633
unknown_type_strings = FrozenDict ({"wkb_blob" : dt .binary })
636
634
637
635
@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 :
639
639
return dt .Timestamp (scale = 6 , nullable = nullable )
640
640
641
641
@classmethod
@@ -923,7 +923,9 @@ def _from_sqlglot_DATETIME(cls, nullable: bool | None = None) -> dt.Timestamp:
923
923
return dt .Timestamp (timezone = None , nullable = nullable )
924
924
925
925
@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 :
927
929
return dt .Timestamp (timezone = None , nullable = nullable )
928
930
929
931
@classmethod
@@ -1147,7 +1149,9 @@ def _from_sqlglot_DATETIME(cls, n=None, nullable: bool | None = None):
1147
1149
)
1148
1150
1149
1151
@classmethod
1150
- def _from_sqlglot_TIMESTAMP (cls ):
1152
+ def _from_sqlglot_TIMESTAMP (
1153
+ cls , scale : int | None = None , nullable : bool | None = None
1154
+ ):
1151
1155
return dt .Binary (nullable = False )
1152
1156
1153
1157
@classmethod
0 commit comments