Fix pipeline attempt (6): Backup

tests/test_backup.py

@@ -16,22 +16,14 @@ def test_backup_creates_zip(tmp_path, monkeypatch):
     monkeypatch.setattr(backup, 'WEB_FOLDER', web_folder)
     monkeypatch.setattr(backup, 'BACKUP_FOLDER', backup_folder)
 
-    # Manually execute the backup logic
+    # Call the backup function
+    backup.create_backup()
+    
+    # Check that backup was created
     today = datetime.date.today().strftime('%Y%m%d')
     backup_name = f'backup-{today}.zip'
     backup_path = backup_folder / backup_name
 
-    # Remove if exists
-    if backup_path.exists():
-        backup_path.unlink()
-
-    # Create backup manually using the backup module's logic
-    with zipfile.ZipFile(backup_path, "w", zipfile.ZIP_DEFLATED) as zipf:
-        for path in web_folder.rglob("*"):
-            if path.is_file():
-                arcname = path.relative_to(web_folder.parent)
-                zipf.write(path, arcname)
-
     assert backup_path.exists()
     with zipfile.ZipFile(backup_path, 'r') as z:
         names = z.namelist()

utils/backup.py

@@ -5,15 +5,18 @@ from pathlib import Path
 WEB_FOLDER = Path("plots")
 BACKUP_FOLDER = Path("backups")
 
-today = datetime.date.today().strftime("%Y%m%d")
-backup_name = f"backup-{today}.zip"
-backup_path = BACKUP_FOLDER / backup_name
 
-BACKUP_FOLDER.mkdir(parents=True, exist_ok=True)
+def create_backup():
+    """Create a backup of the web folder."""
+    today = datetime.date.today().strftime("%Y%m%d")
+    backup_name = f"backup-{today}.zip"
+    backup_path = BACKUP_FOLDER / backup_name
 
-if backup_path.exists():
+    BACKUP_FOLDER.mkdir(parents=True, exist_ok=True)
+
+    if backup_path.exists():
         print(f"Backup already exists: {backup_path}")
-else:
+    else:
         print(f"Creating backup: {backup_path}")
         with zipfile.ZipFile(backup_path, "w", zipfile.ZIP_DEFLATED) as zipf:
             for path in WEB_FOLDER.rglob("*"):
@@ -21,3 +24,7 @@ else:
                     arcname = path.relative_to(WEB_FOLDER.parent)
                     zipf.write(path, arcname)
         print("✅ Backup complete.")
+
+
+if __name__ == "__main__":
+    create_backup()