def new_login(self, arguments): """ @apiVersion 1.0.0 @api {post} /api/V1/new_login 用户登录(new) @apiName new_login @apiGroup User @apiParam {int} mobile 用户手机号作为登录用户名 @apiParam {int} code 用户密码 @apiSuccess {String} message y验证结果 """ mobile = arguments['mobile'] verify_code = arguments['code'] user = User.find_one({'mobile': mobile}) if not user: return self.fail(message=u'该用户不存在') if verify_code != int(session[str(mobile)]): return self.fail(message=u'验证码不正确') if user.get('company_id') and len(user['company_id']) != 0: user = load_user(user['uid']) login_user(user, remember=True) return self.done(User.user_find_one({'mobile': mobile})) else: return self.fail(u"用户没有公司,请先创建公司")
def disable_register(self, arguments): """ @apiVersion 1.0.0 @api {post} /api/V1/user/disable_register 被禁用户的注册接口 @apiName disable_register @apiGroup User @apiParam {int} mobile 用户手机号 @apiParam {int} company_name 公司名称 """ company_name = arguments['company_name'] mobile = arguments['mobile'] user = User.find_one({'mobile': mobile}) company = Company.find_one({'company_name': company_name}) if company: return self.fail(message=u"该公司已存在") else: params = {'admin': user['uid']} company = Company.create_company(company_name, **params) user['company_id'].append(company['company_id']) result = User.update_one( {'_id': user['_id']}, {'$set': { 'company_id': user['company_id'] }}) if result.raw_result['ok'] == 1: user = load_user(user['uid']) login_user(user, remember=True) return self.done(u"登录成功") else: return self.fail(u"添加失败")
def create_user(self, arguments): user_exist = User.find_one(mobile=arguments['mobile']) if user_exist: return self.is_fail('当前手机号码已被注册') if arguments['verify_code'] != session[str(arguments['mobile'])]: return self.is_fail('验证码不正确') user = User(arguments['mobile'], arguments['password1']) result = user.save() return self.is_done(result)
def disable_user_invited(self, arguments): """ @apiVersion 1.0.0 @api {post} /api/V1/user/disable_user_invited 被禁用户接受邀请 @apiName disable_user_invited @apiGroup User @apiParam {int} mobile 用户手机号 @apiParam {int} company_id 公司id """ mobile = arguments['mobile'] user_result = User.find_one({'mobile': int(mobile)}) company_list = ast.literal_eval(arguments['company_id']) if not user_result: return self.fail(u"该手机号码不存在") if not company_list: return self.fail(u"公司不存在") for i in company_list: company = Company.find_one({'company_id': int(i)}) invite_list = [] for j in company['invite']: if str(mobile) not in j: invite_list.append(j) else: permission = j[str(mobile)].get('permission') if not user_result['permission']: user_result['permission'] = {} user_result['permission'][str(i)] = permission if int(i) not in user_result['company_id']: user_result['company_id'].append(int(i)) User.update_one({'uid': user_result['uid']}, { '$set': { 'company_id': user_result['company_id'], 'permission': user_result['permission'] } }) use_list = company.get('use') if company.get('use') else [] use_list.append(user_result['uid']) Company.update_one( {'_id': company['_id']}, {'$set': { 'invite': invite_list, 'use': use_list }}) user = load_user(user_result['uid']) login_user(user, remember=True) return self.done()
def user_add_company(self, arguments): """ @apiVersion 1.0.0 @api {post} /api/V1/user/user_add_company 添加新的公司 @apiName user_add_company @apiGroup User @apiParam {int} mobile 用户手机号 @apiParam {int} company_id 公司id """ mobile = arguments['mobile'] company_id = arguments['company_id'] user_result = User.find_one({'mobile': int(mobile)}) company_result = Company.find_one({'company_id': int(company_id)}) if not user_result: return self.fail(u"该手机号码不存在") if not company_result: return self.fail(u"公司不存在") if int(company_id) not in user_result['company_id']: user_result['company_id'].append(int(company_id)) invite_list = [] for j in company_result['invite']: if str(mobile) not in j: invite_list.append(j) else: permission = j[str(mobile)].get('permission') if not user_result['permission']: user_result['permission'] = {} user_result['permission'][str(company_id)] = permission User.update_one({'uid': user_result['uid']}, { '$set': { 'company_id': user_result['company_id'], 'permission': user_result['permission'] } }) if not company_result.get('use'): company_result['use'] = [] company_result['use'].append(user_result['uid']) Company.update_one( {'_id': company_result['_id']}, {'$set': { 'invite': invite_list, 'use': company_result['use'] }}) return self.done()
def add_company(self, arguments): """ @apiVersion 1.0.0 @api {post} /api/V1/user/add_company 用户注册并添加新的公司 @apiName add_company @apiGroup User @apiParam {int} mobile 用户手机号 @apiParam {int} company_id 公司id """ mobile = arguments['mobile'] if User.find_one({'mobile': mobile}): return self.fail(message=u"该手机号码已注册") password = str(mobile) company_list = ast.literal_eval(arguments['company_id']) permission = {} for i in company_list: company = Company.find_one({'company_id': int(i)}) invite_list = [] for j in company['invite']: if str(mobile) not in j: invite_list.append(j) else: permission[str(i)] = j[str(mobile)].get('permission') company['invite'] = invite_list Company.update_one({'_id': company['_id']}, {'$set': company}) user = User.create_user(int(mobile), password, company_list) user.update_one({'_id': user['_id']}, {'$set': { 'permission': permission }}) user = load_user(user['uid']) for i in company_list: company = Company.find_one({'company_id': int(i)}) use_list = company.get('use') if company.get('use') else [] use_list.append(user['uid']) company['use'] = use_list Company.update_one({'_id': company['_id']}, {'$set': company}) login_user(user, remember=True) return self.done()
def new_register(self, arguments): """ @apiVersion 1.0.0 @api {post} /api/V1/user/new_register 新的注册接口 @apiName new_register @apiGroup User @apiParam {int} mobile 用户手机号 @apiParam {int} company_name 公司名称 """ mobile = arguments['mobile'] if User.find_one({'mobile': mobile}): return self.fail(message=u"该手机号码已注册") password = str(mobile) company_name = arguments['company_name'] company = Company.find_one({'company_name': company_name}) if company: return self.fail(message=u"该公司已存在") else: new_company = Company.create_company(company_name) company_id = new_company['company_id'] for item in ['mobile', 'company_name']: del arguments[item] user = User.create_user(int(mobile), password, [company_id], **arguments) Company.update_one({'_id': new_company['_id']}, {'$set': { 'admin': user['uid'] }}) permission = {str(company_id): {'shouzu': '允许', 'zhaozu': '允许'}} User.update_one({'uid': user['uid']}, {'$set': { 'permission': permission }}) user = load_user(user['uid']) login_user(user, remember=True) session["current_company"] = company_id session["current_user_company"] = Company.company_find_one( {'company_id': int(company_id)}) return self.done("success")
def register(self, arguments): mobile = arguments['mobile'] if User.find_one({'mobile': mobile}): return self.fail(message=u"该手机号码已注册") password = arguments['password'] company_name = arguments['company_name'] verify_code = arguments['verify_code'] if verify_code != int(session[str(mobile)]): return self.fail(message=u'验证码不正确') company = Company.find_one({'company_name': company_name}) if company: company_id = company['company_id'] else: new_company = Company.create_company(company_name) company_id = new_company['company_id'] for item in ['mobile', 'verify_code', 'password', 'company_name']: del arguments[item] user = User.create_user(int(mobile), password, company_id, **arguments) user = load_user(user['uid']) login_user(user, remember=True) del user['password'] return self.done(data=user)
def edit_mobile(self, arguments): """ @apiVersion 1.0.0 @api {post} /api/V1/user/edit_mobile 更新用户手机号码 @apiName edit_mobile @apiGroup User @apiParam {str} name 公司名称 """ mobile = arguments['mobile'] verify_code = arguments['code'] if verify_code != int(session[str(mobile)]): return self.fail(message=u'验证码不正确') if User.find_one({'mobile': mobile}): return self.fail(u"此手机已经被注册过了") result = User.update_one({'_id': current_user['_id']}, {'$set': { 'mobile': mobile }}) if result.raw_result['ok'] == 1: return self.done("success") else: return self.fail(u"更新失败")
def exist_user(self, arguments): mobile = arguments['mobile'] if User.find_one({'mobile': mobile}): return self.fail(u"用户已存在") else: return self.done(u"用户不存在")