Merge pull request #5 from arunmathaisk/origin/test123

fix(pre-commit): unwanted ruff changes reverted.

agent/database.py

@@ -154,6 +154,8 @@ def modify_user_permissions(self, username: str, mode: str, permissions: dict |
         self._run_sql(queries_str, commit=True, allow_all_stmt_types=True)
 
     def fetch_database_table_sizes(self) -> dict:
+        from agent.utils import cint
+
         data = self._run_sql(
             f"SELECT table_name, data_length, index_length FROM INFORMATION_SCHEMA.TABLES WHERE TABLE_SCHEMA"
             f"='{self.database_name}'",
@@ -165,12 +167,12 @@ def fetch_database_table_sizes(self) -> dict:
         tables = {}
         for d in data:
             tables[d["table_name"]] = {
-                "data_length": int(d["data_length"]),
-                "index_length": int(d["index_length"]),
-                "total_size": int(d["data_length"]) + int(d["index_length"]),
+                "data_length": cint(d["data_length"]),
+                "index_length": cint(d["index_length"]),
+                "total_size": cint(d["data_length"]) + cint(d["index_length"]),
             }
-            d["data_length"] = int(d["data_length"])
-            d["index_length"] = int(d["index_length"])
+            d["data_length"] = cint(d["data_length"])
+            d["index_length"] = cint(d["index_length"])
             d["total_size"] = d["data_length"] + d["index_length"]
         return tables
 

agent/utils.py

@@ -65,6 +65,8 @@ def get_size(folder):
 
 def cint(x):
     """Convert to integer"""
+    if x is None:
+        return 0
     try:
         num = int(float(x))
     except Exception: