1

import datetime

     2

import urllib

     3

     4

from django import http, shortcuts, forms

     5

from django.conf import settings, urls

     6

from django.contrib import admin

     7

from django.core.urlresolvers import reverse

     8

from django.utils import timezone, formats

     9

import django.utils.html

    10

import django.forms.models

    11

    12

from qrurls.models import URL, URLItem, URLImage

    13

    14

"""

    15

    Private backend UI, using django-admin.

    16

"""

    17

    18

class URLItemFormset (django.forms.models.BaseInlineFormSet) :

    19

    """

    20

        Uses the existing URLItems for the URLFeed to determine the initial publishing

    21

        times for new items.

    22

    """

    23

    24

    def __init__ (self, *args, **kwargs) :

    25

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

    26

    32

        kwargs.update(initial=[

    35

        ])

    36

        super(URLItemFormset, self).__init__(*args, **kwargs)

    37

    38

class URLItemInline (admin.TabularInline) :

    39

    """

    40

        Inline set of URLItems for an URLFeed.

    41

    """

    42

    43

    model = URLItem

    44

    formset = URLItemFormset

    45

    46

class URLAdmin (admin.ModelAdmin) :

    47

    @classmethod

    52

        else:

    61

    62

    def timezone (self, obj) :

    63

        now = timezone.localtime(obj.now())

    64

        tz = now.tzinfo

    65

        td = now.tzinfo.utcoffset(now)

    66

    67

        if td :

    68

            minutes, seconds = divmod(td.total_seconds(), 60)

    69

            hours, minutes = divmod(minutes, 60)

    70

            offset = "(UTC%+03d:%02d)" % (hours, minutes)

    71

        else :

    72

            offset = ""

    73

    74

        return u"%s %s" % (tz, offset)

    75

    76

    def qrcode_url (self, obj) :

    77

        warn = None

    78

    79

        if obj.shorturl.upper() != obj.shorturl :

    80

            warn = "Shorturl should be UPPERCASE for most compact QR code"

    81

    82

        return '<a href="{url}">{url}</a>{warn}'.format(

    83

                url     = django.utils.html.escape(obj.qrcode_url()),

    84

                warn    = '<p class="errornote">{warn}</p>'.format(warn=warn) if warn else '',

    85

        )

    86

    qrcode_url.allow_tags = True

    87

    88

    def qrcode_img (self, obj) :

    89

        return '<img src="{img}" />'.format(

    90

                img=django.utils.html.escape(obj.qrcode_img()),

    91

        )

    92

    qrcode_img.allow_tags = True

    93

    94

    # XXX: a whole bunch of ugly datetime-formatting for display...

    95

    def active (self, obj) :

    96

        item = obj.active_item()

    97

        if item :

    98

            return "%s %s" % (formats.localize(timezone.localtime(item.published)), item)

    99

        else :

   100

            return ""

   101

   102

    def now (self, obj) :

   103

        return formats.localize(timezone.localtime(obj.now()))

   104

   105

    def upcoming (self, obj) :

   106

        item = obj.upcoming_item()

   107

        if item :

   108

            return "%s %s" % (formats.localize(timezone.localtime(item.published)), item)

   109

        else :

   110

            return ""

   111

   112

    readonly_fields = (

   113

        'timezone',

   114

        'qrcode_url',

   115

        'qrcode_img',

   116

        'active_item',

   117

        'upcoming_item',

   118

        'active',

   119

        'now',

   120

        'upcoming',

   121

    )

   122

    list_display = (

   123

        'get_absolute_url',

   124

        'active_item',

   125

        'upcoming_item',

   126

    )

   127

    fieldsets = (

   128

        (None, {

   129

            'fields': (

   130

                'shorturl', 

   131

                'publishing_time', 

   132

                'timezone',

   133

                'title',

   134

            ),

   135

        }),

   136

        ("QRCode", {

   137

            'fields': ('qrcode_url', 'qrcode_img'),

   138

        }),

   139

        ("Item publishing overview", {

   140

            'fields': (

   141

                'active',

   142

                'now',

   143

                'upcoming',

   144

            ),

   145

        }),

   146

    )

   147

    inlines = (URLItemInline, )

   148

   149

    ## Import

   150

    class ImportForm (forms.Form) :

   151

        shorturl = forms.ModelChoiceField(URL.objects)

   152

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

   153

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

   154

        # Django forms doesn't really understand multi-value fields, so we just hack this ourselves.

   155

        #image = forms.ImageField()

   156

   157

    def import_images_handler (self, url_feed, data, images=()) :

   158

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

   159

   163

   164

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

   165

            url_image = URLImage(image=image)

   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)

   179

            images = request.FILES.getlist('image')

   180

   181

            if form.is_valid() :

   182

                return self.import_images_handler(url_feed, form.cleaned_data, images)

   183

        else :

   185

   186

            form = self.ImportForm(initial=dict(

   187

                shorturl        = url_feed.id,

   188

                publishing_date = publishing_date,

   189

                publishing_time = publishing_time,

   190

            ))

   191

   192

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

   193

            current_app     = self.admin_site.name,

   194

   195

            media           = self.media + form.media,

   196

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

   197

            form            = form,

   198

        ))

   199

   200

    def get_urls (self) :

   201

        return urls.patterns('',

   202

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

   203

                name='qrurls_import_images'),

   204

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

   205

   206

    def import_images_action (self, request, queryset) :

   207

        url_feed, = [url_feed for url_feed in queryset]

   208

        return http.HttpResponseRedirect(

   209

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

   210

        )

   211

    import_images_action.short_description = "Import images for selected feeds"

   212

   213

    actions = [import_images_action]

   214

   215

class URLItemAdmin (admin.ModelAdmin) :

   216

    list_display = (

   217

        'shorturl', 'published_state', 'get_absolute_url', 'image', 'published',

   218

    )

   219

    readonly_fields = ('published_state', 'published')

   220

    fieldsets = (

   221

        ("Publishing", {

   222

            'fields': ('shorturl', 'published', ),

   223

        }),

   224

        ("Target", {

   225

            'fields': ('url', 'image'),

   226

        }),

   227

    )

   228

   229

class URLImageAdmin (admin.ModelAdmin) :

   230

    list_display = (

   231

        'title',

   232

        'uploaded',

   233

    )

   234

    # hide the "uploaded" field

   235

    fields = (

   236

        'image',

   237

        'title',

   238

    )

   239

    readonly_fields = (

   240

        'uploaded',

   241

    )

   242

   243

    inlines = (URLItemInline, )

   244

   245

admin.site.register(URL, URLAdmin)

   246

admin.site.register(URLItem, URLItemAdmin)

   247

admin.site.register(URLImage, URLImageAdmin)