def handle_add(self, converted): customer = Customer() self._apply_data(customer, converted) session = DBSession() session.add(customer) statusmessage.show(self.request, u"Customer added.", "success") return HTTPFound(location=route_url('customers', self.request))
def _apply_data(self, invoice, converted): changed = False # Apply schema fields to the customer object field_names = [p.key for p in class_mapper(Invoice).iterate_properties] for field_name in field_names: if field_name in converted.keys(): if getattr(invoice, field_name) != converted[field_name]: setattr(invoice, field_name, converted[field_name]) changed = True if invoice.due_date != invoice.date + datetime.timedelta( days=converted['payment_term']): invoice.due_date = invoice.date + datetime.timedelta( days=converted['payment_term']) changed = True if converted[ 'recurring_term'] and invoice.recurring_date != invoice.date + datetime.timedelta( days=converted['recurring_term']): invoice.recurring_date = invoice.date + datetime.timedelta( days=converted['recurring_term']) changed = True # Apply data of the items subforms session = DBSession() item_map = {} for item in invoice.items: item_map[item.id] = item for index, item_data in enumerate(converted['item_list']): if item_data['item_id']: item_id = item_data['item_id'] item = item_map[item_id] del item_map[item_id] else: item = InvoiceItem() item.invoice = invoice session.add(item) changed = True # Apply schema fields to the invoice item object field_names = [ p.key for p in class_mapper(InvoiceItem).iterate_properties ] for field_name in field_names: if field_name in item_data.keys(): if getattr(item, field_name) != item_data[field_name]: setattr(item, field_name, item_data[field_name]) changed = True if item.item_number != index: item.item_number = index changed = True # Remove invoice items that have been removed in the form for item in item_map.values(): invoice.items.remove(item) changed = True return changed
def test_next_invoice_number_used(self): from seantisinvoice.models import DBSession from seantisinvoice.models import Company from seantisinvoice.models import Invoice from seantisinvoice.models import next_invoice_number session = DBSession() company = session.query(Company).one() company.invoice_start_number = 20 invoice = Invoice(invoice_number=20) session.add(invoice) session.flush() result = next_invoice_number() self.assertEqual(21, result) self.assertEqual(22, company.invoice_start_number)
def handle_add(self, converted): session = DBSession() invoice = Invoice() invoice.company = session.query(Company).first() self._apply_data(invoice, converted) session.add(invoice) # Get and add unique invoice number if invoice.invoice_number is None: invoice.invoice_number = next_invoice_number() statusmessage.show(self.request, u"Invoice added.", "success") return HTTPFound(location=route_url('invoices', self.request))
def test_recurring(self): from seantisinvoice.models import DBSession from seantisinvoice.models import Invoice, InvoiceItem from seantisinvoice.recurring import copy_recurring session = DBSession() copy_recurring() self.assertEquals(0, session.query(Invoice).count()) # Add a new invoice invoice = Invoice() invoice_date = datetime.date.today() - datetime.timedelta(days=100) invoice.date = invoice_date invoice.due_date = datetime.date.today() - datetime.timedelta(days=70) invoice.project_description = u'Invoice project description' session.add(invoice) # Add an item to the invoice item = InvoiceItem() item.item_number = 0 item.amount = 1000 item.service_description = u'Item description' item.service_title = u'Invoice item' item.invoice = invoice session.merge(item) copy_recurring() # Invoice not cloned as it has no recurring_date self.assertEquals(1, session.query(Invoice).count()) invoice = session.query(Invoice).one() # Set the recurring date on the invoice recurring_date = datetime.date.today() - datetime.timedelta(days=70) invoice.recurring_date = recurring_date copy_recurring() # There are now two invoices self.assertEquals(2, session.query(Invoice).count()) invoice = session.query(Invoice).filter_by(date=invoice_date).one() # Recurring date of the original invoice has been reset self.failIf(invoice.recurring_date) cloned_invoice = session.query(Invoice).filter_by( date=recurring_date).one() # New invoice has the recurring date of the original one as date self.assertEquals(recurring_date, cloned_invoice.date) self.assertEquals(recurring_date + datetime.timedelta(days=30), cloned_invoice.recurring_date) # Check whether attributes on invoice and invoice item have been cloned self.assertEquals(invoice.project_description, cloned_invoice.project_description) cloned_item = cloned_invoice.items[0] self.assertEquals(item.amount, cloned_item.amount) self.assertEquals(item.service_description, cloned_item.service_description) self.assertEquals(item.service_title, cloned_item.service_title)
def _apply_data(self, invoice, converted): changed = False # Apply schema fields to the customer object field_names = [ p.key for p in class_mapper(Invoice).iterate_properties ] for field_name in field_names: if field_name in converted.keys(): if getattr(invoice, field_name) != converted[field_name]: setattr(invoice, field_name, converted[field_name]) changed = True if invoice.due_date != invoice.date + datetime.timedelta(days=converted['payment_term']): invoice.due_date = invoice.date + datetime.timedelta(days=converted['payment_term']) changed = True if converted['recurring_term'] and invoice.recurring_date != invoice.date + datetime.timedelta(days=converted['recurring_term']): invoice.recurring_date = invoice.date + datetime.timedelta(days=converted['recurring_term']) changed = True # Apply data of the items subforms session = DBSession() item_map = {} for item in invoice.items: item_map[item.id] = item for index, item_data in enumerate(converted['item_list']): if item_data['item_id']: item_id = item_data['item_id'] item = item_map[item_id] del item_map[item_id] else: item = InvoiceItem() item.invoice = invoice session.add(item) changed = True # Apply schema fields to the invoice item object field_names = [ p.key for p in class_mapper(InvoiceItem).iterate_properties ] for field_name in field_names: if field_name in item_data.keys(): if getattr(item, field_name) != item_data[field_name]: setattr(item, field_name, item_data[field_name]) changed = True if item.item_number != index: item.item_number = index changed = True # Remove invoice items that have been removed in the form for item in item_map.values(): invoice.items.remove(item) changed = True return changed
def test_recurring(self): from seantisinvoice.models import DBSession from seantisinvoice.models import Invoice, InvoiceItem from seantisinvoice.recurring import copy_recurring session = DBSession() copy_recurring() self.assertEquals(0, session.query(Invoice).count()) # Add a new invoice invoice = Invoice() invoice_date = datetime.date.today() - datetime.timedelta(days=100) invoice.date = invoice_date invoice.due_date = datetime.date.today() - datetime.timedelta(days=70) invoice.project_description = u"Invoice project description" session.add(invoice) # Add an item to the invoice item = InvoiceItem() item.item_number = 0 item.amount = 1000 item.service_description = u"Item description" item.service_title = u"Invoice item" item.invoice = invoice session.merge(item) copy_recurring() # Invoice not cloned as it has no recurring_date self.assertEquals(1, session.query(Invoice).count()) invoice = session.query(Invoice).one() # Set the recurring date on the invoice recurring_date = datetime.date.today() - datetime.timedelta(days=70) invoice.recurring_date = recurring_date copy_recurring() # There are now two invoices self.assertEquals(2, session.query(Invoice).count()) invoice = session.query(Invoice).filter_by(date=invoice_date).one() # Recurring date of the original invoice has been reset self.failIf(invoice.recurring_date) cloned_invoice = session.query(Invoice).filter_by(date=recurring_date).one() # New invoice has the recurring date of the original one as date self.assertEquals(recurring_date, cloned_invoice.date) self.assertEquals(recurring_date + datetime.timedelta(days=30), cloned_invoice.recurring_date) # Check whether attributes on invoice and invoice item have been cloned self.assertEquals(invoice.project_description, cloned_invoice.project_description) cloned_item = cloned_invoice.items[0] self.assertEquals(item.amount, cloned_item.amount) self.assertEquals(item.service_description, cloned_item.service_description) self.assertEquals(item.service_title, cloned_item.service_title)
def _add_customer(self): from seantisinvoice.models import Customer from seantisinvoice.models import CustomerContact from seantisinvoice.models import DBSession session = DBSession() customer = Customer() customer.name = u'Customers Inc.' customer.address1 = u'Street' customer.postal_code = u'12234' customer.city = u'Dublin' session.add(customer) # Each customer needs at least one contact contact = CustomerContact() contact.first_name = u'Buck' contact.last_name = u'Mulligan' contact.title = u'Mr.' contact.customer = customer session.add(contact) session.flush() return customer
def copy_recurring(): today = datetime.date.today() session = DBSession() query = session.query(Invoice) query = query.filter(Invoice.recurring_date <= today) query = query.filter( or_(Invoice.recurring_stop == None, Invoice.recurring_stop > today)) for invoice in query.all(): # Clone invoice and invoice items invoice_clone = Invoice() invoice_clone.company = invoice.company invoice_clone.contact = invoice.contact invoice_clone.project_description = invoice.project_description invoice_clone.currency = invoice.currency invoice_clone.tax = invoice.tax session.add(invoice_clone) for item in invoice.items: item_clone = InvoiceItem() item_clone.item_number = item.item_number item_clone.amount = item.amount item_clone.hours = item.hours item_clone.days = item.days item_clone.service_description = item.service_description item_clone.service_title = item.service_title item_clone.invoice = invoice_clone session.add(item_clone) # Get new invoice number invoice_clone.invoice_number = next_invoice_number() # Adjust dates on cloned invoice invoice_clone.date = invoice.recurring_date invoice_clone.due_date = invoice_clone.date + (invoice.due_date - invoice.date) invoice_clone.recurring_date = invoice_clone.date + ( invoice.recurring_date - invoice.date) invoice_clone.recurring_stop = invoice.recurring_stop # Old invoice is not recurring anymore invoice.recurring_date = None transaction.commit()
def _add_customer(self): from seantisinvoice.models import Customer from seantisinvoice.models import CustomerContact from seantisinvoice.models import DBSession session = DBSession() customer = Customer() customer.name = u"Customers Inc." customer.address1 = u"Street" customer.postal_code = u"12234" customer.city = u"Dublin" session.add(customer) # Each customer needs at least one contact contact = CustomerContact() contact.first_name = u"Buck" contact.last_name = u"Mulligan" contact.title = u"Mr." contact.customer = customer session.add(contact) session.flush() return customer
def _apply_data(self, customer, converted): changed = False session = DBSession() # Apply schema fields to the customer object field_names = [ p.key for p in class_mapper(Customer).iterate_properties ] for field_name in field_names: if field_name in converted.keys(): if getattr(customer, field_name) != converted[field_name]: setattr(customer, field_name, converted[field_name]) changed = True # Apply data of the contact subforms contact_map = {} for contact in customer.contacts: contact_map[contact.id] = contact for contact_data in converted['contact_list']: if contact_data['contact_id']: contact_id = contact_data['contact_id'] contact = contact_map[contact_id] del contact_map[contact_id] else: contact = CustomerContact() contact.customer = customer session.add(contact) changed = True # Apply schema fields to the customer contact object field_names = [ p.key for p in class_mapper(CustomerContact).iterate_properties ] for field_name in field_names: if field_name in contact_data.keys(): if getattr(contact, field_name) != contact_data[field_name]: setattr(contact, field_name, contact_data[field_name]) changed = True # Remove contact items that have been removed in the form for contact in contact_map.values(): # FIXME: what happens to existing invoices that loose their contact? customer.contacts.remove(contact) changed = True return changed
def copy_recurring(): today = datetime.date.today() session = DBSession() query = session.query(Invoice) query = query.filter(Invoice.recurring_date <= today) query = query.filter(or_(Invoice.recurring_stop == None, Invoice.recurring_stop > today)) for invoice in query.all(): # Clone invoice and invoice items invoice_clone = Invoice() invoice_clone.company = invoice.company invoice_clone.contact = invoice.contact invoice_clone.project_description = invoice.project_description invoice_clone.currency = invoice.currency invoice_clone.tax = invoice.tax session.add(invoice_clone) for item in invoice.items: item_clone = InvoiceItem() item_clone.item_number = item.item_number item_clone.amount = item.amount item_clone.hours = item.hours item_clone.days = item.days item_clone.service_description = item.service_description item_clone.service_title = item.service_title item_clone.invoice = invoice_clone session.add(item_clone) # Get new invoice number invoice_clone.invoice_number = next_invoice_number() # Adjust dates on cloned invoice invoice_clone.date = invoice.recurring_date invoice_clone.due_date = invoice_clone.date + (invoice.due_date - invoice.date) invoice_clone.recurring_date = invoice_clone.date + (invoice.recurring_date - invoice.date) invoice_clone.recurring_stop = invoice.recurring_stop # Old invoice is not recurring anymore invoice.recurring_date = None transaction.commit()
def _add_invoice(self): from seantisinvoice.models import Invoice from seantisinvoice.models import InvoiceItem from seantisinvoice.models import DBSession session = DBSession() customer = self._add_customer() invoice = Invoice() invoice.project_description = u'Project' invoice.date = datetime.date.today() invoice.due_date = invoice.date + datetime.timedelta(days=30) invoice.currency = u'CHF' invoice.contact = customer.contacts[0] session.add(invoice) # Add invoice item to the invoice item = InvoiceItem() item.item_number = 0 item.service_title = u'Testing' item.service_description = u'A lot of work!' item.amount = 1000.0 item.invoice = invoice session.add(item) session.flush() return invoice
def _add_invoice(self): from seantisinvoice.models import Invoice from seantisinvoice.models import InvoiceItem from seantisinvoice.models import DBSession session = DBSession() customer = self._add_customer() invoice = Invoice() invoice.project_description = u"Project" invoice.date = datetime.date.today() invoice.due_date = invoice.date + datetime.timedelta(days=30) invoice.currency = u"CHF" invoice.contact = customer.contacts[0] session.add(invoice) # Add invoice item to the invoice item = InvoiceItem() item.item_number = 0 item.service_title = u"Testing" item.service_description = u"A lot of work!" item.amount = 1000.0 item.invoice = invoice session.add(item) session.flush() return invoice