def queryUser(): _openid = request.json.get('openid') _account = request.json.get('account') user_dao = UserDao() res = user_dao.query_by_openid_account(_account, _openid) size = len(res) if size > 0: return json.dumps(return_success(UserDao.to_dict(res))) else: return json.dumps(return_unsuccess('Error: No such user'))
def login(): # token登陆 if request.method == 'GET': return decode_token() _json = request.json login_type = _json.get('type') account = _json.get('account') password = _json.get('passwd') web = _json.get('web') # 生成token login_time = int(time.time()) token = Auth.create_jwt({'account': account, 'login_time': login_time}) # 账号密码登陆 if login_type == 0: store_in = base64.b64decode(password) if not web: store_in = binascii.hexlify(store_in) strpass = str(store_in, 'utf-8') print(strpass) user_dao = UserDao() res = user_dao.query_check_login(account, strpass) size = len(res) if size == 1: resp = return_success(UserDao.to_dict(res)) resp['token'] = token return jsonify(resp) else: return jsonify(return_unsuccess('账号或密码错误')) # 验证码登陆 elif login_type == 1: true_veri = redis_store.get('veri' + account) if not true_veri: return jsonify(return_unsuccess("验证码过期")) elif password != true_veri: return jsonify(return_unsuccess("验证码错误")) res = json.loads(check_account()) suc = res.get("success") if not suc: print(res) redis_store.delete('veri' + account) user = UserDao().query_by_account(account) resp = return_success(UserDao.to_dict(user)) resp['token'] = token else: resp = return_unsuccess('Error: No such user') return jsonify(resp) # openid登陆 elif login_type == 2: openid = _json.get("openid") user_dao = UserDao() res = user_dao.query_by_openid_account(account, openid) size = len(res) if size == 1: resp = return_success(UserDao.to_dict(res)) token = Auth.create_jwt({ 'account': res[0][0], 'login_time': login_time }) resp['token'] = token return jsonify(resp) else: return jsonify(return_unsuccess('Error: No such user')) else: return jsonify(return_unsuccess('Error: Wrong Login Method'))