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)
