uncloud/uncloud_pay/admin.py

93 lines
2.9 KiB
Python
Raw Normal View History

from django.contrib import admin
from django.template.response import TemplateResponse
from django.urls import path
2020-08-09 08:14:49 +00:00
from django.shortcuts import render
from django.conf.urls import url
from uncloud_pay.views import BillViewSet
from hardcopy import bytestring_to_pdf
from django.core.files.temp import NamedTemporaryFile
from django.http import FileResponse
from django.template.loader import render_to_string
2020-10-08 17:54:04 +00:00
from uncloud_pay.models import *
2020-06-21 11:46:54 +00:00
2020-08-04 16:56:36 +00:00
2020-06-21 14:08:00 +00:00
class BillRecordInline(admin.TabularInline):
model = BillRecord
2020-06-21 11:46:54 +00:00
class RecurringPeriodInline(admin.TabularInline):
model = ProductToRecurringPeriod
class ProductAdmin(admin.ModelAdmin):
inlines = [ RecurringPeriodInline ]
2020-08-04 16:56:36 +00:00
2020-06-21 14:08:00 +00:00
class BillAdmin(admin.ModelAdmin):
inlines = [ BillRecordInline ]
2020-06-21 14:42:55 +00:00
def get_urls(self):
"""
Create URLs for PDF view
"""
info = "%s_%s" % (self.model._meta.app_label, self.model._meta.model_name)
pat = lambda regex, fn: url(regex, self.admin_site.admin_view(fn), name='%s_%s' % (info, fn.__name__))
url_patterns = [
2020-08-09 08:14:49 +00:00
pat(r'^([0-9]+)/as_pdf/$', self.as_pdf),
pat(r'^([0-9]+)/as_html/$', self.as_html),
] + super().get_urls()
return url_patterns
2020-08-09 08:14:49 +00:00
def as_pdf(self, request, object_id):
bill = self.get_object(request, object_id=object_id)
print(bill)
if bill is None:
raise self._get_404_exception(object_id)
output_file = NamedTemporaryFile()
2020-06-21 21:54:57 +00:00
bill_html = render_to_string("bill.html.j2", {'bill': bill,
'bill_records': bill.billrecord_set.all()
2020-06-21 21:54:57 +00:00
})
bytestring_to_pdf(bill_html.encode('utf-8'), output_file)
response = FileResponse(output_file, content_type="application/pdf")
response['Content-Disposition'] = f'filename="bill_{bill}.pdf"'
return response
2020-08-09 08:14:49 +00:00
def as_html(self, request, object_id):
bill = self.get_object(request, object_id=object_id)
if bill is None:
raise self._get_404_exception(object_id)
return render(request, 'bill.html.j2',
{'bill': bill,
'bill_records': bill.billrecord_set.all()
})
bill_html = render_to_string("bill.html.j2", {'bill': bill,
'bill_records': bill.billrecord_set.all()
})
bytestring_to_pdf(bill_html.encode('utf-8'), output_file)
response = FileResponse(output_file, content_type="application/pdf")
response['Content-Disposition'] = f'filename="bill_{bill}.pdf"'
2020-08-09 08:14:49 +00:00
return HttpResponse(template.render(context, request))
return response
2020-06-21 11:46:54 +00:00
admin.site.register(Bill, BillAdmin)
admin.site.register(ProductToRecurringPeriod)
admin.site.register(Product, ProductAdmin)
2020-10-08 17:54:04 +00:00
for m in [ Order, BillRecord, BillingAddress, RecurringPeriod, VATRate ]:
admin.site.register(m)