import StringIO
import zipfile

from django.http import HttpResponse
from django.utils.translation import ugettext_lazy as _

from orchestra.utils.html import html_to_pdf


def render_bills(modeladmin, request, queryset):
    for bill in queryset:
        bill.html = bill.render()
        bill.save()
render_bills.verbose_name = _("Render")
render_bills.url_name = 'render'


def download_bills(modeladmin, request, queryset):
    if queryset.count() > 1:
        stringio = StringIO.StringIO()
        archive = zipfile.ZipFile(stringio, 'w')
        for bill in queryset:
            pdf = html_to_pdf(bill.html)
            archive.writestr('%s.pdf' % bill.number, pdf)
        archive.close()
        response = HttpResponse(stringio.getvalue(), content_type='application/pdf')
        response['Content-Disposition'] = 'attachment; filename="orchestra-bills.zip"'
        return response
    bill = queryset.get()
    pdf = html_to_pdf(bill.html)
    return HttpResponse(pdf, content_type='application/pdf')
download_bills.verbose_name = _("Download")
download_bills.url_name = 'download'


def view_bill(modeladmin, request, queryset):
    bill = queryset.get()
    bill.html = bill.render()
    return HttpResponse(bill.html)
view_bill.verbose_name = _("View")
view_bill.url_name = 'view'


def close_bills(modeladmin, request, queryset):
    for bill in queryset:
        bill.close()
close_bills.verbose_name = _("Close")
close_bills.url_name = 'close'