Hack to support reuse of memory command, closes #643

sqlite_utils/cli.py

@@ -1815,6 +1815,10 @@ def query(
     )
 
 
+# So named memory db "sqlite_utils_memory" isn't dropped when it goes out of scope
+_sqlite_utils_memory_db = []
+
+
 @cli.command()
 @click.argument(
     "paths",
@@ -1921,7 +1925,14 @@ def memory(
     \b
         sqlite-utils memory animals.csv --schema
     """
-    db = sqlite_utils.Database(memory=True)
+    if getattr(sys, "_sqlite_utils_memory_test", False) or not getattr(
+        sys, "_called_from_test", False
+    ):
+        db = sqlite_utils.Database(memory_name="sqlite_utils_memory")
+        _sqlite_utils_memory_db.append(db)
+    else:
+        db = sqlite_utils.Database(memory=True)
+
     # If --dump or --save or --analyze used but no paths detected, assume SQL query is a path:
     if (dump or save or schema or analyze) and not paths:
         paths = [sql]
@@ -1954,6 +1965,7 @@ def memory(
             rows = tracker.wrap(rows)
         if flatten:
             rows = (_flatten(row) for row in rows)
+
         db[file_table].insert_all(rows, alter=True)
         if tracker is not None:
             db[file_table].transform(types=tracker.types)
@@ -1964,6 +1976,7 @@ def memory(
         for view_name in view_names:
             if not db[view_name].exists():
                 db.create_view(view_name, "select * from [{}]".format(file_table))
+
         if fp:
             fp.close()
 

sqlite_utils/db.py

@@ -325,6 +325,8 @@ class Database:
         execute_plugins: bool = True,
         strict: bool = False,
     ):
+        self.memory_name = None
+        self.memory = False
         assert (filename_or_conn is not None and (not memory and not memory_name)) or (
             filename_or_conn is None and (memory or memory_name)
         ), "Either specify a filename_or_conn or pass memory=True"
@@ -335,8 +337,11 @@ class Database:
                 uri=True,
                 check_same_thread=False,
             )
+            self.memory = True
+            self.memory_name = memory_name
         elif memory or filename_or_conn == ":memory:":
             self.conn = sqlite3.connect(":memory:")
+            self.memory = True
         elif isinstance(filename_or_conn, (str, pathlib.Path)):
             if recreate and os.path.exists(filename_or_conn):
                 try: