nvac removed

Author: wtbates99

src/providers/bigquery_provider.py

@@ -13,7 +13,6 @@ def __init__(self, project_id: str, credentials_path: Optional[str] = None):
             )
             self.client = bigquery.Client(project=project_id, credentials=credentials)
         else:
-            # Use default credentials
             self.client = bigquery.Client(project=project_id)
 
     def execute_query(self, sql_query: str) -> List[Dict[str, Any]]:

src/providers/mysql_provider.py

@@ -17,7 +17,6 @@ def __init__(
         self.database = database
         self.port = port
         try:
-            # Use MySQLConnectionAbstract as the base type for the union
             self.connection: (
                 MySQLConnection | PooledMySQLConnection | MySQLConnectionAbstract
             ) = mysql.connector.connect(
@@ -33,7 +32,6 @@ def __init__(
     def execute_query(self, sql_query: str) -> List[Dict[str, Any]]:
         cursor = self.connection.cursor(dictionary=True)
         cursor.execute(sql_query)
-        # Cast the result to List[Dict[str, Any]] since dictionary=True ensures dicts
         results = cast(List[Dict[str, Any]], cursor.fetchall())
         cursor.close()
         return results
@@ -94,7 +92,6 @@ def get_compact_tables(
                 """,
                 (schema_name,),
             )
-            # Cast to List[Dict[str, str]] since dictionary=True ensures dicts
             results = cast(List[Dict[str, str]], cursor.fetchall())
             if not results:
                 raise Exception(f"No tables found in schema '{schema_name}'")
@@ -115,7 +112,6 @@ def get_compact_tables(
                 """,
                 (schema_name, table_name),
             )
-            # Cast to List[Dict[str, str]] since dictionary=True ensures dicts
             columns = cast(List[Dict[str, str]], cursor.fetchall())
 
             cursor.execute(
@@ -126,7 +122,6 @@ def get_compact_tables(
                 """,
                 (schema_name, table_name),
             )
-            # Cast to Optional[Dict[str, str]] since dictionary=True ensures dict or None
             description_row = cast(Optional[Dict[str, str]], cursor.fetchone())
             table_description = (
                 description_row["description"]

src/providers/postgres_provider.py

@@ -90,7 +90,6 @@ def get_compact_tables(
         """
         cursor = self.connection.cursor()
 
-        # Get all tables if table_names is None
         if table_names is None:
             cursor.execute(
                 """
@@ -106,7 +105,6 @@ def get_compact_tables(
         compact_tables = []
 
         for table_name in table_names:
-            # Get column information
             cursor.execute(
                 """
                 SELECT column_name, data_type, is_nullable
@@ -118,7 +116,6 @@ def get_compact_tables(
             )
             columns = cursor.fetchall()
 
-            # Get table description if available
             cursor.execute(
                 """
                 SELECT obj_description(
@@ -137,10 +134,7 @@ def get_compact_tables(
                 col_name = column[0]
                 col_type = column[1].lower()
 
-                # Map the PostgreSQL type to our compact type
-                mapped_type = type_map.get(
-                    col_type, "S"
-                )  # Default to string if type not found
+                mapped_type = type_map.get(col_type, "S")
                 fields.append({"n": col_name, "t": mapped_type})
 
             compact_tables.append(

src/providers/sqlite_provider.py

@@ -14,7 +14,6 @@ def __init__(self, database_path: str):
         """
         self.database_path = database_path
         self.connection = sqlite3.connect(database_path)
-        # Enable dictionary access to rows
         self.connection.row_factory = sqlite3.Row
 
     def execute_query(self, sql_query: str) -> List[Dict[str, Any]]:
@@ -71,7 +70,6 @@ def get_compact_tables(
         """
         cursor = self.connection.cursor()
 
-        # Get all tables if table_names is None
         if table_names is None:
             cursor.execute(
                 "SELECT name FROM sqlite_master WHERE type='table' AND name NOT LIKE 'sqlite_%'"
@@ -82,32 +80,22 @@ def get_compact_tables(
         compact_tables = []
 
         for table_name in table_names:
-            # Get table info
             cursor.execute(f"PRAGMA table_info('{table_name}')")
             columns = cursor.fetchall()
 
-            # Debug: Print raw column data to see what we're getting
-            print(f"Table: {table_name}, Columns: {columns}")
-
             fields = []
             for column in columns:
-                # PRAGMA table_info returns: (cid, name, type, notnull, dflt_value, pk)
-                col_name = column[1]  # name is at index 1
-                col_type = (
-                    column[2].upper() if column[2] else "TEXT"
-                )  # type is at index 2
-
-                # Map the SQLite type to our compact type
-                mapped_type = type_map.get(
-                    col_type, "S"
-                )  # Default to string if type not found
+                col_name = column[1]
+                col_type = column[2].upper() if column[2] else "TEXT"
+
+                mapped_type = type_map.get(col_type, "S")
+                fields.append({"n": col_name, "t": mapped_type})
                 fields.append({"n": col_name, "t": mapped_type})
 
-            # SQLite doesn't have schemas, so we just use the table name
             compact_tables.append(
                 {
                     "t": table_name,
-                    "d": "",  # SQLite doesn't store table descriptions
+                    "d": "",
                     "f": fields,
                 }
             )