feat: support datetime objects in literal instantiation

pyiceberg/expressions/literals.py

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

tests/expressions/test_literals.py

@@ -906,6 +906,10 @@ def test_uuid_to_binary() -> None:
     assert isinstance(binary_literal, BinaryLiteral)  # type: ignore
 
 
+def test_literal_from_datetime() -> None:
+    assert isinstance(literal(datetime.datetime.now()), TimestampLiteral)
+
+
 #   __  __      ___
 #  |  \/  |_  _| _ \_  _
 #  | |\/| | || |  _/ || |