Пример #1
0
 def authenticate(email, password):
     """
     用户登录,登录成功返回token,写将登录时间写入数据库;登录失败返回失败原因
     :param password:
     :return: json
     """
     userInfo = Users().getOne(
         {Users.email == email}, 'id desc',
         ('email', 'password', 'id', 'nick_name', 'head_img'))
     if (userInfo is None):
         return Utils.formatError(Code.BAD_REQUEST, '找不到用户')
     else:
         if (Users.check_password(userInfo['password'], password)):
             updated_at = int(time.time())
             Users().edit({'updated_at': updated_at},
                          {Users.email == email})
             token = UsersAuthJWT.encode_auth_token(userInfo['id'],
                                                    updated_at)
             userInfo.pop('password')  #删除密码
             return Utils.formatBody(
                 {
                     'token': token.decode(),
                     'user': userInfo
                 }, '登陆成功')
         else:
             return Utils.formatError(Code.BAD_REQUEST, '密码不正确')
Пример #2
0
 def add(params,user_info):
     ''' 添加通讯录 '''
     if params['focused_user_id'] == user_info['data']['id']:
         return Utils.formatError(Code.BAD_REQUEST,msg='无法添加自己为新朋友')
     filters = {
         AddressBook.focused_user_id == params['focused_user_id'],
         AddressBook.be_focused_user_id == user_info['data']['id']
     }
     addressBookData = AddressBook().getOne(filters)
     if(addressBookData == None):
         # 加入房间号
         room_uuid = Utils.unique_id()
         # 建立通讯录关系
         status = AddressBook.addRoomAndAddressBook(
             room_uuid, params['focused_user_id'], user_info['data']['id'])
         if status == False:
             return Utils.formatError(Code.BAD_REQUEST,msg='添加失败')
     
         #回复被添加用户
         socketio.emit('beg', Utils.formatBody({
             'action':'beg_add_success',
             'focused_user_id' : user_info['data']['id']
         }), namespace='/room', room='@broadcast.'+str(params['focused_user_id']))
         #添加后同步房间
         addressBookData = AddressBook.get(room_uuid)
         for item in addressBookData:
             roomList = AddressBook.getRoomList(item.be_focused_user_id)['data']
             socketio.emit('room',Utils.formatBody(roomList), namespace="/room",room='@broadcast.'+str(item.be_focused_user_id))   
         return Utils.formatBody({},msg='添加成功')
     return Utils.formatError(Code.BAD_REQUEST,msg='已添加')
Пример #3
0
 def add(self, data=[]):
     #添加
     if 'update_time' in data.keys():
         data['update_time'] = int(time.time())
     #判断用户名
     if 'name' in data.keys():
         filters = {Admin.delete_time == 0, Admin.name == data['name']}
         adminData = Admin().getOne(filters, 'add_time desc')
         if adminData != None:
             return Utils.formatError(CONST['CODE']['ERROR']['value'],
                                      '用户名已经存在')
     #判断手机号码
     if 'mobile' in data.keys():
         filters = {Admin.delete_time == 0, Admin.mobile == data['mobile']}
         admin = Admin().getOne(filters, 'add_time desc')
         if admin:
             return Utils.formatError(CONST['CODE']['ERROR']['value'],
                                      '手机号码已经存在')
     #判断邮箱
     if 'email' in data.keys():
         filters = {Admin.delete_time == 0, Admin.email == data['email']}
         admin = Admin().getOne(filters, Admin.add_time.desc())
         if admin:
             return Utils.formatError(CONST['CODE']['ERROR']['value'],
                                      '邮箱已经存在')
     #生成一个md5对象
     m1 = hashlib.md5()
     #使用md5对象里的update方法md5转换
     m1.update(data['pwd'].encode("utf-8"))
     data['pwd'] = m1.hexdigest()
     res = Admin().add(data)
     if res:
         return Utils.formatBody(msg='添加成功')
     return Utils.formatError(CONST['CODE']['ERROR']['value'], '添加失败')
Пример #4
0
    def adminChat(message: dict) -> dict:
        admin_user_info = UsersAuthJWT().adminIdentify(
            message['Authorization'])
        if isinstance(admin_user_info, str):
            return Utils.formatError(CONST['CODE']['ERROR_AUTH_CHECK_TOKEN_FAIL']['value'], admin_user_info)
        # 整合数据信息
        filters = {
            Admin.id == admin_user_info['data']['id'],
        }
        admin_user_info = Admin().getOne(filters)
        default_img_data = Config().getOne(
            {Config.type == 'img', Config.code == 'default.img', Config.status == 1})
        if default_img_data == None:
            default_img = 'static/img/about/python.jpg'
        else:
            default_img = default_img_data['config']

        admin_user_info['nick_name'] = '系统管理-'+admin_user_info['nick_name']
        admin_user_info['head_img'] = default_img  # 这里后期改成配置的
        # 使用0作为系统id
        msg = message['data']['msg']
        room_uuid = message['data']['room_uuid']
        Type = message['data']['type']
        room_data = Room.get(room_uuid)
        if room_data == None:
            return Utils.formatError(CONST['CODE']['ROOM_NO_EXIST']['value'], "房间不存在")
        room_type = room_data.type
        created_at = int(time.time())
        save_action = message['data']['save_action']
        return ChatService.sendChatMessage(msg, room_uuid, Type, room_data, room_type, created_at, save_action, admin_user_info, 1)
Пример #5
0
 def register(self, params):
     userData = Users().getOne({Users.email == params['email']})
     if (userData == None):
         #昵称首字母
         isChinese = re.compile(u"[\u4e00-\u9fa5]+")
         isEnglish = re.compile('[a-zA-Z]')
         nickNameFirstWord = params['nickName'][0]
         if isChinese.search(nickNameFirstWord):
             first_word = Pinyin().get_initial(nickNameFirstWord)
         elif isEnglish.search(nickNameFirstWord):
             first_word = nickNameFirstWord.upper()
         else:
             first_word = '#'
         data = {
             'email': params['email'],
             'password': Users.set_password(params['password']),
             'nick_name': params['nickName'],
             'head_img': params['headImg'],
             'first_word': first_word,
             'created_at': time.time(),
             'updated_at': time.time()
         }
         user = Users().add(data)
         if user == False:
             return Utils.formatError(Code.BAD_REQUEST, '注册失败')
         else:
             result = UsersAuthJWT.authenticate(params['email'],
                                                params['password'])
             return result
         return Utils.formatError(Code.BAD_REQUEST, '注册失败')
     return Utils.formatError(Code.BAD_REQUEST, '账号已注册')
 def beg(params, user_info):
     if user_info['data']['id'] == params['be_focused_user_id']:
         return Utils.formatError(CONST['CODE']['BAD_REQUEST']['value'],
                                  '无法添加自己为新朋友')
     addressBookdata = AddressBook.getAdddressBookByFocusedUserId(
         user_info['data']['id'], params['be_focused_user_id'])
     if addressBookdata != None:
         return Utils.formatError(CONST['CODE']['BAD_REQUEST']['value'],
                                  '已添加')
     msg_uuid = Utils.unique_id()
     userData = Users().getOne({Users.id == user_info['data']['id']})
     data = {
         'data': userData,
         'action': 'beg_add',
         'msg_uuid': msg_uuid,
         'be_focused_user_id': params['be_focused_user_id'],
         'focused_user_id': user_info['data']['id']
     }
     socketio.emit('beg',
                   Utils.formatBody(data),
                   namespace='/api',
                   room='@broadcast.' +
                   str(params['be_focused_user_id']))  #,callback=success)
     #只缓存最新的一条
     cache.set('@broadcast.beg' + str(params['be_focused_user_id']), data)
     return Utils.formatBody({}, msg='发送成功')
Пример #7
0
 def inner_wrappar(*args, **kwargs):
     if 'Authorization' in args[0].keys():
         result = UsersAuthJWT().adminIdentify(args[0]['Authorization'])
     else:
         return Utils.formatError(CONST['CODE']['ERROR_AUTH_CHECK_TOKEN_FAIL']['value'], '令牌失效')
     kwargs['user_info'] = result
     if isinstance(result, str):
         # socketio.emit(name,Utils.formatError(CONST['CODE']['ERROR']['value']_AUTH_CHECK_TOKEN_FAIL, '令牌失效'), room='@api.'+str(request.sid))
         return Utils.formatError(CONST['CODE']['ERROR_AUTH_CHECK_TOKEN_FAIL']['value'], '令牌失效')
     res = func(*args, **kwargs)
     return res
Пример #8
0
    def edit(self, data, filters = {}):
    
        if 'update_time' in data.keys():
            data['update_time'] = time.time()
        adminData = Admin().getOne(filters, 'add_time desc')
        if adminData == None:
            return Utils.formatError(Code.ERROR, '用户不存在')
        #判断用户名
        if 'name' in data.keys():
            filters = {
                Admin.delete_time == 0,
                Admin.name        == data['name'],
                Admin.id          != adminData['id']
            }
            admin = Admin().getOne(filters, 'add_time desc')
            if admin != None:
                return Utils.formatError(Code.ERROR, '用户名已经存在')


        
        #判断手机号码
        if 'mobile' in data.keys():
            filters = {
                Admin.delete_time == 0,
                Admin.mobile      == data['mobile'],
                Admin.id          != adminData['id']
            }
            admin = Admin().getOne(filters, 'add_time desc')
            if admin != None:
                return Utils.formatError(Code.ERROR, '手机号码已经存在')
        
        #判断邮箱
        if 'email' in data.keys():
            filters = {
                Admin.delete_time == 0,
                Admin.email       == data['email'],
                Admin.id          != adminData['id']
            }
            admin = Admin().getOne(filters, Admin.add_time.desc())
            if admin != None:
                return Utils.formatError(Code.ERROR, '邮箱已经存在')
        
        if 'pwd' in data.keys():
            #生成一个md5对象
            m1 = hashlib.md5()
            #使用md5对象里的update方法md5转换
            m1.update(data['pwd'].encode("utf-8"))
            data['pwd'] = m1.hexdigest()      
        res = Admin().edit(data, filters)
        if res==True:
            return Utils.formatBody(msg='编辑成功')
        return Utils.formatError(Code.ERROR, '编辑失败', 0)
Пример #9
0
 def login(self, data):
     name = data['name']
     #生成一个md5对象
     m1 = hashlib.md5()
     #使用md5对象里的update方法md5转换
     m1.update(data['pwd'].encode("utf-8"))
     pwd = m1.hexdigest()
     #用户名/邮箱/手机登录
     filters = {
         Admin.name == name, Admin.pwd == pwd, Admin.delete_time == 0
     }
     admin = Admin().getOne(filters, 'add_time desc')
     if admin == None:
         filters = {
             Admin.email == name, Admin.pwd == pwd, Admin.delete_time == 0
         }
         admin = Admin().getOne(filters, 'add_time desc')
         if admin == None:
             filters = {
                 Admin.mobile == name, Admin.pwd == pwd,
                 Admin.delete_time == 0
             }
             admin = Admin().getOne(filters, 'add_time desc')
     if admin != None:
         #更新登录时间
         filters = {Admin.id == admin['id']}
         update_time = int(time.time())
         Admin().edit({Admin.update_time: update_time}, filters)
         token = UsersAuthJWT.encode_auth_token(admin['id'], update_time)
         return Utils.formatBody({'token': token, 'user': admin})
     return Utils.formatError(CONST['CODE']['ERROR']['value'])
Пример #10
0
 def inner_wrappar(*args, **kwargs):
     result = UsersAuthJWT().adminIdentify(request.headers.get('Authorization'))
     kwargs['user_info'] = result
     if isinstance(result, str):
         return make_response(jsonify(Utils.formatError(CONST['CODE']['ERROR_AUTH_CHECK_TOKEN_FAIL']['value'], result)))
     res = func(*args, **kwargs)
     return res
Пример #11
0
 def inner_wrappar(*args, **kwargs):
     result = UsersAuthJWT().identify(args[0]['Authorization'])
     kwargs['user_info'] = result
     if isinstance(result, str):
         return Utils.formatError(
             Code.ERROR_AUTH_CHECK_TOKEN_FAIL, '令牌失效'
         )  #socketio.emit(name,Utils.formatError(Code.ERROR_AUTH_CHECK_TOKEN_FAIL, '令牌失效'), room='@api.'+str(request.sid))
     res = func(*args, **kwargs)
     return res
Пример #12
0
 def create(params, user_info):
     # 加入房间号
     data = ChatService().groupChatCreate(params, user_info)
     if data:
         #添加后同步房间
         user_room_relation_data = Utils.db_l_to_d(UserRoomRelation.get(data['room_uuid']))
         for item in user_room_relation_data:
             roomList = UserRoomRelation.getRoomList(item['user_id'])
             socketio.emit('groupRoom', Utils.formatBody(roomList), namespace='/api', room='@broadcast.'+str(item['user_id']))
         return Utils.formatBody(data, msg='创建成功')
     return Utils.formatError(CONST['CODE']['BAD_REQUEST']['value'],msg='创建失败')
Пример #13
0
 def create(user_info, params):
     # 加入房间号
     data = ChatService().groupChatCreate(user_info, params)
     if data:
         #添加后同步房间
         user_room_relation_data = Utils.db_l_to_d(UserRoomRelation.get(data['room_uuid']))
         for item in user_room_relation_data:
             roomList = UserRoomRelation.getRoomList(item['user_id'])['data']
             socketio.emit('groupRoom', Utils.formatBody(roomList), namespace='/room', room='@broadcast.'+str(item['user_id']))
         return Utils.formatBody(data, msg='创建成功')
     return Utils.formatError(Code.BAD_REQUEST,msg='创建失败')
Пример #14
0
 def register(params):
     userData = Users().getOne({Users.email == params['email']})
     if (userData == None):
         #昵称首字母
         isChinese = re.compile(u"[\u4e00-\u9fa5]+")
         isEnglish = re.compile('[a-zA-Z]')
         nickNameFirstWord = params['nickName'][0]
         nowTime = time.time()
         if isChinese.search(nickNameFirstWord):
             first_word = Pinyin().get_initial(nickNameFirstWord)
         elif isEnglish.search(nickNameFirstWord):
             first_word = nickNameFirstWord.upper()
         else:
             first_word = '#'
         data = {
             'email': params['email'],
             'password': Users.set_password(params['password']),
             'nick_name': params['nickName'],
             'head_img': params['headImg'],
             'first_word': first_word,
             'created_at': nowTime,
             'updated_at': nowTime
         }
         user = Users().add(data)
         if user == False:
             return Utils.formatError(CONST['CODE']['BAD_REQUEST']['value'],
                                      '注册失败')
         else:
             result = UsersAuthJWT.authenticate(params['email'],
                                                params['password'])
             # 发送延时推广进群广告
             invite = Invite()
             invite.setAction("invite")
             invite.setId(result['data']['user']['id'])
             # 延时2分钟推送
             delayQueue.product(invite.__dict__, 120)
             return result
     return Utils.formatError(CONST['CODE']['BAD_REQUEST']['value'],
                              '账号已注册')
Пример #15
0
 def adminCreateRoom(message):
     admin_user_info = UsersAuthJWT().adminIdentify(
         message['Authorization'])
     if isinstance(admin_user_info, str):
         return Utils.formatError(CONST['CODE']['ERROR_AUTH_CHECK_TOKEN_FAIL']['value'], admin_user_info)
     filters = {
         Admin.id == admin_user_info['data']['id'],
     }
     admin_user_info = Admin().getOne(filters)
     filters = {
         AddressBook.be_focused_user_id == message['user_id'],
         AddressBook.focused_user_id == admin_user_info['id'],
         AddressBook.type == 1
     }
     addressBookInfo = AddressBook().getOne(filters)
     if addressBookInfo == None:
         room_uuid = Utils.unique_id()
         # 建立通讯录关系
         status = AddressBook.adminAddRoomAndAddressBook(
             room_uuid, admin_user_info, message['user_id'])
         if status == False:
             return Utils.formatError(CONST['CODE']['BAD_REQUEST']['value'], msg='添加失败')
         # 添加后同步房间
         addressBookData = AddressBook.get(room_uuid)
         for item in addressBookData:
             roomList = AddressBook.getRoomList(
                 item.be_focused_user_id)['list']
             if item['type'] == CONST['ADDRESSBOOK']['ADMIN']['value']:
                 socketio.emit('room', Utils.formatBody(
                     roomList), namespace="/api", room='@broadcast.'+str(item.be_focused_user_id))
             else:
                 socketio.emit('room', Utils.formatBody(
                     roomList), namespace="/api", room='@broadcast.'+str(item.be_focused_user_id))
     else:
         room_uuid = addressBookInfo['room_uuid']
     return Utils.formatBody({'room_uuid': room_uuid})
Пример #16
0
 def chat(message, user_info):
     """
         @param  dict message
         @param  dict user_info
         @return dict 
     """
     msg = message['data']['msg']
     room_uuid = message['data']['room_uuid']
     Type = message['data']['type']
     room_data = Room.get(room_uuid)
     if room_data == None:
         return Utils.formatError(CONST['CODE']['ROOM_NO_EXIST']['value'], "房间不存在")
     room_type = room_data.type
     created_at = message['data']['created_at']
     save_action = message['data']['save_action']
     user_data = Users().getOne({Users.id == user_info['data']['id']})
     return ChatService.sendChatMessage(msg, room_uuid, Type, room_data, room_type, created_at, save_action, user_data)
Пример #17
0
 def upload(params):
     filename = secure_filename(params['name'])
     base64Data = params['dataUrl'].split(',')[1]
     if (base64Data and Utils.allowed_file(filename)):
         file_suffix = params['name'].split('.')[1]
         path = UPLOAD_FOLDER + Utils.unique_id() + '.' + file_suffix
         full_path = os.getcwd() + path
         f = open(full_path, 'wb')
         data = base64.b64decode(base64Data)
         f.write(data)
         #f.write(params['arrayBuffer'])
         f.close
         return Utils.formatBody({
             'path': path,
             'name': filename
         },
                                 msg='上传成功')
     return Utils.formatError(Code.BAD_REQUEST, '文件类型不允许')
Пример #18
0
 def sendChatMessage(msg, room_uuid, Type, room_data, room_type, created_at, save_action, user_data, user_type=0):
     data = {
         'msg': msg,
         'name': user_data['nick_name'],
         'user_id': user_data['id'],
         'type':  Type,
         'head_img': user_data['head_img'],
         'room_uuid': room_uuid,
         'created_at': created_at,
         'read_status': 0,
         'user_type': user_type
         }
     if room_data != None and room_type == CONST['ROOM']['ADMIN']['value']:
         address_book_data = AddressBook.get(room_uuid)
         # 发送消息
         socketio.emit('chat',  Utils.formatBody(data),
                       namespace='/api', room=room_uuid)
         # 视频消息则直接返回
         if Type ==  CONST['CHAT']['VIDEO']['value']:
             return
         # 如果是云端存储则记录,这边判断不判断都存储
         # if save_action == CONST['SAVE']['CLOUD']['value']:
         res = Msg().getOne({Msg.room_uuid == room_uuid, Msg.created_at ==
                             created_at, Msg.user_id == user_data['id']})
         if res == None:
             copy_data = data.copy()
             copy_data['msg'] = json.dumps(msg)
             copy_data['send_status'] = CONST['STATUS']['SUCCESS']['value']
             Msg().add(copy_data)
         else:
             copy_data = data.copy()
             copy_data['msg'] = json.dumps(msg)
             copy_data['send_status'] = CONST['STATUS']['SUCCESS']['value']
             filters = {
                 Msg.name == copy_data['name'],
                 Msg.created_at == copy_data['created_at'],
                 Msg.room_uuid == room_uuid
             }
             Msg().edit(copy_data, filters)
         # 聊天时同步房间信息
         Room.updateLastMsgRoom(room_uuid, data, created_at)
         # 更新聊天提示数字
         AddressBook.updateUnreadNumber(room_uuid, user_data['id'])
         AddressBook.cleanUnreadNumber(room_uuid, user_data['id'])
         # 更新客户端房间信息
         for item in address_book_data:
             roomList = AddressBook.getRoomList(item.be_focused_user_id)
             socketio.emit('room', Utils.formatBody(
                 roomList), namespace='/api', room='@broadcast.'+str(item.be_focused_user_id))
     if room_data != None and room_type == CONST['ROOM']['ALONE']['value']:
         address_book_data = AddressBook.get(room_uuid)
         # 视频消息则直接返回
         if Type ==  CONST['CHAT']['VIDEO']['value']:
             videoData = data
             videoData["room_type"] = room_type
             # 发送消息
             for item in address_book_data:
                 if item['be_focused_user_id'] != user_data['id']:
                     emit('video',  Utils.formatBody(videoData), room='@broadcast.'+str(item.be_focused_user_id))
             return 
         # 发送消息
         emit('chat',  Utils.formatBody(data), room=room_uuid)
         # 如果是云端存储则记录,这边判断不判断都存储
         # if save_action == CONST['SAVE']['CLOUD']['value']:
         res = Msg().getOne({Msg.room_uuid == room_uuid, Msg.created_at ==
                             created_at, Msg.user_id == user_data['id']})
         if res == None:
             copy_data = data.copy()
             copy_data['msg'] = json.dumps(msg)
             copy_data['send_status'] = CONST['STATUS']['SUCCESS']['value']
             Msg().add(copy_data)
         else:
             copy_data = data.copy()
             copy_data['msg'] = json.dumps(msg)
             copy_data['send_status'] = CONST['STATUS']['SUCCESS']['value']
             filters = {
                 Msg.name == copy_data['name'],
                 Msg.created_at == copy_data['created_at'],
                 Msg.room_uuid == room_uuid
             }
             Msg().edit(copy_data, filters)
         # 聊天时同步房间信息
         Room.updateLastMsgRoom(room_uuid, data, created_at)
         # 更新聊天提示数字
         AddressBook.updateUnreadNumber(room_uuid, user_data['id'])
         AddressBook.cleanUnreadNumber(room_uuid, user_data['id'])
         # 更新客户端房间信息
         for item in address_book_data:
             print(AddressBook,item)
             roomList = AddressBook.getRoomList(item.be_focused_user_id)
             socketio.emit('room', Utils.formatBody(
                 roomList), namespace='/api', room='@broadcast.'+str(item.be_focused_user_id))
     elif room_data != None and room_type == CONST['ROOM']['GROUP']['value']:
         # 获取用户的权限,如果禁言状态返回错误
         filters = {
             UserRoomRelation.user_id == user_data['id'],
             UserRoomRelation.room_uuid == room_uuid
         }
         selfUserRoomRelationData = UserRoomRelation().getOne(filters)
         if selfUserRoomRelationData['status'] == CONST['GROUP']['BLOCK']['value']:
             return Utils.formatError(CONST['CODE']['ERROR']['value'], '禁言中')
         # 获取群组内用户
         user_room_relation_data = UserRoomRelation.get(room_uuid)
         # 发送消息
         emit('chat', Utils.formatBody(data), room=room_uuid)
         # 如果是云端存储则记录
         # if save_action == CONST['SAVE']['CLOUD']['value']:
         res = Msg().getOne({Msg.room_uuid == room_uuid, Msg.created_at ==
                             created_at, Msg.user_id == user_data['id']})
         if res == None:
             copy_data = data.copy()
             copy_data['msg'] = json.dumps(msg)
             copy_data['send_status'] = CONST['STATUS']['SUCCESS']['value']
             Msg().add(copy_data)
         else:
             copy_data = data.copy()
             copy_data['msg'] = json.dumps(msg)
             copy_data['send_status'] = CONST['STATUS']['SUCCESS']['value']
             filters = {
                 Msg.name == copy_data['name'],
                 Msg.created_at == copy_data['created_at'],
                 Msg.room_uuid == room_uuid
             }
             Msg().edit(copy_data, filters)
         # 聊天时同步房间信息
         Room.updateLastMsgRoom(room_uuid, data, created_at)
         # 更新聊天提示数字
         UserRoomRelation.updateUnreadNumber(room_uuid, user_data['id'])
         UserRoomRelation.cleanUnreadNumber(room_uuid, user_data['id'])
         # 更新客户端房间信息
         for item in user_room_relation_data:
             roomList = UserRoomRelation.getRoomList(item.user_id)
             socketio.emit('groupRoom', Utils.formatBody(
                 roomList), namespace='/api', room='@broadcast.'+str(item.user_id))
     return Utils.formatBody({'action': "chat", "data": data})