8e69178e07
Instead of inserting the path of the apps into the Python app, the apps are now referenced by their full import path. This app name claves with external or native Python libraries. Example: Mayan statistics app vs. Python new statistics library. Every app reference is now prepended with 'mayan.apps'. Existing config.yml files need to be updated manually. Signed-off-by: Roberto Rosario <roberto.rosario.gonzalez@gmail.com>
96 lines
3.1 KiB
Python
96 lines
3.1 KiB
Python
from __future__ import unicode_literals
|
|
|
|
import errno
|
|
import os
|
|
|
|
from pathlib2 import Path
|
|
|
|
from django.conf import settings
|
|
from django.core import management
|
|
from django.core.management.base import CommandError
|
|
from django.utils.encoding import force_text
|
|
from django.utils.translation import ugettext_lazy as _
|
|
|
|
from mayan.apps.common.utils import fs_cleanup
|
|
from mayan.apps.documents.models import DocumentType
|
|
|
|
CONVERTDB_FOLDER = 'convertdb'
|
|
CONVERTDB_OUTPUT_FILENAME = 'migrate.json'
|
|
|
|
|
|
class Command(management.BaseCommand):
|
|
help = 'Convert from a database backend to another one.'
|
|
|
|
def add_arguments(self, parser):
|
|
parser.add_argument(
|
|
'args', metavar='app_label[.ModelName]', nargs='*',
|
|
help=_(
|
|
'Restricts dumped data to the specified app_label or '
|
|
'app_label.ModelName.'
|
|
)
|
|
)
|
|
parser.add_argument(
|
|
'--from', action='store', default='default', dest='from',
|
|
help=_(
|
|
'The database from which data will be exported. If omitted '
|
|
'the database named "default" will be used.'
|
|
),
|
|
)
|
|
parser.add_argument(
|
|
'--to', action='store', default='default', dest='to',
|
|
help=_(
|
|
'The database to which data will be imported. If omitted '
|
|
'the database named "default" will be used.'
|
|
),
|
|
)
|
|
parser.add_argument(
|
|
'--force', action='store_true', dest='force',
|
|
help=_(
|
|
'Force the conversion of the database even if the receiving '
|
|
'database is not empty.'
|
|
),
|
|
)
|
|
|
|
def handle(self, *app_labels, **options):
|
|
# Create the media/convertdb folder
|
|
convertdb_folder_path = force_text(
|
|
Path(
|
|
settings.MEDIA_ROOT, CONVERTDB_FOLDER
|
|
)
|
|
)
|
|
|
|
try:
|
|
os.makedirs(convertdb_folder_path)
|
|
except OSError as exception:
|
|
if exception.errno == errno.EEXIST:
|
|
pass
|
|
|
|
convertdb_file_path = force_text(
|
|
Path(
|
|
convertdb_folder_path, CONVERTDB_OUTPUT_FILENAME
|
|
)
|
|
)
|
|
|
|
management.call_command('purgeperiodictasks')
|
|
|
|
management.call_command(
|
|
'dumpdata', *app_labels, all=True,
|
|
database=options['from'], natural_primary=True,
|
|
natural_foreign=True, output=convertdb_file_path,
|
|
interactive=False, format='json'
|
|
)
|
|
|
|
if DocumentType.objects.using(options['to']).count() and not options['force']:
|
|
fs_cleanup(convertdb_file_path)
|
|
raise CommandError(
|
|
'There is existing data in the database that will be '
|
|
'used for the import. If you proceed with the conversion '
|
|
'you might lose data. Please check your settings.'
|
|
)
|
|
|
|
management.call_command(
|
|
'loaddata', convertdb_file_path, database=options['to'],
|
|
interactive=False, verbosity=3
|
|
)
|
|
fs_cleanup(convertdb_file_path)
|