def register_phone(): """用户手机注册 (GET|POST) :uri: /users/register_phone :param phone: 手机号 :param password: 密码 :param nickname: 昵称 :param code: 短信验证码 :param gender: 性别(可选)(1:男, 2:女) :returns: {'user': object, 'ut': string} """ params = request.values phone = params.get('phone', None) code = params.get('code', None) password = params.get("password", None) nickname = params.get("nickname", None) gender = params.get("gender", 0) if not phone or not code or not password or not nickname: return error.InvalidArguments invalid_error = User.invalid_password(password) if invalid_error: return invalid_error invalid_error = User.invalid_nickname(nickname) if invalid_error: return invalid_error if User.get_by_phone(phone): return error.UserExists if not SMS.verify_code(phone, code): return error.VerifyCodeFailed user = User.init() name = '$mb$%s%s' % (phone[-4:], random.randint(1000, 9999)) user.name = name user.phone = phone user.nickname = nickname user.gender = gender salt = os.urandom(const.PWD_HASH_LEN) pwd = User.gen_pwd_hash(password, salt) user._salt = Binary(salt) user._password = Binary(pwd) uid = user.create_model() new_user = User.get_one(uid) token = User.gen_token(str(uid)) return {'user': new_user.format(), 'ut': token}
def register(): """用户注册 (GET|POST) :uri: /users/register :param name: 用户名 :param password: 密码 :param nickname: 昵称 :returns: {'user': object, 'ut': string} """ params = request.values.to_dict() name = params.get("name", None) # delete password from data so that we don't save it to mongo password = str(params.pop("password", None)) nickname = params.get('nickname', None) if not name or not password or not nickname: return error.InvalidArguments invalid_error = User.invalid_password(password) if invalid_error: return invalid_error invalid_error = User.invalid_nickname(nickname) if invalid_error: return invalid_error if User.get_by_name(name): return error.UserExists user = User.init() user.update(params) salt = os.urandom(const.PWD_HASH_LEN) pwd = User.gen_pwd_hash(password, salt) user._salt = Binary(salt) user._password = Binary(pwd) uid = user.create_model() new_user = User.get_one(uid) # 初始化用户任务 UserTask.create_and_init_user_tasks(str(new_user._id)) token = User.gen_token(str(uid)) return {'user': new_user.format(), 'ut': token}