feat: add bodyinfo field age
continuous-integration/drone/push Build is passing Details

This commit is contained in:
BryantHe 2023-09-01 15:59:19 +08:00
parent 5a10303133
commit 721a5f1aa8
2 changed files with 8 additions and 3 deletions

View File

@ -1,10 +1,11 @@
from apiflask import Schema from apiflask import Schema
from apiflask.fields import String, Float, URL, Boolean from apiflask.fields import String, Float, URL, Boolean, Integer
class UserInfoIn(Schema): class UserInfoIn(Schema):
nickname = String() nickname = String()
gender = String() gender = String()
age = Integer()
height = Float() height = Float()
is_hidden_weight = Boolean() is_hidden_weight = Boolean()
@ -17,6 +18,7 @@ class UserInfoOut(Schema):
user_id = String(attribute='_id') user_id = String(attribute='_id')
nickname = String() nickname = String()
gender = String() gender = String()
age = Integer()
avatar_id = String() avatar_id = String()
avatar_url = URL() avatar_url = URL()
height = Float(allow_nan=True) height = Float(allow_nan=True)

View File

@ -25,7 +25,8 @@ def set_user_info(json_data):
rpc.body_record.set_body_info(session['user_id'], rpc.body_record.set_body_info(session['user_id'],
height=json_data.get('height', None), height=json_data.get('height', None),
is_hidden_weight=json_data.get('is_hidden_weight', None), is_hidden_weight=json_data.get('is_hidden_weight', None),
gender=json_data.get('gender', None)) gender=json_data.get('gender', None),
age=json_data.get('age', None))
except Exception as e: except Exception as e:
raise UserInfoError(extra_data={'error_docs': str(e)}) raise UserInfoError(extra_data={'error_docs': str(e)})
@ -33,7 +34,7 @@ def set_user_info(json_data):
@user.get('/info') @user.get('/info')
@user.doc(summary='查询用户信息', description='查询用户信息。注gender 1 代表男性2 代表女性') @user.doc(summary='查询用户信息', description='查询用户信息。注gender 类型str 1 代表男性2 代表女性')
@user.input(UserIdIn, location='query') @user.input(UserIdIn, location='query')
@user.output(UserInfoOut) @user.output(UserInfoOut)
@login_required @login_required
@ -62,6 +63,7 @@ def get_user_info(query_data):
if body_info_dict: if body_info_dict:
result['height'] = body_info_dict.get('height', None) result['height'] = body_info_dict.get('height', None)
result['gender'] = body_info_dict.get('gender', None) result['gender'] = body_info_dict.get('gender', None)
result['age'] = body_info_dict.get('age', None)
if body_info_dict.get('is_hidden_weight', None): if body_info_dict.get('is_hidden_weight', None):
result['is_hidden_weight'] = True result['is_hidden_weight'] = True
else: else:
@ -70,5 +72,6 @@ def get_user_info(query_data):
result['height'] = None result['height'] = None
result['is_hidden_weight'] = False result['is_hidden_weight'] = False
result['gender'] = None result['gender'] = None
result['age'] = None
return result return result