def wrapper(*args, **kwargs):
token = request.headers.get("Authorization", default=None)
if not token:
log.logError("login_required: 未登录请先登陆")
return failReturn("", "未登录请先登陆"), 401
user_id = identify(token)
if not user_id:
log.logError("login_required: 未登录请先登陆")
return failReturn("", "未登录请先登陆"), 401
# 获取到用户并写入到session中,方便后续使用
session["user_id"] = user_id
return f(*args, **kwargs)
def get_patient_by_name():
"""
根据username获取病人
:return: json
---
tags:
- Manager API
parameters:
- name: body
in: body
required: true
schema:
id: 根据username获取病人
required:
- username
properties:
username:
type: string
description: username
- name: Authorization
in: header
type: string
required: true
description: token
responses:
success:
schema:
type: object
properties:
status:
type: string
description: success.
msg:
type: string
description: 获取病人成功
data:
type: object
properties:
patient:
type: object
properties:
id:
type: integer
example: 1
doctor:
type: string
example: zj
name:
type: string
example: zj
sex:
type: string
example: 男
recordID:
type: string
example: 123546
age:
type: string
example: 21
cva:
type: string
example: 血栓性脑埂塞
info:
type: string
example: 信息
state:
type: string
example: 急性期
result:
type: string
example: 结果
updateTime:
type: string
example: date-time
createTime:
type: string
example: date-time
description: 成功
fail:
schema:
type: object
properties:
status:
type: string
example: fail.
msg:
type: string
example: getPatientByName出错 or 获取病人失败
data:
type: string
example: error
description: 失败
"""
try:
json = request.get_json()
username = json['username']
patient = _get_patient_name(username)
if not patient:
return failReturn("", "getPatientByName: 获取病人失败")
response_object = {'patient': patient}
return successReturn(response_object, "getPatientByName: 获取病人成功")
except Exception as e:
return failReturn(format(e), "getPatientByName出错")
def get_patients():
"""
获取病人列表
:return: json
---
tags:
- Manager API
parameters:
- name: Authorization
in: header
type: string
required: true
description: token
responses:
success:
schema:
type: object
properties:
status:
type: string
example: success.
msg:
type: string
example: 获取病人列表成功
data:
type: object
properties:
userInfo:
type: array
items:
type: object
properties:
id:
type: integer
example: 1
doctor:
type: string
example: zj
name:
type: string
example: zj
sex:
type: string
example: 男
recordID:
type: string
example: 123546
age:
type: string
example: 21
cva:
type: string
example: 血栓性脑埂塞
info:
type: string
example: 信息
state:
type: string
example: 急性期
result:
type: string
example: 结果
updateTime:
type: string
example: date-time
createTime:
type: string
example: date-time
description: 成功
fail:
schema:
type: object
properties:
status:
type: string
example: fail.
msg:
type: string
example: getPatients出错
data:
type: string
example: error
description: 失败
"""
try:
patientList = _get_patients()
response_object = {'patientList': patientList}
return successReturn(response_object, "getPatients： 获取病人列表成功")
except Exception as e:
return failReturn(format(e), "getPatients出错")
def add_patient():
"""
添加病人或编辑病人状态
:return: json
---
tags:
- Manager API
parameters:
- name: body
in: body
required: true
schema:
id: 添加病人或编辑病人状态
required:
- name
properties:
name:
type: string
description: name
sex:
type: integer
description: 0为男 1为女
age:
type: integer
description: age
info:
type: string
description: 病人基础信息
result:
type: string
description: 检查结果
recordID:
type: string
description: 病例ID
state:
type: string
description: 病人状态
cva:
type: string
description: 脑卒中分类
- name: Authorization
in: header
type: string
required: true
description: token
responses:
success:
schema:
type: object
properties:
status:
type: string
example: success.
msg:
type: string
example: 病人已成功添加 or 病人已存在，已更新数据
data:
type: object
properties:
patient:
type: object
properties:
id:
type: integer
example: 1
doctor:
type: string
example: zj
name:
type: string
example: zj
sex:
type: string
example: 男
recordID:
type: string
example: 123546
age:
type: string
example: 21
cva:
type: string
example: 血栓性脑埂塞
info:
type: string
example: 信息
state:
type: string
example: 急性期
result:
type: string
example: 结果
updateTime:
type: string
example: date-time
createTime:
type: string
example: date-time
description: patient
description: 成功
fail:
schema:
type: object
properties:
status:
type: string
example: fail.
msg:
type: string
example: addPatient出错
data:
type: string
example: error
description: 失败
"""
try:
json = request.get_json()
name = json['name']
sex = json['sex']
age = json['age']
info = json['info']
result = json['result']
recordID = json['recordID']
state = json['state']
cva = json['cva']
patient, msg = _add_patient(name, sex, age, info, result, recordID,
state, cva)
response_object = {'patient': patient}
return successReturn(response_object, msg)
except Exception as e:
return failReturn(format(e), "addPatient出错")
def get_user():
"""
个人信息页
:return: json
---
tags:
- Manager API
parameters:
- name: Authorization
in: header
type: string
required: true
description: token
responses:
success:
schema:
type: object
properties:
status:
type: string
example: success.
msg:
type: string
example: 获取信息成功
data:
type: object
properties:
username:
type: string
example: username
id:
type: integer
example: id
userType:
type: string
example: userType
description: 成功
fail:
schema:
type: object
properties:
status:
type: string
example: fail.
msg:
type: string
example: 获取信息失败 or getUser出错
data:
type: string
example: error 或 空
description: 失败
"""
try:
username, id, userType = _get_username()
if not username:
return failReturn("", "getUser: 获取信息失败")
return successReturn(
{
"username": username,
"id": id,
"userType": userType
}, "getUser: 获取信息成功")
except Exception as e:
return failReturn(format(e), "getUser出错")
def refresh_token():
"""
刷新token，获取新的数据获取token, 当前jwt过期后可请求refresh
:return:
---
tags:
- Manager API
parameters:
- name: body
in: body
required: true
schema:
id: 刷新token
required:
- refresh_token
properties:
refresh_token:
type: string
description: refresh_token
responses:
success:
schema:
type: object
properties:
status:
type: string
example: success.
msg:
type: string
example: 刷新成功
data:
type: object
properties:
access_token:
type: string
example: access_token
description: 成功
fail:
schema:
type: object
properties:
status:
type: string
example: fail.
msg:
type: string
example: 参数错误 or 请登陆 or refreshToken出错
data:
type: string
example: error 或 空
description: 失败
"""
try:
user_data = request.get_json()
refreshToken = user_data['refresh_token']
if not refreshToken:
return failReturn("", "refreshToken: 参数错误")
payload = decode_auth_token(refreshToken)
if not payload:
return failReturn("", "refreshToken: 请登陆")
if "user_id" not in payload:
return failReturn("", "refreshToken: 请登陆")
access_token = generate_access_token(user_id=payload["user_id"])
data = {"access_token": access_token.decode("utf-8")}
return successReturn(data, "refreshToken: 刷新成功")
except Exception as e:
return failReturn(format(e), "refreshToken出错")
def register():
"""
用户注册
:return: json
---
tags:
- Manager API
parameters:
- name: body
in: body
required: true
schema:
id: 用户注册
required:
- username
- password
- realname
properties:
username:
type: string
description: 用户名
password:
type: string
description: 密码
realname:
type: string
description: 真实姓名
responses:
success:
schema:
type: object
properties:
status:
type: string
example: success.
msg:
type: string
example: 注册成功
data:
type: object
example: err
description: 注册成功
fail:
schema:
type: object
properties:
status:
type: string
example: fail.
msg:
type: string
example: 注册失败 or register出错
data:
type: string
example: error
description: 注册失败
"""
try:
user_data = request.get_json()
form = RegisterForm(data=user_data)
if form.validate():
user = User(username=user_data['username'],
password=user_data['password'],
realname=user_data['realname'])
db.session.add(user)
db.session.commit()
return successReturn("", "register: 注册成功")
return failReturn(form.errors, "register: 注册失败")
except Exception as e:
return failReturn(format(e), "register出错")
def login():
"""
用户登录
:return: json
---
tags:
- Manager API
parameters:
- name: body
in: body
required: true
schema:
id: 用户登录
required:
- username
- password
properties:
username:
type: string
description: 用户名
password:
type: string
description: 密码
responses:
success:
schema:
type: object
properties:
status:
type: string
example: success.
msg:
type: string
example: login 登陆成功
data:
type: object
properties:
access_token:
type: string
example: access_token
refresh_token:
type: string
example: refresh_token
description: 登陆成功
fail:
schema:
type: object
properties:
status:
type: string
example: fail.
msg:
type: string
example: login 用户名或密码错误 or login出错
data:
type: string
example: error或空
description: 用户名或密码错误
"""
try:
user_data = request.get_json()
form = LoginForm(data=user_data)
if form.validate_on_submit():
user = form.get_user()
access_token = generate_access_token(user_id=user.id)
refreshToken = generate_refresh_token(user_id=user.id)
data = {
"access_token": access_token.decode("utf-8"),
"refresh_token": refreshToken.decode("utf-8")
}
return successReturn(data, "login: 登陆成功")
return failReturn("", "login: 用户名或密码错误")
except Exception as e:
return failReturn(format(e), "login出错")
def patients_analyze():
"""
统计结果分析
:return: json
---
tags:
- Manager API
parameters:
- name: Authorization
in: header
type: string
required: true
description: token
responses:
success:
schema:
type: object
properties:
status:
type: string
example: success.
msg:
type: string
example: 统计结果分析成功
data:
type: object
properties:
sex_number:
type: object
properties:
manNumber:
type: integer
example: 1
womanNumber:
type: integer
example: 2
phase:
type: object
properties:
亚急性期:
type: integer
example: 1
急性期:
type: integer
example: 2
慢性期:
type: integer
example: 2
超急性期:
type: integer
example: 2
age:
type: object
properties:
"<20":
type: integer
example: 1
20-30:
type: integer
example: 2
30-40:
type: integer
example: 2
40-50:
type: integer
example: 2
50-60:
type: integer
example: 2
60-70:
type: integer
example: 2
70-80:
type: integer
example: 2
">80":
type: integer
example: 2
description: 成功
fail:
schema:
type: object
properties:
status:
type: string
example: fail.
msg:
type: string
example: patientsAnalyze出错
data:
type: string
example: error
description: 失败
"""
try:
# 男女比例
db_sex = db.session.query(func.count(Patient.id)).group_by(
Patient.sex).all()
manNumber = db_sex[0][0]
womanNumber = db_sex[1][0]
sex_number = {"manNumber": manNumber, "womanNumber": womanNumber}
# 疾病阶段
db_state = db.session.query(func.count(Patient.id)).group_by(
Patient.state).all()
subacute_phase = db_state[0][0] # 亚急性期
acute_phase = db_state[1][0] # 急性期
chronic_phase = db_state[2][0] # 慢性期
hyperacute_phase = db_state[3][0] # 超急性期
phase = {
"亚急性期": subacute_phase,
"急性期": acute_phase,
"慢性期": chronic_phase,
"超急性期": hyperacute_phase
}
# 年龄分布
age_1 = db.session.query(func.count(
Patient.id)).filter(Patient.age < 20).all()
age_2 = db.session.query(func.count(Patient.id)).filter(
and_(20 <= Patient.age, Patient.age < 30)).all()
age_3 = db.session.query(func.count(Patient.id)).filter(
and_(30 <= Patient.age, Patient.age < 40)).all()
age_4 = db.session.query(func.count(Patient.id)).filter(
and_(40 <= Patient.age, Patient.age < 50)).all()
age_5 = db.session.query(func.count(Patient.id)).filter(
and_(50 <= Patient.age, Patient.age < 60)).all()
age_6 = db.session.query(func.count(Patient.id)).filter(
and_(60 <= Patient.age, Patient.age < 70)).all()
age_7 = db.session.query(func.count(Patient.id)).filter(
and_(70 <= Patient.age, Patient.age < 80)).all()
age_8 = db.session.query(func.count(
Patient.id)).filter(80 <= Patient.age).all()
total = db.session.query(func.count(Patient.id)).all()
age = {
"<20": age_1[0][0],
"20-30": age_2[0][0],
"30-40": age_3[0][0],
"40-50": age_4[0][0],
"50-60": age_5[0][0],
"60-70": age_6[0][0],
"70-80": age_7[0][0],
">80": age_8[0][0],
"total": total[0][0]
}
response_object = {
'sex_number': sex_number,
'phase': phase,
"age": age
}
return successReturn(response_object, "patientsAnalyze: 统计结果分析成功")
except Exception as e:
return failReturn(format(e), "patientsAnalyze出错")
def update_role():
"""
更新用户权限
:return: json
---
tags:
- Manager API
parameters:
- name: body
in: body
required: true
schema:
id: 更新用户权限
required:
- userID
- role
properties:
userID:
type: integer
description: userID
role:
type: integer
description: role 1管理员 2主任医生 3医生
- name: Authorization
in: header
type: string
required: true
description: token
responses:
success:
schema:
type: object
properties:
status:
type: string
example: success.
msg:
type: string
example: 更新权限成功
data:
type: string
example: err
description: 成功
fail:
schema:
type: object
properties:
status:
type: string
example: fail.
msg:
type: string
example: 权限不足 or 用户不存在 or updateRole出错
data:
type: string
example: error
description: 失败
"""
try:
doctor = _get_current_user()
userID = request.get_json()['userID']
role = request.get_json()['role']
if doctor.userType > role:
return failReturn("", "updateRole; 权限不足")
user = User.query.filter_by(id=userID).first()
if not user:
return failReturn("", "updateRole； 用户不存在")
user.userType = role
db.session.commit()
return successReturn("", "updateRole: 更新权限成功")
except Exception as e:
return failReturn(format(e), "updateRole出错")
def user_detail():
"""
获取用户详细信息
:return: json
---
tags:
- Manager API
parameters:
- name: body
in: body
required: true
schema:
id: 获取用户详细信息
required:
- userID
properties:
userID:
type: integer
description: userID
- name: Authorization
in: header
type: string
required: true
description: token
responses:
success:
schema:
type: object
properties:
status:
type: string
example: success.
msg:
type: string
example: 获取用户详细信息成功
data:
type: object
properties:
userInfo:
type: object
properties:
name:
type: string
example: zj
realname:
type: string
example: zj
role:
type: string
example: 管理员
description: 成功
fail:
schema:
type: object
properties:
status:
type: string
example: fail.
msg:
type: string
example: userDetail出错 or 用户不存在
data:
type: string
example: error
description: 失败
"""
try:
userID = request.get_json()['userID']
user = User.query.filter_by(id=userID).first()
if not user:
return failReturn("", "userDetail: 用户不存在")
role = _userType(user)
response_object = {
'name': user.username,
'realname': user.realname,
'role': role
}
return successReturn(response_object, "userDetail； 获取用户详细信息成功")
except Exception as e:
return failReturn(format(e), "userDetail出错")
def user_info():
"""
获取用户列表
:return: json
---
tags:
- Manager API
parameters:
- name: Authorization
in: header
type: string
required: true
description: token
responses:
success:
schema:
type: object
properties:
status:
type: string
example: success.
msg:
type: string
example: 获取用户列表成功
data:
type: object
properties:
userInfo:
type: array
items:
type: object
properties:
id:
type: integer
example: 1
name:
type: string
example: zj
role:
type: string
example: 管理员
description: 成功
fail:
schema:
type: object
properties:
status:
type: string
example: fail.
msg:
type: string
example: userInfo出错 or 权限不足
data:
type: string
example: error
description: 失败
"""
try:
res = _statistics()
if not res:
return failReturn("", "userInfo: 权限不足")
response_object = {'userInfo': res}
return successReturn(response_object, "userInfo: 获取用户列表成功")
except Exception as e:
return failReturn(format(e), "userInfo出错")