def migu_register(): """用户注册(GET|POST) :uri: /migu/register_phone :param phone: 手机号 :param password: 密码 :param code: 验证码 :param sessionid: 短信sessionid :returns: {'user': object, 'ut': string} """ params = request.values.to_dict() phone = params.get('phone', None) code = params.get('code', None) sessionid = params.get('sessionid', None) password = params.get('password', None) if not phone or not code or not password or not sessionid: return error.InvalidArguments invalid_error = User.invalid_password(password) if invalid_error: return invalid_error # 用户中心注册 ret = Migu.center_register(phone, password, const.CENTER_ACCOUNT_PHONE, code, sessionid) if isinstance(ret, error.ApiError): return ret # 进行用户绑定 migu_uid = Migu.get_identityid(phone, password, const.CENTER_ACCOUNT_PHONE) if isinstance(migu_uid, error.ApiError): return migu_uid user = User.get_platform_user('migu', migu_uid) if not user: user = User.get_by_phone(phone) if user: info = dict(partner_migu={'id': migu_uid}, nickname=u'咪咕用户%s%s' % (migu_uid[-4:], random.randint(1000, 9999)), gender=random.randint(1, 2), name='$mg$%s%s' % (migu_uid[-4:], random.randint(1000, 9999))) user = user.update_model({'$set': info}) else: info = dict(phone=phone, nickname=u'咪咕用户%s%s' % (migu_uid[-4:], random.randint(1000, 9999)), gender=random.randint(1, 2), name='$mg$%s%s' % (migu_uid[-4:], random.randint(1000, 9999))) user = User.create_platform_user('migu', migu_uid, data=info) else: # 如果用户没有绑定手机并且手机号没有被绑定, 则自动进行手机号绑定 if not user.phone and not User.get_by_phone(phone): info = dict(phone=phone) user.update_model({'$set': info}) ut = User.gen_token(str(user._id)) return {'user': user.format(), 'ut': ut}
def login(): """用户登录 (GET|POST) :uri: /users/login :param name: 用户名 :param password: 密码 :param type: 登陆类型(name, phone,) :returns: {'user': object, 'ut': string} """ params = request.values login_type = params.get('type', 'name') name = params.get("name", None) password = params.get("password", None) if name is None or password is None or login_type not in ['name', 'phone']: return error.InvalidArguments user = User.login(name, password, login_type=login_type) if not user: return error.LoginFailed # 初始化用户任务 UserTask.create_and_init_user_tasks(str(user._id)) token = User.gen_token(str(user._id)) return {'user': user.format(), 'ut': token}
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}
def platform_login(): """第三方平台token登录 (GET|POST) :uri: /platform/users/login :param platform: 平台标识{'csdk':付费SDK, 'weixin':微信, 'qq':QQ, 'migu':咪咕} :param token: 用户平台token :returns: {'user': object, 'ut': string} """ params = request.values platform = params.get('platform', None) token = params.get("token", None) if not token or platform not in const.PARTNER: return error.InvalidArguments data = Migu.token_validate(token) if isinstance(data, error.ApiError): return data openid = data.get('msisdn', None) migu_uid = data.get('identityID', None) passid = data.get('passID', None) if isinstance(migu_uid, error.ApiError): return migu_uid if platform == 'migu': # 进行用户绑定 user = User.get_platform_user('migu', migu_uid) if not user: user = User.get_by_phone(openid) if user: info = dict(partner_migu={ 'id': migu_uid, 'passid': passid }, name='$mg$%s%s' % (migu_uid[-4:], random.randint(1000, 9999))) user = user.update_model({'$set': info}) else: info = dict(phone=openid, nickname=u'咪咕用户%s%s' % (migu_uid[-4:], random.randint(1000, 9999)), name='$mg$%s%s' % (migu_uid[-4:], random.randint(1000, 9999)), partner_migu={ 'id': migu_uid, 'passid': passid }) user = User.create_platform_user('migu', migu_uid, data=info) else: info = {} # 如果用户没有绑定手机并且手机号没有被绑定, 则自动进行手机号绑定 if not user.phone and not User.get_by_phone(openid): info = dict(phone=openid) # 如果用户未绑定passid,则自动进行passid绑定 if passid not in user.partner_migu: info.update( dict(partner_migu={ 'id': migu_uid, 'passid': passid })) if info: user = user.update_model({'$set': info}) # 同步咪咕用户名密码 if user: User.change_pwd(user, token) else: # 判断平台用户是否已经用户系统中存在 user = User.get_platform_user(platform, openid) first_login = not user info = {} if platform == 'csdk': info = ChargeSDK(token).get_open_info() elif platform == 'weixin': info = WeiXin(token, openid).get_open_info(first_login) elif platform == 'qq': info = QQ(token, openid).get_open_info(first_login) if not info: return error.LoginFailed # 如果平台用户在用户系统中不存在, 则创建, 通过平台ID(openid)进行关联 if first_login: open_id = info.pop('openid') user = User.create_platform_user(platform, open_id, data=info) # 给咪咕平台发送请求进行咪咕账号注册并登录绑定 try: if platform in ['qq', 'weixin'] and user: password = '******' % (openid[-4:]) ret = Migu.center_register(open_id, password, const.CENTER_ACCOUNT_INDIV) if not isinstance(ret, error.ApiError): openid = Migu.get_identityid( open_id, password, const.CENTER_ACCOUNT_INDIV) if not isinstance(openid, error.ApiError): if not User.get_platform_user('migu', openid): info = {'partner_migu': {'id': openid}} user.update_model({'$set': info}) except: pass if not user: return error.LoginFailed ut = User.gen_token(str(user._id)) return {'user': user.format(include_fields=['passid']), 'ut': ut}