47 lines
1.6 KiB
Python
47 lines
1.6 KiB
Python
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.')
|