class UAVPartsAdd(Resource):
def __init__(self):
self.dao = PartsDao()
self.userDao = UserDAO()
def post(self):
if (request.data != ""):
data = json.loads(request.data)
token = data['token']
parts = data['parts']
parts_dict = json.loads(json.dumps(parts))
parts_obj = Parts()
parts_obj.parts_id = parts_dict[0]['parts_id']
parts_obj.parts_ver = parts_dict[0]['parts_ver']
parts_obj.parts_type = parts_dict[0]['parts_type']
parts_obj.parts_fact = parts_dict[0]['parts_fact']
parts_obj.parts_date = parts_dict[0]['parts_date']
parts_obj.user_team = parts_dict[0]['user_team']
parts_obj.parts_status = '在库'
parts_obj.parts_use_number = 0
user = self.userDao.verify_token(token, '')
if (not user):
return make_response(jsonify({'error': 'Unauthorized access'}),
401)
rs = self.dao.add_parts(user, parts_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 UAVPartsAdd(Resource):
def __init__(self):
self.dao = PartsDao()
self.userDao = UserDAO()
def post(self):
if (request.data != ""):
data = json.loads(request.data)
token = data['token']
parts = data['parts']
parts_dict = json.loads(json.dumps(parts))
parts_obj = Parts()
parts_obj.parts_id = parts_dict[0]['parts_id']
parts_obj.parts_ver = parts_dict[0]['parts_ver']
parts_obj.parts_type = parts_dict[0]['parts_type']
parts_obj.parts_fact = parts_dict[0]['parts_fact']
parts_obj.parts_date = datetime.strptime(
parts_dict[0]['parts_date'], '%Y-%m-%d').date()
parts_obj.user_team = parts_dict[0]['user_team']
parts_obj.parts_use_dpartment = parts_dict[0]['use_department']
parts_obj.parts_status = '在库'
parts_obj.parts_use_number = 0
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.add_parts(user, parts_obj)
if rs == 1:
return make_response(jsonify({'success': '添加设备成功'}), 200)
elif rs == 2030701:
return make_response(
jsonify({
'existed': '添加的设备已经存在',
'error': rs
}), 401)
elif rs == 2030702:
return make_response(
jsonify({
'existed': '无权限添加设备',
'error': rs
}), 401)
else:
return make_response(
jsonify({
'error': '输入参数有误',
'errorcode': 10000000
}), 401)
def get(self):
return self.post()