Invoice info, Service Provider info, Client info
This commit is contained in:
parent
e40c392761
commit
529dcb86d3
4 changed files with 107 additions and 27 deletions
|
@ -25,7 +25,7 @@ class SimpleTable(Table):
|
|||
])
|
||||
|
||||
def __init__(self, data, horizontal_align=None):
|
||||
Table.__init__(self, data, style=self.style, hAlign=horizontal_align)
|
||||
Table.__init__(self, data, hAlign=horizontal_align)
|
||||
|
||||
|
||||
class PaidStamp(object):
|
||||
|
|
|
@ -1,9 +1,12 @@
|
|||
from __future__ import unicode_literals
|
||||
|
||||
|
||||
class PDFInfo(object):
|
||||
def __init__(self, title, author, subject):
|
||||
def __init__(self, title=None, author=None, subject=None):
|
||||
self.title = title
|
||||
self.author = author
|
||||
self.subject = subject
|
||||
self.creator = 'pyinvlice (https://ciciapp.com/pyinvoice)'
|
||||
self.creator = 'PyInvoice (https://ciciapp.com/pyinvoice)'
|
||||
|
||||
|
||||
class InvoiceInfo(object):
|
||||
|
|
|
@ -1,10 +1,19 @@
|
|||
from __future__ import unicode_literals
|
||||
from datetime import datetime, date
|
||||
from reportlab.lib.pagesizes import letter
|
||||
from reportlab.lib.units import inch
|
||||
from reportlab.platypus import SimpleDocTemplate
|
||||
from pyinvoice.components import SimpleTable
|
||||
from pyinvoice.models import PDFInfo, Item, Transaction, InvoiceInfo, ServiceProviderInfo, ClientInfo
|
||||
|
||||
|
||||
class SimpleInvoice(SimpleDocTemplate):
|
||||
def __init__(self, invoice_path):
|
||||
default_pdf_info = PDFInfo(title='Invoice', author='CiCiApp.com', subject='Invoice')
|
||||
|
||||
def __init__(self, invoice_path, pdf_info=None):
|
||||
if not pdf_info:
|
||||
pdf_info = self.default_pdf_info
|
||||
|
||||
SimpleDocTemplate.__init__(
|
||||
self,
|
||||
invoice_path,
|
||||
|
@ -12,5 +21,70 @@ class SimpleInvoice(SimpleDocTemplate):
|
|||
rightMargin=inch,
|
||||
leftMargin=inch,
|
||||
topMargin=inch,
|
||||
bottomMargin=inch
|
||||
bottomMargin=inch,
|
||||
**pdf_info.__dict__
|
||||
)
|
||||
|
||||
self.invoice_info = None
|
||||
self.service_provider_info = None
|
||||
self.client_info = None
|
||||
self._items = []
|
||||
self._transactions = []
|
||||
|
||||
@property
|
||||
def items(self):
|
||||
return self._items[:]
|
||||
|
||||
def add_item(self, item):
|
||||
if isinstance(item, Item):
|
||||
self._items.append(item)
|
||||
|
||||
@property
|
||||
def transactions(self):
|
||||
return self._transactions[:]
|
||||
|
||||
def add_transaction(self, t):
|
||||
if isinstance(t, Transaction):
|
||||
self._transactions.append(t)
|
||||
|
||||
@staticmethod
|
||||
def __attribute_to_table_data(instance, attribute_verbose_name_list):
|
||||
data = []
|
||||
|
||||
for p, v in attribute_verbose_name_list:
|
||||
attr = getattr(instance, p)
|
||||
if attr is not None:
|
||||
# TODO: datetime format
|
||||
if isinstance(attr, datetime):
|
||||
attr = attr.strftime('%Y-%m-%d %H:%M')
|
||||
elif isinstance(attr, date):
|
||||
attr = attr.strftime('%Y-%m-%d')
|
||||
data.append(['{0}:'.format(v), attr])
|
||||
|
||||
return data
|
||||
|
||||
def finish(self):
|
||||
story = []
|
||||
|
||||
if isinstance(self.invoice_info, InvoiceInfo):
|
||||
props = [('invoice_id', 'Invoice id'), ('invoice_datetime', 'Invoice date'),
|
||||
('due_datetime', 'Invoice due date')]
|
||||
|
||||
story.append(
|
||||
SimpleTable(self.__attribute_to_table_data(self.invoice_info, props), horizontal_align='LEFT')
|
||||
)
|
||||
|
||||
if isinstance(self.service_provider_info, ServiceProviderInfo):
|
||||
props = [('name', 'Name'), ('street', 'Street'), ('city', 'City'), ('state', 'State'),
|
||||
('country', 'Country'), ('post_code', 'Post code')]
|
||||
|
||||
story.append(
|
||||
SimpleTable(self.__attribute_to_table_data(self.service_provider_info, props), horizontal_align='RIGHT')
|
||||
)
|
||||
|
||||
if isinstance(self.client_info, ClientInfo):
|
||||
props = [('name', 'Name'), ('street', 'Street'), ('city', 'City'), ('state', 'State'),
|
||||
('country', 'Country'), ('post_code', 'Post code'), ('email', 'Email'), ('client_id', 'Client id')]
|
||||
story.append(SimpleTable(self.__attribute_to_table_data(self.client_info, props), horizontal_align='LEFT'))
|
||||
|
||||
self.build(story)
|
45
test.py
45
test.py
|
@ -1,25 +1,28 @@
|
|||
from reportlab.lib.units import inch
|
||||
from pyinvoice.components import SimpleTable, PaidStamp
|
||||
from datetime import datetime
|
||||
from pyinvoice.models import InvoiceInfo, ServiceProviderInfo, ClientInfo
|
||||
from pyinvoice.templates import SimpleInvoice
|
||||
|
||||
story = []
|
||||
|
||||
address_table = SimpleTable([
|
||||
['Name', 'zhangshine'],
|
||||
['City', 'Jining'],
|
||||
['City', 'Jining'],
|
||||
['City', 'Jining'],
|
||||
['City', 'Jining'],
|
||||
['City', 'Jining']
|
||||
], horizontal_align='RIGHT')
|
||||
story.append(address_table)
|
||||
|
||||
merchant_table = SimpleTable([
|
||||
['Name', 'CiCiApp'],
|
||||
['xxxx', 'yyyy']
|
||||
], horizontal_align='LEFT')
|
||||
story.append(merchant_table)
|
||||
|
||||
doc = SimpleInvoice('test.pdf')
|
||||
doc.invoice_info = InvoiceInfo('1024', datetime.now(), datetime.now())
|
||||
doc.service_provider_info = ServiceProviderInfo(
|
||||
name='PyInvoice',
|
||||
street='My Street',
|
||||
city='My City',
|
||||
state='My State',
|
||||
country='My Country',
|
||||
post_code='My Post code'
|
||||
)
|
||||
|
||||
doc.build(story, onFirstPage=PaidStamp(7*inch, 5.8*inch))
|
||||
doc.client_info = ClientInfo(
|
||||
email='My Email',
|
||||
client_id='My Client Id',
|
||||
name='Client Name',
|
||||
street='Client Street',
|
||||
city='Client City',
|
||||
state='Client State',
|
||||
country='Client country',
|
||||
post_code='Client Post code'
|
||||
)
|
||||
|
||||
# doc.build(story, onFirstPage=PaidStamp(7*inch, 5.8*inch))
|
||||
doc.finish()
|
Loading…
Reference in a new issue