def update_department(department_name):
try:
department = Department.json_parse(request.json)
department.department_name = department_name
DepartmentService.update_department(department)
return jsonify(department.json()), 200
except KeyError:
return "Please enter a new Department Head for this Department.", 400
except ResourceNotFound as r:
return r.message, 404
def query_department(shopId, parentId):
node = request.args.get('node')
ds = DepartmentService()
if not shopId.isdigit() or not parentId.isdigit():
res = {}
res['success'] = False
res['msg'] = MessageConstants_CN.MSG_INVALID_ARGS
return json.dumps(res)
if not node or not node.isdigit():
res = ds.query(shopId=shopId, parentId=-1)
return res
else:
res = ds.query(shopId=shopId, parentId=node)
return res
def remove_department():
print(request.form)
ds = DepartmentService()
try:
shopId = request.form['shopId']
nodeId = request.form['id']
if not shopId or not nodeId:
res = {}
res['success'] = False
res['msg'] = MessageConstants_CN.MSG_INVALID_ARGS
return json.dumps(res)
_, result = ds.remove(shopId=shopId, nodeId=nodeId)
return json.dumps(result)
except Exception as ex:
result = {}
result['success'] = True
result['msg'] = MessageConstants_CN.MSG_INTER_ERROR
return json.dumps(result)
def post_department():
try:
department = Department.json_parse(request.json)
department = DepartmentService.add_department(department)
return jsonify(department.json()), 201
except KeyError:
return "A name and department head must be entered to add a department.", 400
except ResourceNotFound as r:
return r.message, 404
def update_department():
print(request.form)
ds = DepartmentService()
try:
nodeId = request.form['id']
shopId = request.form['shopId']
name = request.form['name']
newdata = {}
newdata['id'] = nodeId
newdata['shopId'] = shopId
newdata['name'] = name
_, result = ds.update(newdata)
print(result)
return json.dumps(result)
except Exception as ex:
print(ex)
res = {}
res['success'] = False
res['msg'] = MessageConstants_CN.MSG_INTER_ERROR
return json.dumps(res)
def add_department():
print(request.form)
ds = DepartmentService()
try:
name = request.form['name']
shopId = request.form['shopId']
parentId = request.form['pid']
if not name or not shopId or not parentId:
result = {}
result['success'] = False
result['msg'] = MessageConstants_CN.MSG_INVALID_ARGS
return json.dumps(result)
_, res = ds.add(shopId=shopId, parentId=parentId, name=name)
print(res)
return json.dumps(res)
except Exception as ex:
print(ex)
res = {}
res['success'] = False
res['msg'] = MessageConstants_CN.MSG_INTER_ERROR
return json.dumps(res)
def delete_department(department_name):
try:
return DepartmentService.delete_department(department_name)
except KeyError:
return f"No Department with name {department_name} found.", 400
def get_department(department_name):
try:
department = DepartmentService.get_department(department_name)
return jsonify(department.json()), 200
except ResourceNotFound as r:
return r.message, 404
def get_all_departments():
return jsonify(DepartmentService.get_all_departments())