Exemplo n.º 1
0
    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"用户没有公司,请先创建公司")
Exemplo n.º 2
0
    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"添加失败")
Exemplo n.º 3
0
 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)
Exemplo n.º 4
0
    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()
Exemplo n.º 5
0
    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()
Exemplo n.º 6
0
    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()
Exemplo n.º 7
0
    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")
Exemplo n.º 8
0
 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)
Exemplo n.º 9
0
    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"更新失败")
Exemplo n.º 10
0
 def exist_user(self, arguments):
     mobile = arguments['mobile']
     if User.find_one({'mobile': mobile}):
         return self.fail(u"用户已存在")
     else:
         return self.done(u"用户不存在")