Fix compairing tz-naive and tz-aware datetimes in Pydantic models

Author: dolfinus

data_rentgen/server/schemas/v1/lineage.py

@@ -71,7 +71,7 @@ class BaseLineageQueryV1(BaseModel):
     @classmethod
     def _check_until(cls, value: datetime | None, info: ValidationInfo) -> datetime | None:
         since = info.data.get("since")
-        if since and value and since >= value:
+        if since and value and since.timestamp() >= value.timestamp():
             msg = "'since' should be less than 'until'"
             raise ValueError(msg)
         return value

data_rentgen/server/schemas/v1/operation.py

@@ -127,7 +127,7 @@ class OperationQueryV1(PaginateQueryV1):
     @classmethod
     def _check_until(cls, value: datetime | None, info: ValidationInfo) -> datetime | None:
         since = info.data.get("since")
-        if since and value and since >= value:
+        if since and value and since.timestamp() >= value.timestamp():
             msg = "'since' should be less than 'until'"
             raise ValueError(msg)
         return value

data_rentgen/server/schemas/v1/run.py

@@ -218,7 +218,7 @@ class RunsPaginateQueryV1(PaginateQueryV1):
     @classmethod
     def _check_until(cls, value: datetime | None, info: ValidationInfo) -> datetime | None:
         since = info.data.get("since")
-        if since and value and since >= value:
+        if since and value and since.timestamp() >= value.timestamp():
             msg = "'since' should be less than 'until'"
             raise ValueError(msg)
         return value