Produits Entr'ouvert » Combo

# 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

class Command(BaseCommand):

    help = 'Export the site'

    def add_arguments(self, parser):

        parser.add_argument(

                '--output', metavar='FILE', default=None,

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

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

        else:

# 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

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

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(

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

                help='Clean site before importing')

        parser.add_argument(

                '--if-empty', action='store_true', default=False,

                help='Import only if site is empty')

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

        if filename == '-':

            fd = sys.stdin

        else:

        try:

            raise CommandError(e)

# 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.contrib.auth.models import Group

from django.db import transaction

from django.utils import six

from django.utils.encoding import python_2_unicode_compatible

from django.utils.translation import ugettext_lazy as _

from combo.apps.assets.models import Asset

from combo.apps.maps.models import MapLayer

from combo.apps.pwa.models import PwaSettings, PwaNavigationEntry

from .models import Page

@python_2_unicode_compatible

    def __init__(self, names):

        self.names = names

    def __str__(self):

        return _('Missing groups: %s') % ', '.join(self.names)

def export_site():

        MapLayer.load_serialized_objects(data.get('map-layers') or [])

        Asset.load_serialized_objects(data.get('assets') or [])

        Page.load_serialized_pages(data.get('pages') or [])

        if data.get('pwa'):

            PwaSettings.load_serialized_settings(data['pwa'].get('settings'))

            PwaNavigationEntry.load_serialized_objects(data['pwa'].get('navigation'))

# 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 hashlib

import json

from django.conf import settings

from django.contrib import messages

from django.core.exceptions import ObjectDoesNotExist

from django.core.urlresolvers import reverse, reverse_lazy

from django.http import HttpResponse, HttpResponseRedirect, Http404

from django.shortcuts import redirect

from django.shortcuts import get_object_or_404

from django.utils.translation import ugettext_lazy as _

from django.utils.encoding import force_text, force_bytes

from django.utils.formats import date_format

from django.utils.timezone import localtime

from django.views.decorators.csrf import requires_csrf_token

from django.views.generic import (RedirectView, DetailView,

        CreateView, UpdateView, ListView, DeleteView, FormView)

from combo.data.models import Page, CellBase, ParentContentCell, PageSnapshot, LinkListCell

from combo.data.library import get_cell_class

from combo import plugins

from .forms import (PageEditTitleForm, PageVisibilityForm, SiteImportForm,

        PageEditRedirectionForm, PageSelectTemplateForm, PageEditSlugForm,

        PageEditPictureForm, PageEditIncludeInNavigationForm,

        PageEditDescriptionForm, CellVisibilityForm)

homepage = HomepageView.as_view()

class SiteExportView(ListView):

    model = Page

    def render_to_response(self, context, **response_kwargs):

site_export = SiteExportView.as_view()

class SiteImportView(FormView):

    form_class = SiteImportForm

    template_name = 'combo/site_import.html'

    success_url = reverse_lazy('combo-manager-homepage')

    def form_valid(self, form):

        try:

        try:

        except MissingGroups as e:

            form.add_error('site_json', force_text(e))

            return self.form_invalid(form)

        return super(SiteImportView, self).form_valid(form)

site_import = SiteImportView.as_view()

import base64

import json

import os

import shutil

import sys

import tempfile

import pytest

from django.contrib.auth.models import Group

from django.core.files import File

from django.core.management import call_command

from django.core.management.base import CommandError

from django.utils.encoding import force_bytes, force_text

@pytest.fixture

def some_assets():

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

    Asset(key='favicon', asset=File(BytesIO(b'test2'), 'test2.png')).save()

def get_output_of_command(command, *args, **kwargs):

    old_stdout = sys.stdout

    output = sys.stdout = StringIO()

    sys.stdout = old_stdout

    return output.getvalue()

def test_import_export(app, some_data):

    output = get_output_of_command('export_site')

    assert len(json.loads(output)['pages']) == 3

    import_site(data={}, clean=True)

    assert Page.objects.all().count() == 0

    output = get_output_of_command('export_site')

    import_site(data={}, clean=True)

    assert Image.objects.all().count() == 0

    import_site(data=json.loads(output))

    assert Image.objects.all().count() == 2

    assert image1.title == 'foo'

    assert image1.image == 'path/foo.jpg'

    page4 = Page(title='Four', slug='four', parent=page1, template_name='standard')

    random_list = [page3, page4, page1, page2]

    ordered_list = Page.get_as_reordered_flat_hierarchy(random_list)

    assert ordered_list[0] == page1

    assert ordered_list[1] == page4

    assert ordered_list[2] == page2

    assert ordered_list[3] == page3

def test_site_export_import(app, admin_user):

    Page.objects.all().delete()

    page1 = Page(title='One', slug='one', template_name='standard')

    page1.save()

    page2 = Page(title='Two', slug='two', parent=page1, template_name='standard')

    page2.save()

    page3 = Page(title='Three', slug='three', parent=page2, template_name='standard')

    page3.save()

    cell.save()

    cell = LinkCell(page=page2, placeholder='content', link_page=page1, order=0)

    cell.save()

    app = login(app)

    resp = app.get('/manage/')

    resp = resp.click('Export Site')

    site_export = resp.body

    Page.objects.all().delete()

    assert LinkCell.objects.count() == 0

    app = login(app)

    resp = app.get('/manage/')

    resp = resp.click('Import Site')

    resp = resp.form.submit()

    assert Page.objects.count() == 4

    assert LinkCell.objects.count() == 2

    assert LinkCell.objects.get(page__slug='one').link_page.slug == 'two'

    assert LinkCell.objects.get(page__slug='two').link_page.slug == 'one'

    # check with invalid file

    resp = app.get('/manage/')

    resp = resp.click('Import Site')

    resp = resp.form.submit()

def test_site_export_import_missing_group(app, admin_user):

    Page.objects.all().delete()

    group = Group.objects.create(name='foobar')

    page1 = Page(title='One', slug='one', template_name='standard')

    page1.save()

    page1.groups.set([group])

    app = login(app)

    resp = app.get('/manage/')

    resp = resp.click('Export Site')

    site_export = resp.body

    Page.objects.all().delete()

    group.delete()

    app = login(app)

    resp = app.get('/manage/')

    resp = resp.click('Import Site')

    resp = resp.form.submit()

    assert 'Missing groups: foobar' in resp.text

def test_duplicate_page(app, admin_user):

    page = Page.objects.create(title='One', slug='one', template_name='standard', exclude_from_navigation=False)

    TextCell.objects.create(page=page, placeholder='content', text='Foobar', order=0)

    cell = TextCell(page=page2, text='bar', order=0, placeholder='content')

    cell.save()

    export_filename = os.path.join(settings.MEDIA_ROOT, 'site-export.json')

    if os.path.exists(export_filename):

        os.unlink(export_filename)

    cmd = ExportSiteCommand()

    assert os.path.exists(export_filename)

    stdout = sys.stdout

    try:

        sys.stdout = StringIO()

        assert sys.stdout.getvalue() == open(export_filename).read()

    finally:

        sys.stdout = stdout

    Page.objects.all().delete()

    cmd = ImportSiteCommand()

    cmd.handle(filename=export_filename, if_empty=False, clean=False)