import datetime
from django.contrib import admin
import django.utils.html
import django.forms.models
from qrurls.models import URL, URLItem, URLImage
class URLItemFormset (django.forms.models.BaseInlineFormSet) :
def __init__ (self, *args, **kwargs) :
urlfeed = kwargs.get('instance')
publishing_date = None
publishing_time = datetime.time()
if urlfeed and isinstance(urlfeed, URL) :
publishing_date = urlfeed.last_item().published.date()
publishing_time = urlfeed.publishing_schedule
def publishing_schedule (count) :
if not publishing_date :
# starting today
start_date = datetime.date.today()
offset = 0
else :
# starting from the following day
start_date = publishing_date
offset = 1
for days in xrange(offset, count + offset) :
yield datetime.datetime.combine(
start_date + datetime.timedelta(days=days),
publishing_time
)
kwargs.update(initial=[
dict(published=publish) for publish in publishing_schedule(count=5)
])
super(URLItemFormset, self).__init__(*args, **kwargs)
class URLItemInline (admin.TabularInline) :
model = URLItem
formset = URLItemFormset
class URLAdmin (admin.ModelAdmin) :
def qrcode_url (self, obj) :
warn = None
if obj.shorturl.upper() != obj.shorturl :
warn = "Shorturl should be UPPERCASE for most compact QR code"
return '<a href="{url}">{url}</a>{warn}'.format(
url = django.utils.html.escape(obj.qrcode_url()),
warn = '<p class="errornote">{warn}</p>'.format(warn=warn) if warn else '',
)
qrcode_url.allow_tags = True
def qrcode_img (self, obj) :
return '<img src="{img}" />'.format(
img=django.utils.html.escape(obj.qrcode_img()),
)
qrcode_img.allow_tags = True
readonly_fields = (
'qrcode_url',
'qrcode_img',
'active_item',
'upcoming_item',
)
list_display = (
'get_absolute_url',
'active_item',
'upcoming_item',
)
fieldsets = (
(None, {
'fields': ('shorturl', 'active_item', 'upcoming_item', 'publishing_schedule')
}),
("QRCode", {
'fields': ('qrcode_url', 'qrcode_img'),
}),
)
inlines = (URLItemInline, )
class URLItemAdmin (admin.ModelAdmin) :
list_display = (
'shorturl', 'get_absolute_url', 'published_state',
)
readonly_fields = ('published_state',)
fieldsets = (
("Publishing", {
'fields': ('shorturl', 'published', ),
}),
("Target", {
'fields': ('url', 'image'),
}),
)
class URLImageAdmin (admin.ModelAdmin) :
# hide the "uploaded" field
#fields = ( )
readonly_fields = (
'uploaded',
'image',
)
inlines = (URLItemInline, )
admin.site.register(URL, URLAdmin)
admin.site.register(URLItem, URLItemAdmin)
admin.site.register(URLImage, URLImageAdmin)