class UAVDeviceModify(Resource): def __init__(self): self.dao = DeviceDAO() self.userDao = UserDAO() def post(self): if (request.data != ""): data = json.loads(request.data) token = data['token'] device_json = data['device'] device_dict = json.loads(json.dumps(device_json)) device_obj = Device() device_obj.device_id = int(device_dict[0]['device_id']) device_obj.device_ver = device_dict[0]['device_ver'] device_obj.device_type = device_dict[0]['device_type'] device_obj.uad_code = device_dict[0]['uad_code'] device_obj.device_fact = device_dict[0]['device_fact'] device_obj.device_date = device_dict[0]['device_date'] device_obj.user_team = device_dict[0]['user_team'] device_obj.uad_camera = device_dict[0]['uad_camera'] device_obj.uav_yuntai = device_dict[0]['uav_yuntai'] device_obj.uad_rcontrol = device_dict[0]['uad_rcontrol'] user = self.userDao.verify_token(token, '') if (not user): return make_response(jsonify({'error': 'Unauthorized access'}), 401) rs = self.dao.modify_device(user, device_obj) if rs == 1: return make_response( jsonify({'success': 'add device success'}), 200) else: return make_response(jsonify({'failed': 'add device failed'}), 401) else: return make_response(jsonify({'error': 'Unauthorized access'}), 401) def get(self): return self.post()
class UAVDeviceModify(Resource): def __init__(self): self.dao = DeviceDAO() self.userDao = UserDAO() def post(self): if (request.data != ""): data = json.loads(request.data) token = data['token'] device_json = data['device'] device_dict = json.loads(json.dumps(device_json)) device_obj = Device() device_obj.device_id = int(device_dict[0]['device_id']) device_obj.device_ver = device_dict[0]['device_ver'] device_obj.device_type = device_dict[0]['device_type'] device_obj.uad_code = device_dict[0]['uad_code'] device_obj.device_fact = device_dict[0]['device_fact'] device_obj.device_date = datetime.strptime( device_dict[0]['device_date'], '%Y-%m-%d').date() device_obj.user_team = device_dict[0]['user_team'] device_obj.uad_camera = device_dict[0]['uad_camera'] device_obj.uav_yuntai = device_dict[0]['uav_yuntai'] device_obj.uad_rcontrol = device_dict[0]['uad_rcontrol'] device_obj.device_use_dpartment = device_dict[0]['use_department'] user = self.userDao.verify_token(token, '') if (not user): return make_response( jsonify({ 'error': '用户不存在或登录过期', 'errorcode': 10000000 }), 400) if user == 1010301: return make_response( jsonify({ 'error': '登录过期', 'errorcode': user }), 400) if user == 1010302: return make_response( jsonify({ 'error': '用户验证错误', 'errorcode': user }), 400) rs = self.dao.modify_device(user, device_obj) if rs == 1: return make_response(jsonify({'success': '修改设备成功'}), 200) elif rs == 2011201: return make_response( jsonify({ 'failed': '待修改的设备不存在', 'errorcode': rs }), 401) else: return make_response( jsonify({ 'failed': '无权限修改设备', 'errorcode': rs }), 401) else: return make_response( jsonify({ 'error': '输入参数有误', 'errorcode': 10000000 }), 401) def get(self): return self.post()