def get_list(): token = request.headers.get('token') user_id = user_id_by_token(token) data_list = search_data('list', {'user_id': user_id}, True) body = [] if data_list == None else data_list body = list(reversed(body)) return api_factory(body)
def change_log(log_id): id_ = unquote(log_id) api_param = request.get_json() api_param = api_param if api_param != None else {} db_value = search_data('log', {'id': id_}) keys = db_value.keys() if isinstance(db_value, dict) else [] for key in keys: if key in api_param.keys(): db_value[key] = api_param[key] update_data('log', db_value, {'id': id_}) return api_factory({'msg': '修改成功'})
def login(): api_param = request.get_json() user = api_param.get('user') pwd = api_param.get('pwd') db_value = search_data('user', {'user': user}) if db_value == None or db_value['pwd'] != pwd: return err_factory('用户名或者密码错误', None, 403) else: token = rand_string() set_token(user, token) print('更新token', user, token) return api_factory(token)
def register(): api_param = request.get_json() user = api_param.get('user') pwd = api_param.get('pwd') db_value = search_data('user', {'user': user}) if db_value == None: insert_data('user', {'id': rand_string(), 'user': user, 'pwd': pwd}) token = rand_string() set_token(user, token) create_menu(user) return api_factory(token) else: return err_factory('该用户已存在', None, 403)
def create_menu(user): """ 创建用户的基础清单 """ keys = [] values = [] user_id = search_data('user', {"user": user})['id'] with open('./base_menu.json', 'r', encoding='utf-8') as f: read_file = json.loads(f.read()) for menu in read_file: menu['id'] = rand_string() menu['user_id'] = user_id list_menu = list(menu.values()) values.append(tuple(list_menu)) keys = list(read_file[0].keys()) insert_batch('list', keys, values)
def get_log(list_id): id_ = unquote(list_id) data = search_data('log', {'list_id': id_}, True) result = data if isinstance(data, list) else [] return api_factory(result)