Produits Entr'ouvert » Combo

# combo - content management system

# Copyright (C) 2021  Entr'ouvert

#

# This program is free software: you can redistribute it and/or modify it

# under the terms of the GNU Affero General Public License as published

# by the Free Software Foundation, either version 3 of the License, or

# (at your option) any later version.

#

# This program is distributed in the hope that it will be useful,

# but WITHOUT ANY WARRANTY; without even the implied warranty of

# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

# GNU Affero General Public License for more details.

#

# You should have received a copy of the GNU Affero General Public License

# along with this program.  If not, see <http://www.gnu.org/licenses/>.

from django.core.management.base import BaseCommand, CommandError

from combo.apps.assets.utils import export_assets

class Command(BaseCommand):

    help = 'Export assets'

    def add_arguments(self, parser):

        parser.add_argument(

                '--output', metavar='FILE', default='assets_export.tar',

                help='name of a file to write output to')

    def handle(self, *args, **options):

        try:

            output = open(options['output'], 'wb')

        except IOError as e:

            raise CommandError(e)

        export_assets(output)

        output.close()

# combo - content management system

# Copyright (C) 2016  Entr'ouvert

#

# This program is free software: you can redistribute it and/or modify it

# under the terms of the GNU Affero General Public License as published

# by the Free Software Foundation, either version 3 of the License, or

# (at your option) any later version.

#

# This program is distributed in the hope that it will be useful,

# but WITHOUT ANY WARRANTY; without even the implied warranty of

# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

# GNU Affero General Public License for more details.

#

# You should have received a copy of the GNU Affero General Public License

# along with this program.  If not, see <http://www.gnu.org/licenses/>.

import json

import sys

import tarfile

from django.core.management.base import BaseCommand, CommandError

from combo.apps.assets.utils import import_assets

class Command(BaseCommand):

    help = 'Import an exported site'

    def add_arguments(self, parser):

        parser.add_argument('filename', metavar='FILENAME', type=str,

                help='name of file to import')

        parser.add_argument(

                '--overwrite', action='store_true', default=False,

                help='Overwrite asset files')

    def handle(self, filename, *args, **options):

        fd = open(filename, 'rb')

        try:

            import_assets(fd, overwrite=options['overwrite'])

        except ImportSiteError as e:

            raise CommandError(e)

from django.core.management import call_command

    clean_assets_files()

def test_cmd_import_export_assets(tmpdir, some_assets):

    assert Asset.objects.count() == 2

    assert count_asset_files() == 2

    filename = os.path.join(str(tmpdir), 'file.tar')

    call_command('export_assets', '--output', filename)

    Asset.objects.all().delete()

    assert Asset.objects.count() == 0

    clean_assets_files()

    Asset(key='banner', asset=File(BytesIO(b'modified'), 'test.png')).save()

    call_command('import_assets', filename)  # default behaviour

    assert Asset.objects.count() == 2

    assert count_asset_files() == 2

    assert open('%s/assets/test.png' % default_storage.path('')).read() == 'modified'

    call_command('import_assets', filename, '--overwrite')

    assert Asset.objects.count() == 2

    assert count_asset_files() == 2

    assert open('%s/assets/test.png' % default_storage.path('')).read() == 'test'

    clean_assets_files()