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

This commit is contained in:
BryantHe 2023-08-23 10:57:14 +08:00
parent d039cc69e1
commit 0ef80c8816
2 changed files with 12 additions and 3 deletions

View File

@ -1,14 +1,17 @@
from apiflask import Schema from apiflask import Schema
from apiflask.fields import String, Float, URL from apiflask.fields import String, Float, URL, Boolean
class UserInfoIn(Schema): class UserInfoIn(Schema):
nickname = String() nickname = String()
height = Float() height = Float()
is_hidden_weight = Boolean()
class UserInfoOut(Schema): class UserInfoOut(Schema):
user_id = String(attribute='_id')
nickname = String() nickname = String()
avatar_id = String() avatar_id = String()
avatar_url = URL() avatar_url = URL()
height = Float(allow_nan=True) height = Float(allow_nan=True)
is_hidden_weight = Boolean()

View File

@ -23,7 +23,8 @@ def set_user_info(json_data):
try: try:
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),)
except Exception as e: except Exception as e:
raise UserInfoError(extra_data={'error_docs': str(e)}) raise UserInfoError(extra_data={'error_docs': str(e)})
@ -57,7 +58,12 @@ def get_user_info():
result.update(user_dict) result.update(user_dict)
if body_info_dict: if body_info_dict:
result['height'] = body_info_dict.get('height', None) result['height'] = body_info_dict.get('height', None)
if body_info_dict.get('is_hidden_weight', None):
result['is_hidden_weight'] = True
else:
result['is_hidden_weight'] = False
else: else:
result['height'] = None result['height'] = None
result['is_hidden_weight'] = False
return result return result