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', 'image', 'published_state', 
    )
    readonly_fields = ('published_state',)
    fieldsets = (
        ("Publishing", {
            'fields': ('shorturl', 'published', ),
        }),
        ("Target", {
            'fields': ('url', 'image'),
        }),
    )
    

class URLImageAdmin (admin.ModelAdmin) :
    list_display = (
        'uploaded', 'image',
    )
    # hide the "uploaded" field
    fields = (
        'image',
    )
    readonly_fields = (
        'uploaded',
    )
    
    inlines = (URLItemInline, )

admin.site.register(URL, URLAdmin)
admin.site.register(URLItem, URLItemAdmin)
admin.site.register(URLImage, URLImageAdmin)