Produits Entr'ouvert » Combo

class AssetsImportForm(forms.Form):

    assets_file = forms.FileField(label=_('Assets File'))

    overwrite = forms.BooleanField(label=_('Overwrite Existing Files'), required=False)

<a class="extra-actions-menu-opener"></a>

<ul class="extra-actions-menu">

  <li><a href="{% url 'combo-manager-assets-export' %}">{% trans 'Export Assets' %}</a></li>

  <li><a rel="popup" href="{% url 'combo-manager-assets-import' %}">{% trans 'Import Assets' %}</a></li>

</ul>

{% extends "combo/manager_base.html" %}

{% load i18n %}

{% block appbar %}

<h2>{% trans "Assets Import" %}</h2>

{% endblock %}

{% block content %}

<form method="post" enctype="multipart/form-data">

  {% csrf_token %}

  {{ form.as_p }}

  <div class="buttons">

    <button class="submit-button">{% trans "Import" %}</button>

    <a class="cancel" href="{% url 'combo-manager-assets' %}">{% trans 'Cancel' %}</a>

  </div>

</form>

{% endblock %}

    url(r'^export/$', views.assets_export, name='combo-manager-assets-export'),

    url(r'^import/$', views.assets_import, name='combo-manager-assets-import'),

import tarfile

from django.contrib import messages

from django.http import Http404, HttpResponse

from django.utils.six import BytesIO

from django.utils.translation import ugettext_lazy as _

from .forms import AssetUploadForm, AssetsImportForm

class AssetsImport(FormView):

    form_class = AssetsImportForm

    template_name = 'combo/manager_assets_import.html'

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

    def form_valid(self, form):

        overwrite = form.cleaned_data.get('overwrite')

        try:

            assets = tarfile.open(fileobj=form.cleaned_data['assets_file'])

        except tarfile.TarError:

            messages.error(self.request, _('The assets file is not valid.'))

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

        media_prefix = default_storage.path('')

        for tarinfo in assets.getmembers():

            filepath = default_storage.path(tarinfo.name)

            if not overwrite and os.path.exists(filepath):

                continue

            assets.extract(tarinfo, path=media_prefix)

        messages.success(self.request, _('The assets file has been imported.'))

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

assets_import = AssetsImport.as_view()

def assets_export(request, *args, **kwargs):

    fd = BytesIO()

    assets_file = tarfile.open('assets.tar', 'w', fileobj=fd)

    media_prefix = default_storage.path('')

    for basedir, dirnames, filenames in os.walk(media_prefix):

        for filename in filenames:

            assets_file.add(

                    os.path.join(basedir, filename),

                    os.path.join(basedir, filename)[len(media_prefix):])

    assets_file.close()

    return HttpResponse(fd.getvalue(), content_type='application/x-tar')

import shutil

def test_asset_export_import(app, admin_user):

    for path in ('uploads', 'assets', 'cache'):

        if os.path.exists(default_storage.path(path)):

            shutil.rmtree(default_storage.path(path))

    app = login(app)

    # upload a file

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

    resp = resp.click('Upload')

    resp.form['upload'] = Upload('test.png',

            base64.decodestring(b'iVBORw0KGgoAAAANSUhEUgAAAAEAAAABAQAAAAA3bvkkAAAACklEQVQI12NoAAAAggCB3UNq9AAAAABJRU5ErkJggg=='),

            'image/png')

    resp = resp.form.submit()

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

    resp = resp.click('Export')

    assert resp.content_type == 'application/x-tar'

    content = resp.content

    for path in ('uploads', 'assets'):

        if os.path.exists(default_storage.path(path)):

            shutil.rmtree(default_storage.path(path))

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

    assert 'have any asset yet.' in resp.text

    resp = resp.click('Import')

    resp.form['assets_file'] = Upload('test.tar', content)

    resp = resp.form.submit()

    assert sum([len(x[2]) for x in os.walk(default_storage.path(''))]) == 2

    resp = resp.follow()

    assert 'The assets file has been imported.' in resp.text

    # test no overwrite

    filename = re.findall('data-href="(.*?)"', resp.text)[0][7:]  # strip /media/

    with open(default_storage.path(filename), 'w') as fd:

        fd.write('test')  # 4 bytes

    assert os.stat(default_storage.path(filename)).st_size == 4

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

    resp = resp.click('Import')

    resp.form['assets_file'] = Upload('test.tar', content)

    resp = resp.form.submit()

    resp = resp.follow()

    assert 'The assets file has been imported.' in resp.text

    assert os.stat(default_storage.path(filename)).st_size == 4

    # test overwrite

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

    resp = resp.click('Import')

    resp.form['overwrite'] = True

    resp.form['assets_file'] = Upload('test.tar', content)

    resp = resp.form.submit()

    resp = resp.follow()

    assert 'The assets file has been imported.' in resp.text

    assert os.stat(default_storage.path(filename)).st_size == 67

    # test uploading garbage

    for path in ('uploads', 'assets'):

        if os.path.exists(default_storage.path(path)):

            shutil.rmtree(default_storage.path(path))

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

    resp = resp.click('Import')

    resp.form['assets_file'] = Upload('test.tar', b'garbage')

    resp = resp.form.submit()

    assert sum([len(x[2]) for x in os.walk(default_storage.path(''))]) == 0

    resp = resp.follow()

    assert 'The assets file is not valid.' in resp.text