sqlsaber 0.10.0__py3-none-any.whl → 0.12.0__py3-none-any.whl

sqlsaber/__init__.py

@@ -1,3 +1 @@
-"""SQLSaber CLI - SQL like Claude Code."""
-
-__version__ = "0.1.0"
+"""SQLSaber CLI - Agentic SQL assistant like Claude Code but for SQL."""

sqlsaber/cli/commands.py

@@ -43,7 +43,7 @@ def meta_handler(
         str | None,
         cyclopts.Parameter(
             ["--database", "-d"],
-            help="Database connection name (uses default if not specified)",
+            help="Database connection name or direct file path for CSV/SQLite files (uses default if not specified)",
         ),
     ] = None,
 ):
@@ -54,6 +54,8 @@ def meta_handler(
         saber                                  # Start interactive mode
         saber "show me all users"              # Run a single query with default database
         saber -d mydb "show me users"          # Run a query with specific database
+        saber -d data.csv "show me users"      # Run a query with ad-hoc CSV file
+        saber -d data.db "show me users"       # Run a query with ad-hoc SQLite file
         echo "show me all users" | saber       # Read query from stdin
         cat query.txt | saber                  # Read query from file via stdin
     """
@@ -73,7 +75,7 @@ def query(
         str | None,
         cyclopts.Parameter(
             ["--database", "-d"],
-            help="Database connection name (uses default if not specified)",
+            help="Database connection name or direct file path for CSV/SQLite files (uses default if not specified)",
         ),
     ] = None,
 ):
@@ -88,6 +90,8 @@ def query(
     Examples:
         saber                             # Start interactive mode
         saber "show me all users"         # Run a single query
+        saber -d data.csv "show users"    # Run a query with ad-hoc CSV file
+        saber -d data.db "show users"     # Run a query with ad-hoc SQLite file
         echo "show me all users" | saber  # Read query from stdin
     """
 
@@ -101,7 +105,7 @@ def query(
                 # If stdin was empty, fall back to interactive mode
                 actual_query = None
 
-        # Get database configuration or handle direct CSV file
+        # Get database configuration or handle direct file paths
         if database:
             # Check if this is a direct CSV file path
             if database.endswith(".csv"):
@@ -110,6 +114,13 @@ def query(
                     raise CLIError(f"CSV file '{database}' not found.")
                 connection_string = f"csv:///{csv_path}"
                 db_name = csv_path.stem
+            # Check if this is a direct SQLite file path
+            elif database.endswith((".db", ".sqlite", ".sqlite3")):
+                sqlite_path = Path(database).expanduser().resolve()
+                if not sqlite_path.exists():
+                    raise CLIError(f"SQLite file '{database}' not found.")
+                connection_string = f"sqlite:///{sqlite_path}"
+                db_name = sqlite_path.stem
             else:
                 # Look up configured database connection
                 db_config = config_manager.get_database(database)

sqlsaber/cli/display.py

@@ -125,7 +125,6 @@ class DisplayManager:
                 {"name": "Schema", "style": "cyan"},
                 {"name": "Table Name", "style": "white"},
                 {"name": "Type", "style": "yellow"},
-                {"name": "Row Count", "justify": "right", "style": "magenta"},
             ]
             table = self._create_table(columns)
 
@@ -134,12 +133,8 @@ class DisplayManager:
                 schema = table_info.get("schema", "")
                 name = table_info.get("name", "")
                 table_type = table_info.get("type", "")
-                row_count = table_info.get("row_count", 0)
 
-                # Format row count with commas for readability
-                formatted_count = f"{row_count:,}" if row_count else "0"
-
-                table.add_row(schema, name, table_type, formatted_count)
+                table.add_row(schema, name, table_type)
 
             self.console.print(table)
 

sqlsaber/database/schema.py

@@ -42,7 +42,7 @@ class BaseSchemaIntrospector(ABC):
         pass
 
     @abstractmethod
-    async def list_tables_info(self, connection) -> dict[str, Any]:
+    async def list_tables_info(self, connection) -> list[dict[str, Any]]:
         """Get list of tables with basic information."""
         pass
 
@@ -182,32 +182,31 @@ class PostgreSQLSchemaIntrospector(BaseSchemaIntrospector):
             """
             return await conn.fetch(pk_query)
 
-    async def list_tables_info(self, connection) -> dict[str, Any]:
+    async def list_tables_info(self, connection) -> list[dict[str, Any]]:
         """Get list of tables with basic information for PostgreSQL."""
         pool = await connection.get_pool()
         async with pool.acquire() as conn:
-            # Get tables with row counts
+            # Get tables without row counts for better performance
             tables_query = """
-                WITH table_stats AS (
-                    SELECT
-                        schemaname,
-                        relname as tablename,
-                        n_live_tup as approximate_row_count
-                    FROM pg_stat_user_tables
-                )
                 SELECT
                     t.table_schema,
                     t.table_name,
-                    t.table_type,
-                    COALESCE(ts.approximate_row_count, 0) as row_count
+                    t.table_type
                 FROM information_schema.tables t
-                LEFT JOIN table_stats ts
-                    ON t.table_schema = ts.schemaname
-                    AND t.table_name = ts.tablename
                 WHERE t.table_schema NOT IN ('pg_catalog', 'information_schema')
                 ORDER BY t.table_schema, t.table_name;
             """
-            return await conn.fetch(tables_query)
+            records = await conn.fetch(tables_query)
+
+            # Convert asyncpg.Record objects to dictionaries
+            return [
+                {
+                    "table_schema": record["table_schema"],
+                    "table_name": record["table_name"],
+                    "table_type": record["table_type"],
+                }
+                for record in records
+            ]
 
 
 class MySQLSchemaIntrospector(BaseSchemaIntrospector):
@@ -353,24 +352,33 @@ class MySQLSchemaIntrospector(BaseSchemaIntrospector):
                 await cursor.execute(pk_query)
                 return await cursor.fetchall()
 
-    async def list_tables_info(self, connection) -> dict[str, Any]:
+    async def list_tables_info(self, connection) -> list[dict[str, Any]]:
         """Get list of tables with basic information for MySQL."""
         pool = await connection.get_pool()
         async with pool.acquire() as conn:
             async with conn.cursor() as cursor:
-                # Get tables with row counts
+                # Get tables without row counts for better performance
                 tables_query = """
                     SELECT
                         t.table_schema,
                         t.table_name,
-                        t.table_type,
-                        COALESCE(t.table_rows, 0) as row_count
+                        t.table_type
                     FROM information_schema.tables t
                     WHERE t.table_schema NOT IN ('information_schema', 'performance_schema', 'mysql', 'sys')
                     ORDER BY t.table_schema, t.table_name;
                 """
                 await cursor.execute(tables_query)
-                return await cursor.fetchall()
+                rows = await cursor.fetchall()
+
+                # Convert rows to dictionaries
+                return [
+                    {
+                        "table_schema": row["table_schema"],
+                        "table_name": row["table_name"],
+                        "table_type": row["table_type"],
+                    }
+                    for row in rows
+                ]
 
 
 class SQLiteSchemaIntrospector(BaseSchemaIntrospector):
@@ -496,9 +504,9 @@ class SQLiteSchemaIntrospector(BaseSchemaIntrospector):
 
         return primary_keys
 
-    async def list_tables_info(self, connection) -> dict[str, Any]:
+    async def list_tables_info(self, connection) -> list[dict[str, Any]]:
         """Get list of tables with basic information for SQLite."""
-        # First get the table names
+        # Get table names without row counts for better performance
         tables_query = """
             SELECT
                 'main' as table_schema,
@@ -512,34 +520,15 @@ class SQLiteSchemaIntrospector(BaseSchemaIntrospector):
 
         tables = await self._execute_query(connection, tables_query)
 
-        # Now get row counts for each table
-        result = []
-        for table in tables:
-            table_name = table["table_name"]
-            table_type = table["table_type"]
-
-            # Only count rows for tables, not views
-            if table_type.lower() == "table":
-                try:
-                    count_query = f"SELECT COUNT(*) as count FROM [{table_name}]"
-                    count_result = await self._execute_query(connection, count_query)
-                    row_count = count_result[0]["count"] if count_result else 0
-                except Exception:
-                    # If count fails (e.g., table locked), default to 0
-                    row_count = 0
-            else:
-                # For views, we don't count rows as it could be expensive
-                row_count = 0
-
-            result.append(
-                {
-                    "table_schema": table["table_schema"],
-                    "table_name": table_name,
-                    "table_type": table_type,
-                    "row_count": row_count,
-                }
-            )
-        return result
+        # Convert to expected format
+        return [
+            {
+                "table_schema": table["table_schema"],
+                "table_name": table["table_name"],
+                "table_type": table["table_type"],
+            }
+            for table in tables
+        ]
 
 
 class SchemaManager:
@@ -682,7 +671,7 @@ class SchemaManager:
                 )
 
     async def list_tables(self) -> dict[str, Any]:
-        """Get a list of all tables with basic information like row counts."""
+        """Get a list of all tables with basic information."""
         # Check cache first
         cache_key = "list_tables"
         cached_data = self._get_cached_tables(cache_key)
@@ -702,7 +691,6 @@ class SchemaManager:
                     "name": table["table_name"],
                     "full_name": f"{table['table_schema']}.{table['table_name']}",
                     "type": table["table_type"],
-                    "row_count": table["row_count"],
                 }
             )
 

{sqlsaber-0.10.0.dist-info → sqlsaber-0.12.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: sqlsaber
-Version: 0.10.0
+Version: 0.12.0
 Summary: SQLSaber - Agentic SQL assistant like Claude Code
 License-File: LICENSE
 Requires-Python: >=3.12
@@ -21,6 +21,15 @@ Description-Content-Type: text/markdown
 
 # SQLSaber
 
+```
+███████  ██████  ██      ███████  █████  ██████  ███████ ██████
+██      ██    ██ ██      ██      ██   ██ ██   ██ ██      ██   ██
+███████ ██    ██ ██      ███████ ███████ ██████  █████   ██████
+     ██ ██ ▄▄ ██ ██           ██ ██   ██ ██   ██ ██      ██   ██
+███████  ██████  ███████ ███████ ██   ██ ██████  ███████ ██   ██
+            ▀▀
+```
+
 > Use the agent Luke!
 
 SQLSaber is an agentic SQL assistant. Think Claude Code but for SQL.

{sqlsaber-0.10.0.dist-info → sqlsaber-0.12.0.dist-info}/RECORD

@@ -1,4 +1,4 @@
-sqlsaber/__init__.py,sha256=QCFi8xTVMohelfi7zOV1-6oLCcGoiXoOcKQY-HNBCk8,66
+sqlsaber/__init__.py,sha256=HjS8ULtP4MGpnTL7njVY45NKV9Fi4e_yeYuY-hyXWQc,73
 sqlsaber/__main__.py,sha256=RIHxWeWh2QvLfah-2OkhI5IJxojWfy4fXpMnVEJYvxw,78
 sqlsaber/agents/__init__.py,sha256=LWeSeEUE4BhkyAYFF3TE-fx8TtLud3oyEtyB8ojFJgo,167
 sqlsaber/agents/anthropic.py,sha256=CBHneR5NJhu155d0-D1mSGOcTH7kmbXZSLv2mVQotSM,22128
@@ -7,10 +7,10 @@ sqlsaber/agents/mcp.py,sha256=FKtXgDrPZ2-xqUYCw2baI5JzrWekXaC5fjkYW1_Mg50,827
 sqlsaber/agents/streaming.py,sha256=LaSeMTlxuJFRArJVqDly5-_KgcePiCCKPKfMxfB4oGs,521
 sqlsaber/cli/__init__.py,sha256=qVSLVJLLJYzoC6aj6y9MFrzZvAwc4_OgxU9DlkQnZ4M,86
 sqlsaber/cli/auth.py,sha256=1yvawtS5NGj9dWMRZ8I5T6sqBiRqZpPsjEPrJSQBJAs,4979
-sqlsaber/cli/commands.py,sha256=84KpBtrcoprWVuQjlGuzR6D34Ep2fCf8G1pmxrThswE,6329
+sqlsaber/cli/commands.py,sha256=rhbfpXNrWLQbiEEqn4GDwE175vhOWrfPiTRP32Ui7NM,7156
 sqlsaber/cli/completers.py,sha256=HsUPjaZweLSeYCWkAcgMl8FylQ1xjWBWYTEL_9F6xfU,6430
 sqlsaber/cli/database.py,sha256=tJ8rqGrafZpg3VgDmSiq7eZoPscoGAW3XLTYGoQw8LE,12910
-sqlsaber/cli/display.py,sha256=wC-xYmmD21XyAkpRdMW7Ch2Mn5SlM1X34pbpka2ZIX8,10083
+sqlsaber/cli/display.py,sha256=HtXwPe3VPUh2EJpyvpJVWyisCanu9O7w-rkqq7Y4UaY,9791
 sqlsaber/cli/interactive.py,sha256=NjrhzXt6YK4WL17DMkJTY5dA1v9QMxLdkI1XgPKK8YA,8017
 sqlsaber/cli/memory.py,sha256=OufHFJFwV0_GGn7LvKRTJikkWhV1IwNIUDOxFPHXOaQ,7794
 sqlsaber/cli/models.py,sha256=HByezaeKqj65GzB_FmWuugjkgTq2Pvab_mzNZnHxya0,7690
@@ -30,7 +30,7 @@ sqlsaber/config/oauth_tokens.py,sha256=C9z35hyx-PvSAYdC1LNf3rg9_wsEIY56hkEczelba
 sqlsaber/config/settings.py,sha256=gKhGlErzsBk39RoRSy1b8pb-bN2K7HIaPaBgbJDhY4M,4753
 sqlsaber/database/__init__.py,sha256=a_gtKRJnZVO8-fEZI7g3Z8YnGa6Nio-5Y50PgVp07ss,176
 sqlsaber/database/connection.py,sha256=sZVGNMzMwiM11GrsLLPwR8A5ugzJ5O0TCdkrt0KVRuI,15123
-sqlsaber/database/schema.py,sha256=B4emtbaNiqjz6aGBUQYYwARsTMqBilvWSurNg_zKu9U,28600
+sqlsaber/database/schema.py,sha256=anlxjr_4-gEWlEjQgZ5h4KGMfbS0-aQH1VwhXIw41Q4,27951
 sqlsaber/mcp/__init__.py,sha256=COdWq7wauPBp5Ew8tfZItFzbcLDSEkHBJSMhxzy8C9c,112
 sqlsaber/mcp/mcp.py,sha256=YH4crygqb5_Y94nsns6d-26FZCTlDPOh3tf-ghihzDM,4440
 sqlsaber/memory/__init__.py,sha256=GiWkU6f6YYVV0EvvXDmFWe_CxarmDCql05t70MkTEWs,63
@@ -39,8 +39,8 @@ sqlsaber/memory/storage.py,sha256=ne8szLlGj5NELheqLnI7zu21V8YS4rtpYGGC7tOmi-s,57
 sqlsaber/models/__init__.py,sha256=RJ7p3WtuSwwpFQ1Iw4_DHV2zzCtHqIzsjJzxv8kUjUE,287
 sqlsaber/models/events.py,sha256=89SXKb5GGpH01yTr2kPEBhzp9xv35RFIYuFdAZSIPoE,721
 sqlsaber/models/types.py,sha256=w-zk81V2dtveuteej36_o1fDK3So428j3P2rAejU62U,862
-sqlsaber-0.10.0.dist-info/METADATA,sha256=mnHp2BteZvCNwoG3mUW-ulbBFxvQa4i5JkCffUHCEhc,6150
-sqlsaber-0.10.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-sqlsaber-0.10.0.dist-info/entry_points.txt,sha256=qEbOB7OffXPFgyJc7qEIJlMEX5RN9xdzLmWZa91zCQQ,162
-sqlsaber-0.10.0.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-sqlsaber-0.10.0.dist-info/RECORD,,
+sqlsaber-0.12.0.dist-info/METADATA,sha256=k_OjB22bXlyVvH_nPTEIBjLwwzmdsAntYs6_rl7LSzg,6877
+sqlsaber-0.12.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+sqlsaber-0.12.0.dist-info/entry_points.txt,sha256=qEbOB7OffXPFgyJc7qEIJlMEX5RN9xdzLmWZa91zCQQ,162
+sqlsaber-0.12.0.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+sqlsaber-0.12.0.dist-info/RECORD,,