dar-backup 0.6.17__py3-none-any.whl → 0.6.19__py3-none-any.whl

dar_backup/dar_backup_systemd.py

@@ -0,0 +1,119 @@
+from pathlib import Path
+import subprocess
+import argparse
+
+SERVICE_TEMPLATE = """[Unit]
+Description=dar-backup {mode}
+StartLimitIntervalSec=120
+StartLimitBurst=1
+
+[Service]
+Type=oneshot
+TimeoutSec=infinity
+RemainAfterExit=no
+ExecStart=/bin/bash -c '{exec_command}'
+"""
+
+TIMER_TEMPLATE = """[Unit]
+Description=dar-backup {mode} timer
+
+[Timer]
+OnCalendar={calendar}
+Persistent=true
+
+[Install]
+WantedBy=timers.target
+"""
+
+CLEANUP_SERVICE_TEMPLATE = """[Unit]
+Description=cleanup up old DIFF & INCR backups
+StartLimitIntervalSec=120
+StartLimitBurst=1
+
+[Service]
+Type=oneshot
+TimeoutSec=60
+RemainAfterExit=no
+ExecStart=/bin/bash -c '{exec_command}'
+"""
+
+CLEANUP_TIMER = """[Unit]
+Description=dar-cleanup DIFF & INCR timer
+
+[Timer]
+OnCalendar=*-*-* 21:07:00
+
+[Install]
+WantedBy=timers.target
+"""
+
+TIMINGS = {
+    "FULL": "*-12-30 10:03:00",
+    "DIFF": "*-*-01 19:03:00",
+    "INCR": "*-*-04/3 19:03:00"
+}
+
+FLAGS = {
+    "FULL": "-F",
+    "DIFF": "-D",
+    "INCR": "-I"
+}
+
+def build_exec_command(venv, flag, dar_path=None, tool='dar-backup'):
+    if dar_path:
+        return f"PATH={dar_path}:$PATH && . {venv}/bin/activate && {tool} {flag} --verbose --log-stdout"
+    return f". {venv}/bin/activate && {tool} {flag} --verbose --log-stdout"
+
+def generate_service(mode, venv, dar_path):
+    exec_command = build_exec_command(venv, FLAGS[mode], dar_path)
+    return SERVICE_TEMPLATE.format(mode=mode, exec_command=exec_command)
+
+def generate_timer(mode):
+    return TIMER_TEMPLATE.format(mode=mode, calendar=TIMINGS[mode])
+
+def generate_cleanup_service(venv, dar_path):
+    exec_command = build_exec_command(venv, "", dar_path, tool='cleanup').strip()
+    return CLEANUP_SERVICE_TEMPLATE.format(exec_command=exec_command)
+
+def write_unit_file(path, filename, content):
+    file_path = path / filename
+    file_path.write_text(content)
+    print(f"Generated {filename}")
+
+def enable_and_start_unit(unit_name):
+    subprocess.run(["systemctl", "--user", "enable", unit_name], check=False)
+    subprocess.run(["systemctl", "--user", "start", unit_name], check=False)
+
+def write_unit_files(venv, dar_path, install=False):
+    output_path = Path.home() / ".config/systemd/user" if install else Path.cwd()
+    output_path.mkdir(parents=True, exist_ok=True)
+
+    for mode in FLAGS:
+        service_name = f"dar-{mode.lower()}-backup.service"
+        timer_name = f"dar-{mode.lower()}-backup.timer"
+        write_unit_file(output_path, service_name, generate_service(mode, venv, dar_path))
+        write_unit_file(output_path, timer_name, generate_timer(mode))
+        print(f"  → Fires on: {TIMINGS[mode]}")
+
+    write_unit_file(output_path, "dar-cleanup.service", generate_cleanup_service(venv, dar_path))
+    write_unit_file(output_path, "dar-cleanup.timer", CLEANUP_TIMER)
+    print(f"  → Fires on: *-*-* 21:07:00")
+
+    if install:
+        for mode in FLAGS:
+            enable_and_start_unit(f"dar-{mode.lower()}-backup.timer")
+        enable_and_start_unit("dar-cleanup.timer")
+        subprocess.run(["systemctl", "--user", "daemon-reexec"], check=False)
+        subprocess.run(["systemctl", "--user", "daemon-reload"], check=False)
+        print("Systemd `dar-backup` units and timers installed and user daemon reloaded.")
+
+def main():
+    parser = argparse.ArgumentParser(description="Generate systemd service and timer units for dar-backup.")
+    parser.add_argument("--venv",     required=True,       help="Path to the Python venv with dar-backup")
+    parser.add_argument("--dar-path",                      help="Optional path to dar binary's directory")
+    parser.add_argument("--install",  action="store_true", help="Install the units to ~/.config/systemd/user")
+    args = parser.parse_args()
+    write_unit_files(args.venv, args.dar_path, install=args.install)
+
+if __name__ == "__main__":
+    main()

dar_backup/installer.py

@@ -63,6 +63,7 @@ BACKUP_DEFINITION = '''
 --cache-directory-tagging
 '''
 
+
 def main():
     parser = argparse.ArgumentParser(
         description="Set up `dar-backup` on your system.",
@@ -72,7 +73,6 @@ def main():
         action="store_true",
         help="Deploy a simple config file, use ~/dar-backup/ for log file, archives and restore tests."
     )
-
     parser.add_argument(
         "-v", "--version",
         action="version",
@@ -85,38 +85,54 @@ def main():
         errors = []
         if os.path.exists(CONFIG_DIR):
             errors.append(f"Config directory '{CONFIG_DIR}' already exists.")
-
         if os.path.exists(DAR_BACKUP_DIR):
             errors.append(f"Directory '{DAR_BACKUP_DIR}' already exists.")
 
-        if len(errors) > 0:
+        if errors:
             for error in errors:
                 print(f"Error: {error}")
-            sys.exit(1) 
-
-        os.makedirs(DAR_BACKUP_DIR, exist_ok=False)
-        os.makedirs(os.path.join(DAR_BACKUP_DIR, "backups"), exist_ok=False)
-        os.makedirs(os.path.join(DAR_BACKUP_DIR, "restore"), exist_ok=False)
-        os.makedirs(CONFIG_DIR, exist_ok=False)
-        os.makedirs(os.path.join(CONFIG_DIR, "backup.d"), exist_ok=False)
-        print(f"Directories created: `{DAR_BACKUP_DIR}` and `{CONFIG_DIR}`")
-
-        script_dir = Path(__file__).parent
-        source_file = script_dir / "dar-backup.conf"
-        destination_file = Path(CONFIG_DIR) / "dar-backup.conf"
-        shutil.copy(source_file, destination_file)
-        print(f"Config file deployed to {destination_file}")
-
-
-        backup_definition = BACKUP_DEFINITION.replace("@@HOME_DIR@@", os.path.expanduser("~"))
-        with open(os.path.join(CONFIG_DIR, "backup.d", "default"), "w") as f:
-            f.write(backup_definition)
-        print(f"Default backup definition file deployed to {os.path.join(CONFIG_DIR, 'backup.d', 'default')}")
+            sys.exit(1)
+
+        try:
+            os.makedirs(DAR_BACKUP_DIR, exist_ok=False)
+            os.makedirs(os.path.join(DAR_BACKUP_DIR, "backups"), exist_ok=False)
+            os.makedirs(os.path.join(DAR_BACKUP_DIR, "restore"), exist_ok=False)
+            os.makedirs(CONFIG_DIR, exist_ok=False)
+            os.makedirs(os.path.join(CONFIG_DIR, "backup.d"), exist_ok=False)
+            print(f"Directories created: `{DAR_BACKUP_DIR}` and `{CONFIG_DIR}`")
+
+            script_dir = Path(__file__).parent
+            source_file = script_dir / "dar-backup.conf"
+            destination_file = Path(CONFIG_DIR) / "dar-backup.conf"
+
+            try:
+                shutil.copy2(source_file, destination_file)
+                print(f"Config file deployed to {destination_file}")
+            except Exception as e:
+                print(f"Error: Could not copy config file: {e}")
+                sys.exit(1)
+
+
+            backup_definition = BACKUP_DEFINITION.replace("@@HOME_DIR@@", os.path.expanduser("~"))
+
+            try:
+                with open(os.path.join(CONFIG_DIR, "backup.d", "default"), "w") as f:
+                    f.write(backup_definition)
+                print(f"Default backup definition file deployed to {os.path.join(CONFIG_DIR, 'backup.d', 'default')}")
+            except Exception as e:
+                print(f"Error: Could not write default backup definition: {e}")
+                sys.exit(1)
+        except Exception as e:
+            print(f"Installation failed: {e}")
+            sys.exit(1)
+
         print("1. Now run `manager --create` to create the catalog database.")
         print("2. Then you can run `dar-backup --full-backup` to create a backup.")
         print("3. List backups with `dar-backup --list`")
         print("4. List contents of a backup with `dar-backup --list-contents <backup-name>`")
 
+    sys.exit(0)
+
 
 if __name__ == "__main__":
     main()