Skip to content

Commit bb9a84f

Browse files
committed
chore: map timestampntz to timestamp by default
1 parent 80d02bd commit bb9a84f

File tree

1 file changed

+18
-14
lines changed

1 file changed

+18
-14
lines changed

ibis/backends/sql/datatypes.py

Lines changed: 18 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)