Implement flag handling and migration status checks; refactor database test functions and update requirements

flags/status.py

@@ -0,0 +1,50 @@
+import threading
+from watchdog.events import PatternMatchingEventHandler
+from watchdog.observers import Observer
+import os
+import time
+
+FLAGS_DIR = os.path.dirname(os.path.abspath(__file__))
+
+
+
+class FlagHandler(PatternMatchingEventHandler):
+    def __init__(self, shared_state, lock):
+        super().__init__(patterns=["*.flag"], ignore_directories=True)
+        self.shared_state = shared_state
+        self.lock = lock
+
+    def on_created(self, event):
+        try:
+            with open(event.src_path, "r") as f:
+                content = f.read().strip()
+        except FileNotFoundError:
+            return
+        try:
+            os.remove(event.src_path)
+        except FileNotFoundError:
+            pass
+        with self.lock:
+            self.shared_state["current"] = content
+        time.sleep(2)
+
+shared_state = {"current": None}
+lock = threading.Lock()
+
+observer = Observer()
+observer.schedule(FlagHandler(shared_state, lock), path=FLAGS_DIR, recursive=False)
+observer.start()
+
+def get_status():
+    with lock:
+        current = shared_state["current"]
+    if current == "db_check_done":
+        return False
+    return current
+
+def get_migration():
+    with lock:
+            current = shared_state["current"]
+    if current == "migration_done":
+        return True
+    return False