import shutil

from django.core.management.base import BaseCommand, CommandError
from django.db import connection, transaction

from openslides.utils.main import get_database_path_from_settings


class Command(BaseCommand):
    """
    Command to backup the SQLite3 database.
    """

    help = "Backups the SQLite3 database."

    def add_arguments(self, parser):
        parser.add_argument(
            "--path",
            default="database_backup.sqlite",
            help="Path for the backup file (Default: database_backup.sqlite).",
        )

    def handle(self, *args, **options):
        path = options.get("path")

        @transaction.atomic
        def do_backup(src_path, dest_path):
            # perform a simple file-copy backup of the database
            # first we need a shared lock on the database, issuing a select()
            # will do this for us
            cursor = connection.cursor()
            cursor.execute("SELECT count(*) from sqlite_master")
            # now copy the file
            try:
                shutil.copy(src_path, dest_path)
            except IOError:
                # TODO: use the IOError message as message for the user
                raise CommandError("Database backup failed.")

        database_path = get_database_path_from_settings()
        if database_path:
            do_backup(database_path, path)
            self.stdout.write(
                "Database %s successfully stored at %s." % (database_path, path)
            )
        else:
            raise CommandError(
                "Default database is not SQLite3. Only SQLite3 databases"
                "can currently be backuped."
            )