def edit(id=None): setExits() g.title = "Edit {} Record".format(g.title) category = Category(g.db) if request.form: id = request.form.get('id', None) id = cleanRecordID(id) if id >= 0 and not request.form: if id == 0: rec = category.new() else: rec = category.get(id) if rec: return render_template('category_edit.html', rec=rec) else: flash('Record not Found') if request.form: if not validate_form(): return render_template('category_edit.html', rec=request.form) if id == 0: rec = category.new() else: rec = category.get(id) if rec: category.update(rec, request.form) category.save(rec) try: g.db.commit() except Exception as e: g.db.rollback() flash( printException('Error attempting to save Category record', str(e))) return redirect(g.listURL) else: flash('Record not Found') return redirect(g.listURL)
def import_data(filespec): db = Database('instance/database.sqlite').connect() category = Category(db) uom = Uom(db) item = Item(db) trx = Transaction(db) with open(filespec, newline='') as csvfile: reader = csv.DictReader(csvfile) for row in reader: print('Name: {}, Category: {}'.format(row['item_name'], row['category_name'])) #import pdb;pdb.set_trace() #create a Category if not exists cats = category.select_one( where='name = "{}"'.format(row['category_name'])) if not cats: cats = category.new() cats.name = row["category_name"] category.save(cats) cat_id = cats.id #create UOM if not exists uom_name = row['uom'].strip() if uom_name == '': uom_name = "Ea." uoms = uom.select_one(where='name = "{}"'.format(uom_name)) if not uoms: uoms = uom.new() uoms.name = uom_name uom.save(uoms) items = item.select_one( where='name = "{}"'.format(row['item_name'], )) if not items: items = item.new() items.name = row['item_name'] items.uom = uom_name items.cat_id = cat_id item.save(items) # Create a transaction record trxs = trx.new() trx.update(trxs, row) trxs.created = datetime.strptime(row['created'], '%Y-%m-%d') trxs.item_id = items.id trx.save(trxs) try: db.commit() except Exception as e: db.rollback() print(str(e))