def manage_upload_addmodel(request): data = yield from request.post() vendor = data['vd_add'] model = data['md_add'] select = data['adde'] vendor_db = yield from Vendor.findAll('vendor_name = ?', vendor) vid = vendor_db[0]['vendor_id'] md_count = yield from DM.findAll(where='vendor_id=? and model_name=?', args=[vid,model]) # Add model if select=='1': if len(md_count)==1: return web.StreamResponse(status=999) else: model_new = DM(vendor_id=vid,model_name=model) yield from model_new.save() return 'Success' #Delete model else: if len(md_count)==0: return web.StreamResponse(status=997) else: fw_count = yield from Firmware.findAll(where='fw_vendor_name=? and fw_model_name=?',args=[vendor,model]) if len(fw_count)>0: return web.StreamResponse(status=998) else: model_del = yield from DM.find(md_count[0].model_id) yield from model_del.remove() return 'Success'
def api_drive(): vendors = yield from Vendor.findAll(orderBy='vendor_name') models = yield from DM.findAll(orderBy='model_name') firmwares = yield from Firmware.findAll() return dict(vendor=vendors,model=models,firmware=firmwares)