Skip to content

Commit eed5e0f

Browse files
committed
Ability to create Date type from datetime
1 parent 2582597 commit eed5e0f

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

ydb/types.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -33,8 +33,10 @@ def _from_date(x: ydb_value_pb2.Value, table_client_settings: table.TableClientS
3333
return x.uint32_value
3434

3535

36-
def _to_date(pb: ydb_value_pb2.Value, value: typing.Union[date, int]) -> None:
37-
if isinstance(value, date):
36+
def _to_date(pb: ydb_value_pb2.Value, value: typing.Union[date, datetime, int]) -> None:
37+
if isinstance(value, datetime):
38+
pb.uint32_value = (value.date() - _EPOCH.date()).days
39+
elif isinstance(value, date):
3840
pb.uint32_value = (value - _EPOCH.date()).days
3941
else:
4042
pb.uint32_value = value

0 commit comments

Comments
 (0)