TISbackup/tasks.py

import logging
import os
import subprocess

from huey import RedisHuey

from tisbackup import tis_backup

huey = RedisHuey("tisbackup", host="localhost")


@huey.task()
def run_export_backup(base, config_file, mount_point, backup_sections):
    try:
        # Log
        logger = logging.getLogger("tisbackup")
        logger.setLevel(logging.INFO)
        formatter = logging.Formatter("%(asctime)s %(levelname)s %(message)s")
        handler = logging.StreamHandler()
        handler.setFormatter(formatter)
        logger.addHandler(handler)

        # Main
        logger.info("Running export....")

        if backup_sections:
            backup_sections = backup_sections.split(",")
        else:
            backup_sections = []
        backup = tis_backup(dry_run=False, verbose=True, backup_base_dir=base)
        backup.read_ini_file(config_file)
        mount_point = mount_point
        backup.export_backups(backup_sections, mount_point)
    except Exception as e:
        return str(e)

    finally:
        # Safely unmount using subprocess instead of os.system
        try:
            subprocess.run(["/bin/umount", mount_point], check=True, timeout=30)
            os.rmdir(mount_point)
        except (subprocess.CalledProcessError, subprocess.TimeoutExpired, OSError) as e:
            logger.error(f"Failed to unmount {mount_point}: {e}")
    return "ok"


def get_task():
    return task


def set_task(my_task):
    global task
    task = my_task


task = None