def read_export(self, customer): """Reads the TLW export based on customer name. If the PO number is not already in the program data, adds it""" export_file = self.get_export_file(customer) with open(export_file, 'r') as export: for line in export: line = line.rstrip('\n').rstrip('\r').split(',') if line[1] != 'PO #': if line[1].lstrip('0') not in self.purchase_orders: PO = PurchaseOrder(line[1].lstrip('0'), customer) PO.cancel_ship = line[4] #PO.total_cost = self.get_cost_from_exp(PO.po_number, # export_file) PO.get_items_from_export(export_file) PO.get_stores_from_export(export_file) for item in PO.items.itervalues(): PO.total_cost += (item.cost * item.total_qty) self.purchase_orders[PO.po_number] = PO else: PO = self.purchase_orders[line[1].lstrip('0')] PO.cancel_ship = line[4] #PO.total_cost = self.get_cost_from_exp(PO.po_number, #export_file) PO.get_items_from_export(export_file) PO.get_stores_from_export(export_file) PO.total_cost = 0 for item in PO.items.itervalues(): PO.total_cost += (item.cost * item.total_qty) self.purchase_orders.sync()
def read_db(self, db_file): """Reads POs from supplied CSV file and adds to the database object""" with open(db_file, 'r') as db: for line in db: if len(line) > 1: line = line.replace('\n','').replace('\r','').split(',') PO = PurchaseOrder(line[0], line[1]) PO.creation_date = line[2] PO.start_ship = line[3] PO.cancel_ship = line[4] PO.total_cost = line[5] PO.discount = line[6] PO.label = line[7] PO.complete = line[8] self.purchase_orders[PO.po_number] = PO