Accept date in literal (#1618)

Co-authored-by: Fokko Driesprong <[email protected]>

Author: TennyZhuang

pyiceberg/expressions/literals.py

@@ -23,7 +23,7 @@
 
 import struct
 from abc import ABC, abstractmethod
-from datetime import datetime
+from datetime import date, datetime
 from decimal import ROUND_HALF_UP, Decimal
 from functools import singledispatchmethod
 from math import isnan
@@ -50,6 +50,7 @@
 )
 from pyiceberg.utils.datetime import (
     date_str_to_days,
+    date_to_days,
     datetime_to_micros,
     micros_to_days,
     time_str_to_micros,
@@ -149,6 +150,8 @@ def literal(value: L) -> Literal[L]:
         return DecimalLiteral(value)
     elif isinstance(value, datetime):
         return TimestampLiteral(datetime_to_micros(value))  # type: ignore
+    elif isinstance(value, date):
+        return DateLiteral(date_to_days(value))  # type: ignore
     else:
         raise TypeError(f"Invalid literal value: {repr(value)}")
 

pyiceberg/typedef.py

@@ -17,7 +17,7 @@
 from __future__ import annotations
 
 from abc import abstractmethod
-from datetime import datetime
+from datetime import date, datetime
 from decimal import Decimal
 from functools import lru_cache
 from typing import (
@@ -94,7 +94,7 @@ def __missing__(self, key: K) -> V:
 """A recursive dictionary type for nested structures in PyIceberg."""
 
 # Represents the literal value
-L = TypeVar("L", str, bool, int, float, bytes, UUID, Decimal, datetime, covariant=True)
+L = TypeVar("L", str, bool, int, float, bytes, UUID, Decimal, datetime, date, covariant=True)
 
 
 @runtime_checkable

tests/expressions/test_literals.py

@@ -910,6 +910,10 @@ def test_literal_from_datetime() -> None:
     assert isinstance(literal(datetime.datetime.now()), TimestampLiteral)
 
 
+def test_literal_from_date() -> None:
+    assert isinstance(literal(datetime.date.today()), DateLiteral)
+
+
 #   __  __      ___
 #  |  \/  |_  _| _ \_  _
 #  | |\/| | || |  _/ || |