added openadapt.renew

.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/renew.py

@@ -0,0 +1,19 @@
+import os
+from subprocess import run, PIPE
+from openadapt.config import getenv_fallback
+
+
+def clear_db():
+    db = getenv_fallback("DB_FNAME")
+    if os.path.exists(db):
+        os.remove(db)
+    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__":
+    clear_db()