use timestamp to avoid loops

backend/scripts/backup.py

@@ -1,13 +1,54 @@
 #!/usr/bin/env python3
 import os
 import subprocess
-from huggingface_hub import HfApi
+import datetime
+from huggingface_hub import HfApi, hf_hub_download, hf_hub_url
+
+TIMESTAMP_FILE_PATH = "db_backup/last_backup_time.txt"
+DB_GPG_PATH = "db_backup/webui.db.gpg"
+DB_FILE_PATH = "data/webui.db"
+
+def get_last_backup_time(repo_id, hf_token):
+    """
+    Attempt to download and parse the last_backup_time.txt file
+    from the HF Space. If it doesn't exist, return None.
+    """
+    api = HfApi()
+    try:
+        temp_file = hf_hub_download(
+            repo_id=repo_id,
+            repo_type="space",
+            filename=TIMESTAMP_FILE_PATH,  # "db_backup/last_backup_time.txt"
+            token=hf_token
+        )
+    except Exception as e:
+        # File might not exist or there's some other error
+        print(f"Could not download {TIMESTAMP_FILE_PATH}: {e}")
+        return None
+
+    # If we downloaded successfully, read the timestamp
+    try:
+        with open(temp_file, "r", encoding="utf-8") as f:
+            timestamp_str = f.read().strip()
+        last_backup_dt = datetime.datetime.fromisoformat(timestamp_str)
+        return last_backup_dt
+    except Exception as e:
+        print(f"Error parsing timestamp from {TIMESTAMP_FILE_PATH}: {e}")
+        return None
+
+def save_timestamp_locally():
+    """
+    Save the current UTC time to db_backup/last_backup_time.txt (locally).
+    """
+    now = datetime.datetime.utcnow()
+    os.makedirs("db_backup", exist_ok=True)
+    with open(TIMESTAMP_FILE_PATH, "w", encoding="utf-8") as f:
+        f.write(now.isoformat())
 
 def backup_db():
     passphrase = os.environ.get("BACKUP_PASSPHRASE")
     hf_token = os.environ.get("HF_TOKEN")
     space_id = os.environ.get("SPACE_ID")
-
     if not passphrase:
         raise ValueError("BACKUP_PASSPHRASE is not set.")
     if not hf_token:
@@ -15,30 +56,55 @@ def backup_db():
     if not space_id:
         raise ValueError("SPACE_ID is not set (or define repo_id manually).")
 
-    # Encrypt using GPG
+    # 1) Check if last backup was too recent
+    threshold_hours = 12
+    last_backup_dt = get_last_backup_time(space_id, hf_token)
+    if last_backup_dt is not None:
+        now = datetime.datetime.utcnow()
+        elapsed = now - last_backup_dt
+        if elapsed.total_seconds() < threshold_hours * 3600:
+            print(f"Last backup was only {elapsed.total_seconds()/3600:.2f} hours ago.")
+            print(f"Skipping backup to avoid rebuild loop.")
+            return
+
+    # 2) Encrypt using GPG
     print("Encrypting database with GPG...")
     encrypt_cmd = [
         "gpg", "--batch", "--yes", "--passphrase", passphrase,
         "-c", "--cipher-algo", "AES256",
-        "-o", "db_backup/webui.db.gpg",
-        "data/webui.db"
+        "-o", DB_GPG_PATH,  # "db_backup/webui.db.gpg"
+        DB_FILE_PATH         # "data/webui.db"
     ]
     subprocess.run(encrypt_cmd, check=True)
-    print("Database encrypted successfully to db_backup/webui.db.gpg")
+    print(f"Database encrypted successfully to {DB_GPG_PATH}")
+
+    # 3) Update the timestamp file locally
+    save_timestamp_locally()
 
-    # Upload using huggingface_hub
+    # 4) Upload using huggingface_hub
     print("Uploading to Hugging Face Spaces via huggingface_hub...")
     api = HfApi()
     repo_id = space_id
-    api.upload_file(
-        path_or_fileobj="db_backup/webui.db.gpg",
-        path_in_repo="db_backup/webui.db.gpg",
+
+    # uploads both "db_backup/webui.db.gpg" and "db_backup/last_backup_time.txt"
+    files_to_upload = [
+        (DB_GPG_PATH, DB_GPG_PATH),
+        (TIMESTAMP_FILE_PATH, TIMESTAMP_FILE_PATH)
+    ]
+
+    api.create_commit(
         repo_id=repo_id,
         repo_type="space",
-        token=hf_token,
-        commit_message="Update encrypted database backup"
+        operations=[
+            # Each operation: (operation, local_path, remote_path)
+            # "add_or_update" is the operation
+            {"op": "add_or_update", "path_in_repo": remote, "path_or_fileobj": local}
+            for (local, remote) in files_to_upload
+        ],
+        commit_message="Update encrypted database backup + timestamp",
+        token=hf_token
     )
-    print("DB backup uploaded successfully!")
+    print("DB backup + timestamp uploaded successfully!")
 
 if __name__ == "__main__":
     backup_db()