2

import urllib

     4

from django import http, shortcuts, forms

     5

from django.conf import settings, urls

     7

from django.core.urlresolvers import reverse

    22

        extra = kwargs.get('extra', 5)

    24

        publishing_date, publishing_time = URLAdmin.publish_scheduling(urlfeed)

    25

    27

                dict(published=publish) for publish in URLAdmin.publishing_schedule(

    28

                    publishing_date, publishing_time, count=extra)

    41

    @classmethod

    42

    def publish_scheduling (cls, urlfeed) :

    43

        """Calculate initial URLItem.published values for feed, or defaults."""

    44

        # hack to get at the URLFeed to determine our initial values..

    45

        if urlfeed and isinstance(urlfeed, URL) :

    46

            # starting from the following day

    47

            return (

    48

                    urlfeed.last_item().published.date() + datetime.timedelta(days=1),

    49

                    urlfeed.publishing_time,

    50

            )

    51

        else:

    52

            # no data...

    53

            datetime.date.today(), datetime.time()

    54

    55

    @classmethod

    56

    def publishing_schedule (cls, publishing_date, publishing_time, count) :

    57

        """Yield URLItem.published values."""

    58

        for days in xrange(0, count) :

    59

            yield datetime.datetime.combine(

    60

                    publishing_date + datetime.timedelta(days=days),

    61

                    publishing_time

    62

            )

    63

   150

    ## Import

   151

    class ImportForm (forms.Form) :

   152

        shorturl = forms.ModelChoiceField(URL.objects)

   153

        publishing_date = forms.DateField(widget=admin.widgets.AdminDateWidget)

   154

        publishing_time = forms.TimeField(widget=admin.widgets.AdminTimeWidget)

   155

        image = forms.ImageField()

   156

   157

    def import_images_handler (self, url_feed, data) :

   158

        """Custom for backend for mass-importing images into a feed."""

   159

   160

        images = [data['image']]

   161

        publishing = self.publishing_schedule(data['publishing_date'], data['publishing_time'], len(images))

   162

   163

        for image, publish in zip(images, publishing) :

   164

            # XXX: move title default elsewhere

   165

            url_image = URLImage(image=image, title=image.name)

   166

            url_image.save()

   167

   168

            url_item = URLItem(shorturl=url_feed, published=publish, image=url_image)

   169

            url_item.save()

   170

   171

        return http.HttpResponseRedirect(reverse('admin:qrurls_url_change', args=(url_feed.id, )))

   172

   173

    def import_images (self, request, shorturl_id) :

   174

        """Custom form frontend for mass-importing images into a feed."""

   175

        url_feed = URL.objects.get(id=int(shorturl_id))

   176

   177

        if request.method == 'POST' :

   178

            form = self.ImportForm(request.POST, request.FILES)

   179

   180

            if form.is_valid() :

   181

                return self.import_images_handler(url_feed, form.cleaned_data)

   182

        else :

   183

            publishing_date, publishing_time = self.publish_scheduling(url_feed)

   184

   185

            form = self.ImportForm(initial=dict(

   186

                shorturl        = url_feed.id,

   187

                publishing_date = publishing_date,

   188

                publishing_time = publishing_time,

   189

            ))

   190

   191

        return shortcuts.render(request, 'admin/qrurls_import_images.html', dict(

   192

            current_app     = self.admin_site.name,

   193

   194

            media           = self.media + form.media,

   195

            form_url        = reverse('admin:qrurls_import_images', kwargs=dict(shorturl_id=url_feed.id)),

   196

            form            = form,

   197

        ))

   198

   199

    def get_urls (self) :

   200

        return urls.patterns('',

   201

            urls.url(r'^(?P<shorturl_id>\d+)/import$', self.admin_site.admin_view(self.import_images),

   202

                name='qrurls_import_images'),

   203

        ) + super(URLAdmin, self).get_urls()

   204

   205

    def import_images_action (self, request, queryset) :

   206

        url_feed, = [url_feed for url_feed in queryset]

   207

        return http.HttpResponseRedirect(

   208

                reverse('admin:qrurls_import_images', kwargs=dict(shorturl_id=url_feed.id)) # ) + '?' + urllib.urlencode(qs, doseq=True)

   209

        )

   210

    import_images_action.short_description = "Import images for selected feeds"

   211

   212

    actions = [import_images_action]

   213