def notify_invite_peer(self, client_id, ref_client_id, ref_group_id, ref_workspace_domain, ref_subject_name):
# notify a client_id that he/her was invited to chat peer to peer with ref_client_id in ref_group_id
self.model = Notify(
client_id=client_id,
ref_client_id=ref_client_id,
client_workspace_domain=get_owner_workspace_domain(),
ref_group_id=ref_group_id,
ref_subject_name=ref_subject_name,
ref_workspace_domain=ref_workspace_domain,
notify_type=NEW_PEER,
notify_image=None,
notify_title="New message",
notify_content="Some one want to sent you a message",
notify_platform="all"
)
new_group = self.model.add()
# check queue and push
notify_tokens = NotifyToken().get_client_device_ids(client_id)
for notify_token in notify_tokens:
notify_channel = "notify/{}/{}".format(client_id, notify_token.device_id)
if notify_channel in client_notify_queue:
try:
client_notify_queue[notify_channel].put(new_group)
except Exception as e:
logger.error(e)
def notify_adding_member(
self,
client_id,
client_workspace_domain,
ref_client_id,
ref_workspace_domain,
ref_group_id,
ref_subject_name,
notify_type=MEMBER_ADD):
# notify client_id is about added to ref_group_id by ref_client_id
self.model = Notify(
client_id=client_id,
client_workspace_domain=client_workspace_domain,
ref_client_id=ref_client_id,
ref_workspace_domain=ref_workspace_domain,
ref_group_id=ref_group_id,
ref_subject_name=ref_subject_name,
notify_type=notify_type,
notify_image=None,
notify_title="Member Add",
notify_content="A member have been added to the group.",
notify_platform="all"
)
logger.info('notify_adding_member')
new_notification = self.model.add()
# check queue and push
notify_tokens = NotifyToken().get_client_device_ids(client_id)
for notify_token in notify_tokens:
notify_channel = "notify/{}/{}".format(client_id, notify_token.device_id)
if notify_channel in client_notify_queue:
try:
client_notify_queue[notify_channel].put(new_notification)
except Exception as e:
logger.error(e)
raise ValueError
def notify_removing_member(
self,
client_id,
client_workspace_domain,
ref_client_id,
ref_workspace_domain,
ref_group_id,
ref_subject_name,
notify_type=MEMBER_REMOVAL):
# notify client_id about ref_client_id's leaving
self.model = Notify(
client_id=client_id,
client_workspace_domain=client_workspace_domain,
ref_client_id=ref_client_id,
ref_workspace_domain=ref_workspace_domain,
ref_group_id=ref_group_id,
ref_subject_name=ref_subject_name,
notify_type=notify_type,
notify_image=None,
notify_title="Member Removal (Leave)",
notify_content="A member have been removed or have left the group.",
notify_platform="all"
)
logger.info('notify_removing_member:')
new_notification = self.model.add()
notify_tokens = NotifyToken().get_client_device_ids(client_id)
for notify_token in notify_tokens:
notify_channel = "notify/{}/{}".format(client_id, notify_token.device_id)
if notify_channel in client_notify_queue:
try:
client_notify_queue[notify_channel].put(new_notification)
except Exception as e:
logger.error(e)
raise ValueError
def notify_invite_group(self, client_id, ref_client_id, ref_group_id, ref_workspace_domain, ref_subject_name):
# notify a client_id that he/her was invited to chat in group with inviter is ref_client_id in ref_group_id
self.model = Notify(
client_id=client_id,
client_workspace_domain=get_owner_workspace_domain(),
ref_client_id=ref_client_id,
ref_group_id=ref_group_id,
ref_subject_name=ref_subject_name,
ref_workspace_domain=ref_workspace_domain,
notify_type=NEW_GROUP,
notify_image=None,
notify_title="Group Chat",
notify_content="You are added to the group chat",
notify_platform="all"
)
new_group = self.model.add()
# check queue and push
notify_tokens = NotifyToken().get_client_device_ids(client_id)
for notify_token in notify_tokens:
notify_channel = "notify/{}/{}".format(client_id, notify_token.device_id)
if notify_channel in client_notify_queue:
try:
client_notify_queue[notify_channel].put(new_group)
except Exception as e:
logger.error(e)
def notify_client_update_call(self, notify_type, client_id, ref_client_id, ref_group_id):
# notify client_id about event ref_client_id is update call
notify = Notify(
id=0,
client_id=client_id,
client_workspace_domain=get_owner_workspace_domain(),
ref_client_id=ref_client_id,
ref_group_id=ref_group_id,
ref_subject_name="",
ref_workspace_domain="",
notify_type=notify_type,
notify_image=None,
notify_title="",
notify_content="",
read_flg=False,
created_at=datetime.now()
)
notify_tokens = NotifyToken().get_client_device_ids(client_id)
for notify_token in notify_tokens:
notify_channel = "notify/{}/{}".format(client_id, notify_token.device_id)
if notify_channel in client_notify_queue:
try:
client_notify_queue[notify_channel].put(notify)
return True
except Exception as e:
logger.error(e)
return False
def read_notify(self, notify_id):
# mark a notify as readed with id is notify_id
self.model = Notify(id=notify_id, read_flg=True)
self.model.update()
def __init__(self):
super().__init__(Notify())
class NotifyInAppService(BaseService):
"""
Notify in app service, using for real time notifying user about specific events, described in notify type
"""
def __init__(self):
super().__init__(Notify())
def get_unread_notifies(self, client_id):
# get all unread notifies for client_id
lst_notify = self.model.get_unread_notifies(client_id)
lst_obj_res = []
for obj in lst_notify:
obj_res = notify_pb2.NotifyObjectResponse(
id=obj.id,
client_id=obj.client_id,
notify_type=obj.notify_type,
read_flg=obj.read_flg,
created_at=int(obj.created_at.timestamp() * 1000)
)
if obj.ref_client_id:
obj_res.ref_client_id = obj.ref_client_id
if obj.ref_group_id:
obj_res.ref_group_id = obj.ref_group_id
if obj.ref_group_id:
obj_res.ref_group_id = obj.ref_group_id
if obj.notify_image:
obj_res.notify_image = obj.notify_image
if obj.notify_title:
obj_res.notify_title = obj.notify_title
if obj.notify_content:
obj_res.notify_content = obj.notify_content
lst_obj_res.append(obj_res)
response = notify_pb2.GetNotifiesResponse(
lst_message=lst_obj_res
)
return response
async def subscribe(self, client_id, device_id):
# subscribe a channel for client_id in device_id
notify_channel = "notify/{}/{}".format(client_id, device_id)
if notify_channel in client_notify_queue:
client_notify_queue[notify_channel] = None
del client_notify_queue[notify_channel]
await asyncio.sleep(1)
client_notify_queue[notify_channel] = Queue()
def un_subscribe(self, client_id, device_id):
# unsubscribe a channel for client_id in device_id
notify_channel = "notify/{}/{}".format(client_id, device_id)
if notify_channel in client_notify_queue:
client_notify_queue[notify_channel] = None
del client_notify_queue[notify_channel]
def read_notify(self, notify_id):
# mark a notify as readed with id is notify_id
self.model = Notify(id=notify_id, read_flg=True)
self.model.update()
# add notify
def notify_invite_peer(self, client_id, ref_client_id, ref_group_id, ref_workspace_domain, ref_subject_name):
# notify a client_id that he/her was invited to chat peer to peer with ref_client_id in ref_group_id
self.model = Notify(
client_id=client_id,
ref_client_id=ref_client_id,
client_workspace_domain=get_owner_workspace_domain(),
ref_group_id=ref_group_id,
ref_subject_name=ref_subject_name,
ref_workspace_domain=ref_workspace_domain,
notify_type=NEW_PEER,
notify_image=None,
notify_title="New message",
notify_content="Some one want to sent you a message",
notify_platform="all"
)
new_group = self.model.add()
# check queue and push
notify_tokens = NotifyToken().get_client_device_ids(client_id)
for notify_token in notify_tokens:
notify_channel = "notify/{}/{}".format(client_id, notify_token.device_id)
if notify_channel in client_notify_queue:
try:
client_notify_queue[notify_channel].put(new_group)
except Exception as e:
logger.error(e)
def notify_invite_group(self, client_id, ref_client_id, ref_group_id, ref_workspace_domain, ref_subject_name):
# notify a client_id that he/her was invited to chat in group with inviter is ref_client_id in ref_group_id
self.model = Notify(
client_id=client_id,
client_workspace_domain=get_owner_workspace_domain(),
ref_client_id=ref_client_id,
ref_group_id=ref_group_id,
ref_subject_name=ref_subject_name,
ref_workspace_domain=ref_workspace_domain,
notify_type=NEW_GROUP,
notify_image=None,
notify_title="Group Chat",
notify_content="You are added to the group chat",
notify_platform="all"
)
new_group = self.model.add()
# check queue and push
notify_tokens = NotifyToken().get_client_device_ids(client_id)
for notify_token in notify_tokens:
notify_channel = "notify/{}/{}".format(client_id, notify_token.device_id)
if notify_channel in client_notify_queue:
try:
client_notify_queue[notify_channel].put(new_group)
except Exception as e:
logger.error(e)
def notify_removing_member(
self,
client_id,
client_workspace_domain,
ref_client_id,
ref_workspace_domain,
ref_group_id,
ref_subject_name,
notify_type=MEMBER_REMOVAL):
# notify client_id about ref_client_id's leaving
self.model = Notify(
client_id=client_id,
client_workspace_domain=client_workspace_domain,
ref_client_id=ref_client_id,
ref_workspace_domain=ref_workspace_domain,
ref_group_id=ref_group_id,
ref_subject_name=ref_subject_name,
notify_type=notify_type,
notify_image=None,
notify_title="Member Removal (Leave)",
notify_content="A member have been removed or have left the group.",
notify_platform="all"
)
logger.info('notify_removing_member:')
new_notification = self.model.add()
notify_tokens = NotifyToken().get_client_device_ids(client_id)
for notify_token in notify_tokens:
notify_channel = "notify/{}/{}".format(client_id, notify_token.device_id)
if notify_channel in client_notify_queue:
try:
client_notify_queue[notify_channel].put(new_notification)
except Exception as e:
logger.error(e)
raise ValueError
def notify_adding_member(
self,
client_id,
client_workspace_domain,
ref_client_id,
ref_workspace_domain,
ref_group_id,
ref_subject_name,
notify_type=MEMBER_ADD):
# notify client_id is about added to ref_group_id by ref_client_id
self.model = Notify(
client_id=client_id,
client_workspace_domain=client_workspace_domain,
ref_client_id=ref_client_id,
ref_workspace_domain=ref_workspace_domain,
ref_group_id=ref_group_id,
ref_subject_name=ref_subject_name,
notify_type=notify_type,
notify_image=None,
notify_title="Member Add",
notify_content="A member have been added to the group.",
notify_platform="all"
)
logger.info('notify_adding_member')
new_notification = self.model.add()
# check queue and push
notify_tokens = NotifyToken().get_client_device_ids(client_id)
for notify_token in notify_tokens:
notify_channel = "notify/{}/{}".format(client_id, notify_token.device_id)
if notify_channel in client_notify_queue:
try:
client_notify_queue[notify_channel].put(new_notification)
except Exception as e:
logger.error(e)
raise ValueError
def notify_client_update_peer_key(self, client_id, ref_client_id, ref_group_id):
# notify client_id about event ref_client_id updated peer key
notify = Notify(
id=0,
client_id=client_id,
client_workspace_domain=get_owner_workspace_domain(),
ref_client_id=ref_client_id,
ref_group_id=ref_group_id,
ref_subject_name="",
ref_workspace_domain="",
notify_type=PEER_UPDATE_SIGNAL_KEY,
notify_image=None,
notify_title="",
notify_content="",
read_flg=False,
created_at=datetime.now()
)
notify_tokens = NotifyToken().get_client_device_ids(client_id)
for notify_token in notify_tokens:
notify_channel = "notify/{}/{}".format(client_id, notify_token.device_id)
if notify_channel in client_notify_queue:
try:
client_notify_queue[notify_channel].put(notify)
except Exception as e:
logger.error(e)
def notify_client_update_call(self, notify_type, client_id, ref_client_id, ref_group_id):
# notify client_id about event ref_client_id is update call
notify = Notify(
id=0,
client_id=client_id,
client_workspace_domain=get_owner_workspace_domain(),
ref_client_id=ref_client_id,
ref_group_id=ref_group_id,
ref_subject_name="",
ref_workspace_domain="",
notify_type=notify_type,
notify_image=None,
notify_title="",
notify_content="",
read_flg=False,
created_at=datetime.now()
)
notify_tokens = NotifyToken().get_client_device_ids(client_id)
for notify_token in notify_tokens:
notify_channel = "notify/{}/{}".format(client_id, notify_token.device_id)
if notify_channel in client_notify_queue:
try:
client_notify_queue[notify_channel].put(notify)
return True
except Exception as e:
logger.error(e)
return False