Fixes #28

Florents-Tselai · Jul 8, 2024 · 03845b0 · 03845b0
1 parent a9015dd
commit 03845b0
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/tests/test_tsellm.py b/tests/test_tsellm.py
@@ -315,6 +315,10 @@ def setUp(self):
         print(self.db_fp)
         self.path_args = (self.db_fp,)
 
+    def test_duckdb_is_picked_up(self):
+        # https://github.com/Florents-Tselai/tsellm/issues/28
+        super().test_cli_execute_sql()
+
 
 if __name__ == "__main__":
     unittest.main()
diff --git a/tsellm/cli.py b/tsellm/cli.py
@@ -361,8 +361,9 @@ def cli(*args):
     if args.sqlite and args.duckdb:
         raise ValueError("Only one of --sqlite and --duckdb can be specified.")
 
+    sniffer = DBSniffer(args.filename)
     console = (
-        DuckDBConsole(args.filename) if args.duckdb else SQLiteConsole(args.filename)
+        DuckDBConsole(args.filename) if (args.duckdb or sniffer.is_duckdb) else SQLiteConsole(args.filename)
     )
 
     try: