Merge branch 'clear-data' into add-GUI

.gitignore

@@ -21,4 +21,5 @@ cache
 performance
 
 # Generated when adding editable dependencies in requirements.txt (-e)
-src
+src
+.DS_Store

openadapt/config.py

@@ -13,7 +13,7 @@
     "DB_ECHO": False,
     "DB_FNAME": "openadapt.db",
     "OPENAI_API_KEY": "<set your api key in .env>",
-    #"OPENAI_MODEL_NAME": "gpt-4",
+    # "OPENAI_MODEL_NAME": "gpt-4",
     "OPENAI_MODEL_NAME": "gpt-3.5-turbo",
     # may incur significant performance penalty
     "RECORD_READ_ACTIVE_ELEMENT_STATE": False,
@@ -36,6 +36,7 @@ def getenv_fallback(var_name):
     locals()[key] = val
 
 ROOT_DIRPATH = pathlib.Path(__file__).parent.parent.resolve()
+DB_FNAME = getenv_fallback("DB_FNAME")
 DB_FPATH = ROOT_DIRPATH / DB_FNAME
 DB_URL = f"sqlite:///{DB_FPATH}"
 

openadapt/scripts/reset_db.py

@@ -0,0 +1,25 @@
+import os
+from subprocess import run, PIPE
+from openadapt.config import getenv_fallback
+
+
+def reset_db():
+    """
+    The function clears the database by removing the database file and running a
+    database migration using Alembic.
+    """
+    db = getenv_fallback("DB_FNAME")
+    if os.path.exists(db):
+        os.remove(db)
+
+    # Prevents duplicate logging of config values by piping stderr and filtering the output.
+    result = run(["alembic", "upgrade", "head"], stderr=PIPE, text=True)
+    print(result.stderr[result.stderr.find("INFO  [alembic") :])
+    if result.returncode != 0:
+        raise RuntimeError("Database migration failed.")
+    else:
+        print("Database cleared successfully.")
+
+
+if __name__ == "__main__":
+    reset_db()