def save(self):
item = Item()
for k, v in self.form.elements.items():
setattr(item, k, v.text())
item.save()
self.showItems()
self.window.destroyFrame('Ui_Item_New')
self.window.setMessage("Nieuwe artikel " + str(item.id) + " aangemaakt!")
def saveDelivery(delivery): item = Item() deliveryObj = Delivery() deliveryObj.begindate = getDate(delivery['DATVAN'], 2000, 1, 1) deliveryObj.enddate = getDate(delivery['DATTOT'], 2999, 12, 31) deliveryObj.item_id = item.getItemIdByName(delivery['DIENST']) deliveryObj.client_id = delivery['KLANT'] deliveryObj.days = getDays(delivery) deliveryObj.save() print deliveryObj
def saveItem(item): itemObj = Item() itemObj.name = item['DIENST'] itemObj.description = unicode(item['OMSCHR'], errors='replace') itemObj.days = getDaysItem(item, itemObj) itemObj.freq = getFreq(item) itemObj.save() savePrices(item, itemObj.id)
def setUp(self): self.item = Item()
class TestItemModel(unittest.TestCase):
def setUp(self):
self.item = Item()
def test_isDeliveryDay(self):
self.assertEqual(True, self.item.isDeliveryDay(date(2012, 9, 3), 2), 'Test Monday')
self.assertEqual(True, self.item.isDeliveryDay(date(2012, 9, 4), 4))
self.assertEqual(True, self.item.isDeliveryDay(date(2012, 9, 5), 8))
self.assertEqual(True, self.item.isDeliveryDay(date(2012, 9, 6), 16))
self.assertEqual(True, self.item.isDeliveryDay(date(2012, 9, 7), 32))
self.assertEqual(True, self.item.isDeliveryDay(date(2012, 9, 8), 64))
self.assertEqual(True, self.item.isDeliveryDay(date(2012, 9, 3), 6))
self.assertEqual(True, self.item.isDeliveryDay(date(2012, 9, 3), 10))
self.assertEqual(True, self.item.isDeliveryDay(date(2012, 9, 3), 18))
self.assertEqual(True, self.item.isDeliveryDay(date(2012, 9, 3), 62), 'Test Monday')
self.assertEqual(True, self.item.isDeliveryDay(date(2012, 9, 4), 62))
self.assertEqual(True, self.item.isDeliveryDay(date(2012, 9, 5), 62))
self.assertEqual(True, self.item.isDeliveryDay(date(2012, 9, 6), 62))
self.assertEqual(True, self.item.isDeliveryDay(date(2012, 9, 7), 62))
def test_getDay(self):
self.assertEqual(2, self.item.getDayBitByDate(date(2012, 9, 3)), 'Test Monday')
self.assertEqual(4, self.item.getDayBitByDate(date(2012, 9, 4)), 'Test Tuesday')
self.assertEqual(8, self.item.getDayBitByDate(date(2012, 9, 5)), 'Test Wednesday')
self.assertEqual(16, self.item.getDayBitByDate(date(2012, 9, 6)), 'Test Thursday')
self.assertEqual(32, self.item.getDayBitByDate(date(2012, 9, 7)), 'Test Friday')
self.assertEqual(64, self.item.getDayBitByDate(date(2012, 9, 8)), 'Test Saturday')
def test_getDayBit(self):
self.assertEqual(2, self.item.getDayBit(0))
self.assertEqual(4, self.item.getDayBit(1))
self.assertEqual(8, self.item.getDayBit(2))
self.assertEqual(16, self.item.getDayBit(3))
self.assertEqual(32, self.item.getDayBit(4))
self.assertEqual(64, self.item.getDayBit(5))
