restic-stop-backups.py

#!/usr/bin/env python
"""
restic-stop-backups.py - Unregister Restic backup scheduled tasks.

Removes the Windows Task Scheduler tasks created by restic-start-backups.py
and restic-health-check.py. Use this when you want to temporarily or
permanently stop automatic backups.

Features:
- Unregisters backup task (ResticHelperBackup)
- Unregisters health check task (ResticHelperHealthCheck)
- Safe to run multiple times
- Requires Administrator privileges on Windows

Usage:
    python restic-stop-backups.py              # Unregister all tasks
    python restic-stop-backups.py --backup     # Only unregister backup task
    python restic-stop-backups.py --health     # Only unregister health check task
    python restic-stop-backups.py --dry-run    # Show what would be done

Requires Administrator privileges on Windows to modify Task Scheduler.
"""
import os
import sys
import logging
import argparse

import restic_utils as utils

# Task names (must match restic-start-backups.py and restic-health-check.py)
BACKUP_TASK_NAME = "ResticHelperBackup"
HEALTH_TASK_NAME = "ResticHelperHealthCheck"


def main():
    parser = argparse.ArgumentParser(description="Unregister Restic backup scheduled tasks")
    parser.add_argument("--backup", action="store_true",
                        help="Only unregister backup task")
    parser.add_argument("--health", action="store_true",
                        help="Only unregister health check task")
    parser.add_argument("--dry-run", action="store_true",
                        help="Show what would be done without making changes")
    parser.add_argument("--log-level", default="INFO",
                        choices=["DEBUG", "INFO", "WARNING", "ERROR"],
                        help="Set logging level (default: INFO)")
    args = parser.parse_args()

    # Configure logging
    logging.basicConfig(
        level=getattr(logging, args.log_level),
        format='%(message)s',
        handlers=[logging.StreamHandler(sys.stdout)]
    )

    # Check platform
    if sys.platform != "win32":
        print("Task Scheduler is only supported on Windows.")
        print("On other platforms, remove entries from cron or systemd manually.")
        sys.exit(1)

    # Check admin privileges
    if not utils.is_admin():
        print("ERROR: Administrator privileges required to unregister scheduled tasks.")
        print("Please run this script as Administrator.")
        sys.exit(1)

    # Determine which tasks to unregister
    unregister_backup = args.backup or not (args.backup or args.health)
    unregister_health = args.health or not (args.backup or args.health)

    print("Restic Helpers - Unregister Scheduled Tasks")
    print("=" * 50)

    success = True

    if unregister_backup:
        if args.dry_run:
            is_registered = utils.is_task_registered(BACKUP_TASK_NAME)
            if is_registered:
                print(f"[DRY-RUN] Would unregister task: {BACKUP_TASK_NAME}")
            else:
                print(f"[DRY-RUN] Task not registered: {BACKUP_TASK_NAME}")
        else:
            if not utils.unregister_scheduled_task(BACKUP_TASK_NAME):
                success = False

    if unregister_health:
        if args.dry_run:
            is_registered = utils.is_task_registered(HEALTH_TASK_NAME)
            if is_registered:
                print(f"[DRY-RUN] Would unregister task: {HEALTH_TASK_NAME}")
            else:
                print(f"[DRY-RUN] Task not registered: {HEALTH_TASK_NAME}")
        else:
            if not utils.unregister_scheduled_task(HEALTH_TASK_NAME):
                success = False

    print("=" * 50)
    if success:
        print("Done. Automatic backups are now disabled.")
        print("\nTo re-enable, run: python restic-start-backups.py --register")
    else:
        print("Some tasks could not be unregistered.")

    sys.exit(0 if success else 1)


if __name__ == "__main__":
    main()