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='已添加')
def addressBookAdd(params):
if params['focused_user_id'] == params['be_focused_user_id']:
return BaseController().error(msg='无法添加自己为新朋友')
filters = {
AddressBook.focused_user_id == params['focused_user_id'],
AddressBook.be_focused_user_id == params['be_focused_user_id']
}
addressBookData = AddressBook().getOne(filters)
if(addressBookData == None):
# 加入房间号
room_uuid = Utils.unique_id()
# 建立通讯录关系
status = AddressBook.addRoomAndAddressBook(
room_uuid, params['focused_user_id'], params['be_focused_user_id'])
if status == False:
return BaseController().error(msg='添加失败')
#回复被添加用户
socketio.emit('beg', Utils.formatBody({
'action':'beg_add_success'
}), namespace='/room', room='@broadcast.'+str(params['focused_user_id']))
return BaseController().successData(msg='添加成功')
return BaseController().error(msg='已添加')
