KumiSystems/PyInvoice
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

split finish method

Browse source
This commit is contained in:
zhangshine 2015-06-06 17:31:18 +08:00
parent fab6d5b18f
commit 55141fe361
1 changed files with 43 additions and 23 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

46
pyinvoice/templates.py
View file

@ -33,6 +33,7 @@ class SimpleInvoice(SimpleDocTemplate):
self.is_paid = False self.is_paid = False
self._items = [] self._items = []
self._transactions = [] self._transactions = []
self._story = []
@property @property
def items(self): def items(self):
@ -69,31 +70,34 @@ class SimpleInvoice(SimpleDocTemplate):
return data return data
def finish(self): def __build_invoice_info(self):
story = []
if isinstance(self.invoice_info, InvoiceInfo): if isinstance(self.invoice_info, InvoiceInfo):
props = [('invoice_id', 'Invoice id'), ('invoice_datetime', 'Invoice date'), props = [('invoice_id', 'Invoice id'), ('invoice_datetime', 'Invoice date'),
('due_datetime', 'Invoice due date')] ('due_datetime', 'Invoice due date')]
story.append( self._story.append(
SimpleTable(self.__attribute_to_table_data(self.invoice_info, props), horizontal_align='LEFT') SimpleTable(self.__attribute_to_table_data(self.invoice_info, props), horizontal_align='LEFT')
) )
def __build_service_provider_info(self):
if isinstance(self.service_provider_info, ServiceProviderInfo): if isinstance(self.service_provider_info, ServiceProviderInfo):
props = [('name', 'Name'), ('street', 'Street'), ('city', 'City'), ('state', 'State'), props = [('name', 'Name'), ('street', 'Street'), ('city', 'City'), ('state', 'State'),
('country', 'Country'), ('post_code', 'Post code')] ('country', 'Country'), ('post_code', 'Post code')]
story.append( self._story.append(
SimpleTable(self.__attribute_to_table_data(self.service_provider_info, props), horizontal_align='RIGHT') SimpleTable(self.__attribute_to_table_data(self.service_provider_info, props), horizontal_align='RIGHT')
) )
def __build_client_info(self):
# ClientInfo # ClientInfo
if isinstance(self.client_info, ClientInfo): if isinstance(self.client_info, ClientInfo):
props = [('name', 'Name'), ('street', 'Street'), ('city', 'City'), ('state', 'State'), props = [('name', 'Name'), ('street', 'Street'), ('city', 'City'), ('state', 'State'),
('country', 'Country'), ('post_code', 'Post code'), ('email', 'Email'), ('client_id', 'Client id')] ('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._story.append(
SimpleTable(self.__attribute_to_table_data(self.client_info, props), horizontal_align='LEFT')
)
def __build_items(self):
# Items # Items
item_table_paragraph_style = ParagraphStyle( item_table_paragraph_style = ParagraphStyle(
'ItemTableParagraph', 'ItemTableParagraph',
@ -101,27 +105,43 @@ class SimpleInvoice(SimpleDocTemplate):
alignment=TA_CENTER alignment=TA_CENTER
) )
item_data = [( item_data = [
(
item.item_id, item.item_id,
item.name, item.name,
Paragraph(item.description, item_table_paragraph_style), Paragraph(item.description, item_table_paragraph_style),
item.units, item.units,
item.unit_price, item.unit_price,
item.subtotal item.subtotal
) for item in self._items if isinstance(item, Item)] ) for item in self._items if isinstance(item, Item)
]
if item_data: if item_data:
item_data.insert(0, ('Item id', 'Name', 'Description', 'Units', 'Unit Price', 'Subtotal')) item_data.insert(0, ('Item id', 'Name', 'Description', 'Units', 'Unit Price', 'Subtotal'))
story.append(TableWithHeader(item_data, horizontal_align='LEFT')) self._story.append(TableWithHeader(item_data, horizontal_align='LEFT'))
def __build_transactions(self):
# Transaction # Transaction
transaction_table_data = [( transaction_table_data = [
(
t.transaction_id, t.transaction_id,
t.gateway, t.gateway,
self.__format_value(t.transaction_datetime), self.__format_value(t.transaction_datetime),
t.amount t.amount
) for t in self._transactions if isinstance(t, Transaction)] ) for t in self._transactions if isinstance(t, Transaction)
]
if transaction_table_data: if transaction_table_data:
transaction_table_data.insert(0, ('Transaction id', 'Gateway', 'Transaction date', 'Amount')) transaction_table_data.insert(0, ('Transaction id', 'Gateway', 'Transaction date', 'Amount'))
story.append(TableWithHeader(transaction_table_data, horizontal_align='LEFT')) self._story.append(TableWithHeader(transaction_table_data, horizontal_align='LEFT'))
self.build(story, onFirstPage=PaidStamp(7*inch, 5.8*inch) if self.is_paid else None) def finish(self):
self._story = []
self.__build_invoice_info()
self.__build_service_provider_info()
self.__build_client_info()
self.__build_items()
self.__build_transactions()
self.build(self._story, onFirstPage=PaidStamp(7 * inch, 5.8 * inch) if self.is_paid else None)