Fixes

Author: LuckySting

ydb_sqlalchemy/dbapi/cursor.py

@@ -304,7 +304,7 @@ def executescript(self, script):
         return self.execute(script)
 
     def fetchone(self):
-        return next(self.rows or [], None)
+        return next(self.rows or iter([]), None)
 
     def fetchmany(self, size=None):
         return list(itertools.islice(self.rows, size or self.arraysize))

ydb_sqlalchemy/sqlalchemy/__init__.py

@@ -560,7 +560,7 @@ def _get_column_info(t):
 
 class YdbScanQueryCharacteristic(characteristics.ConnectionCharacteristic):
     def reset_characteristic(self, dialect: "YqlDialect", dbapi_connection: dbapi.Connection) -> None:
-        dialect.reset_isolation_level(dbapi_connection)
+        dialect.reset_ydb_scan_query(dbapi_connection)
 
     def set_characteristic(self, dialect: "YqlDialect", dbapi_connection: dbapi.Connection, value: bool) -> None:
         dialect.set_ydb_scan_query(dbapi_connection, value)
@@ -745,7 +745,7 @@ def get_isolation_level(self, dbapi_connection: dbapi.Connection) -> str:
     def set_ydb_scan_query(self, dbapi_connection: dbapi.Connection, value: bool) -> None:
         dbapi_connection.set_ydb_scan_query(value)
 
-    def reset_isolation_level(self, dbapi_connection: dbapi.Connection):
+    def reset_ydb_scan_query(self, dbapi_connection: dbapi.Connection):
         self.set_ydb_scan_query(dbapi_connection, False)
 
     def get_ydb_scan_query(self, dbapi_connection: dbapi.Connection) -> str: