1

import datetime

     2

     4

from django.contrib import admin

     6

import django.utils.html

     7

import django.forms.models

     8

     9

from qrurls.models import URL, URLItem, URLImage

    10

    11

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

    17

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

    18

        urlfeed = kwargs.get('instance')

    19

    20

        publishing_date = None

    21

        publishing_time = datetime.time()

    22

    24

        if urlfeed and isinstance(urlfeed, URL) :

    25

            publishing_date = urlfeed.last_item().published.date()

    27

    28

        def publishing_schedule (count) :

    29

            if not publishing_date :

    30

                # starting today

    31

                start_date = datetime.date.today()

    32

                offset = 0

    33

            else :

    34

                # starting from the following day

    35

                start_date = publishing_date

    36

                offset = 1

    37

    38

            for days in xrange(offset, count + offset) :

    39

                yield datetime.datetime.combine(

    40

                        start_date + datetime.timedelta(days=days),

    41

                        publishing_time

    42

                )

    43

    44

        kwargs.update(initial=[

    45

                dict(published=publish) for publish in publishing_schedule(count=5)

    46

        ])

    47

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

    48

    49

class URLItemInline (admin.TabularInline) :

    54

    model = URLItem

    55

    formset = URLItemFormset

    56

    57

class URLAdmin (admin.ModelAdmin) :

    72

    def qrcode_url (self, obj) :

    73

        warn = None

    74

    75

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

    76

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

    77

    78

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

    79

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

    80

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

    81

        )

    82

    qrcode_url.allow_tags = True

    83

    84

    def qrcode_img (self, obj) :

    85

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

    86

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

    87

        )

    88

    qrcode_img.allow_tags = True

   107

   108

    readonly_fields = (

   110

        'qrcode_url',

   111

        'qrcode_img',

   112

        'active_item',

   113

        'upcoming_item',

   117

    )

   118

    list_display = (

   119

        'get_absolute_url',

   120

        'active_item',

   121

        'upcoming_item',

   122

    )

   123

    fieldsets = (

   124

        (None, {

   130

        }),

   131

        ("QRCode", {

   132

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

   133

        }),

   141

    )

   142

    inlines = (URLItemInline, )

   143

   144

class URLItemAdmin (admin.ModelAdmin) :

   145

    list_display = (

   147

    )

   149

    fieldsets = (

   150

        ("Publishing", {

   151

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

   152

        }),

   153

        ("Target", {

   154

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

   155

        }),

   156

    )

   157

   158

class URLImageAdmin (admin.ModelAdmin) :

   159

    list_display = (

   160

        'title',

   161

        'uploaded',

   162

    )

   163

    # hide the "uploaded" field

   164

    fields = (

   165

        'image',

   166

        'title',

   167

    )

   168

    readonly_fields = (

   169

        'uploaded',

   170

    )

   171

   172

    inlines = (URLItemInline, )

   173

   174

admin.site.register(URL, URLAdmin)

   175

admin.site.register(URLItem, URLItemAdmin)

   176

admin.site.register(URLImage, URLImageAdmin)