fix type conversions

Author: catsona

lib/dl_connector_ydb/dl_connector_ydb/core/base/adapter.py

@@ -103,8 +103,12 @@ def _convert_bytes(value: bytes) -> str:
         return value.decode("utf-8", errors="replace")
 
     @staticmethod
-    def _convert_interval(value: datetime.timedelta) -> int:
-        return int(value.total_seconds() * 1_000_000)
+    def _convert_interval(value: datetime.timedelta | int) -> int:
+        if value is None:
+            return None
+        if isinstance(value, datetime.timedelta):
+            return int(value.total_seconds() * 1_000_000)
+        return value
 
     @staticmethod
     def _convert_ts(value: int | datetime.datetime) -> datetime.datetime:

lib/dl_sqlalchemy_ydb/dl_sqlalchemy_ydb/dialect.py

@@ -33,10 +33,12 @@ class YqlInterval(sa.types.Interval):
     __visit_name__ = "interval"
 
     def result_processor(self, dialect: sa.engine.Dialect, coltype: typing.Any) -> typing.Any:
-        def process(value: typing.Optional[datetime.timedelta]) -> typing.Optional[int]:
+        def process(value: typing.Optional[datetime.timedelta] | int) -> typing.Optional[int]:
             if value is None:
                 return None
-            return int(value.total_seconds() * 1_000_000)
+            if isinstance(value, datetime.timedelta):
+                return int(value.total_seconds() * 1_000_000)
+            return value
 
         return process