def post(self): from utils import set_user_context context = common.sanitize_string(self.request.get("context").strip()) user_key = self.get_current_user_key() set_user_context(user_key, context) self.response.out.write(context)
def get_update(self): update_id = self.request.get("update_id", None) if update_id is not None: update_id = int(update_id) chat_update_id = self.request.get("chat_update_id", None) if chat_update_id is not None: chat_update_id = int(chat_update_id) userchat_key = db.Key(self.request.get("userchat_key", None)) chat_id = userchat_key.id_or_name() if chat_id: peer_id_holder = self.memcache_fetcher.get( config.MEMCACHE_PEER_ID(userchat_key.parent().id_or_name(), chat_id) ) self.login(prev_update_id=update_id, chat_id=chat_id, prev_chat_update_id=chat_update_id) message = None if chat_id: if userchat_key.parent() != self.user_key: self.response.set_status(404) return chat_key = db.Key.from_path("Chat", chat_id) chat_timestamp = common.str2datetime(self.request.get("chat_timestamp")) message = self.request.get("message", None) peer_id = peer_id_holder.get_result() if peer_id is None: userchat = self.datastore_fetcher.get(userchat_key) peer_userchat_key = common.get_ref_key(userchat.get_result(), "peer_userchat") peer_id = peer_userchat_key.parent().id_or_name() memcache.set(peer_id_holder.get_key(), peer_id, time=600) peer_status = self.memcache_fetcher.get(config.MEMCACHE_LAST_BEEN_ONLINE(peer_id)) message_entity = None if message: message = common.htmlize_string(common.sanitize_string(message)) message_entity = models.Message(parent=chat_key, message_string=message, sender=userchat_key) peer_chat_open = self.memcache_fetcher.get(config.MEMCACHE_USER_OPEN_CHAT(peer_id, chat_id)) if peer_chat_open.get_result() is None: peer_unreadchat_key = db.Key.from_path("User", peer_id, "UnreadChat", chat_id) peer_unreadchat = models.UnreadChat(key=peer_unreadchat_key) userchat_holder = self.datastore_fetcher.get(userchat_key) peer_userchat_holder = self.datastore_fetcher.get( db.Key.from_path("User", peer_id, "UserChat", chat_id) ) userchat = userchat_holder.get_result() peer_userchat = peer_userchat_holder.get_result() userchat.last_updated = self.now peer_userchat.last_updated = self.now db.put([message_entity, peer_unreadchat, peer_userchat, userchat]) peer_update_id = memcache.incr(config.MEMCACHE_USER_UPDATE_ID(peer_id), initial_value=0) memcache.set( config.MEMCACHE_USER_NOTIFICATION(peer_id, peer_update_id), { "username": models.User.get_username(self.user_key), "chat_id": chat_id, "message": message, "link": "/chat/%s" % models.User.get_username(self.user_key), "timestamp": message_entity.date_time, }, time=config.NOTIFICATION_DURATION, ) else: db.put(message_entity) if self.chat_update_id: new_messages = ( db.Query(models.Message) .ancestor(chat_key) .filter("date_time >", chat_timestamp) .order("-date_time") .fetch(10) ) try: self.client_update["chat_timestamp"] = str(new_messages[0].date_time) except: pass new_messages.reverse() template_values = { "username": models.User.get_username(self.user_key), "messages": [ { "message_string": msg.message_string, "username": models.User.get_username(common.get_ref_key(msg, "sender").parent()), } for msg in new_messages ], } path = os.path.join(os.path.dirname(__file__), "_messages.html") self.client_update["messages_html"] = template.render(path, template_values).decode("utf-8") elif message_entity is not None: self.client_update["chat_timestamp"] = str(message_entity.date_time) template_values = { "username": models.User.get_username(self.user_key), "messages": [ { "message_string": message_entity.message_string, "username": models.User.get_username(common.get_ref_key(message_entity, "sender").parent()), } ], } path = os.path.join(os.path.dirname(__file__), "_messages.html") self.client_update["messages_html"] = template.render(path, template_values).decode("utf-8") if chat_id and message_entity: self.chat_update_id = memcache.incr(config.MEMCACHE_CHAT_UPDATE_ID(chat_id), delta=1, initial_value=0) self.client_update["status_class"] = "offline" if peer_status.get_result() is None else "online" self._get_client_update() return self.client_update
def post(self): from utils import set_user_context context = common.sanitize_string(self.request.get("context").strip()) user_key = self.get_current_user_key() set_user_context(user_key, context) self.response.out.write(context)
def get_update(self): update_id = self.request.get('update_id', None) if update_id is not None: update_id = int(update_id) chat_update_id = self.request.get('chat_update_id', None) if chat_update_id is not None: chat_update_id = int(chat_update_id) userchat_key = db.Key(self.request.get("userchat_key", None)) chat_id = userchat_key.id_or_name() if chat_id: peer_id_holder = self.memcache_fetcher.get( config.MEMCACHE_PEER_ID(userchat_key.parent().id_or_name(), chat_id)) self.login(prev_update_id=update_id, chat_id=chat_id, prev_chat_update_id=chat_update_id) message = None if chat_id: if userchat_key.parent() != self.user_key: self.response.set_status(404) return chat_key = db.Key.from_path('Chat', chat_id) chat_timestamp = common.str2datetime( self.request.get('chat_timestamp')) message = self.request.get("message", None) peer_id = peer_id_holder.get_result() if peer_id is None: userchat = self.datastore_fetcher.get(userchat_key) peer_userchat_key = common.get_ref_key(userchat.get_result(), 'peer_userchat') peer_id = peer_userchat_key.parent().id_or_name() memcache.set(peer_id_holder.get_key(), peer_id, time=600) peer_status = self.memcache_fetcher.get( config.MEMCACHE_LAST_BEEN_ONLINE(peer_id)) message_entity = None if message: message = common.htmlize_string( common.sanitize_string(message)) message_entity = models.Message(parent=chat_key, message_string=message, sender=userchat_key) peer_chat_open = self.memcache_fetcher.get( config.MEMCACHE_USER_OPEN_CHAT(peer_id, chat_id)) if peer_chat_open.get_result() is None: peer_unreadchat_key = db.Key.from_path( 'User', peer_id, 'UnreadChat', chat_id) peer_unreadchat = models.UnreadChat( key=peer_unreadchat_key) userchat_holder = self.datastore_fetcher.get(userchat_key) peer_userchat_holder = self.datastore_fetcher.get( db.Key.from_path('User', peer_id, 'UserChat', chat_id)) userchat = userchat_holder.get_result() peer_userchat = peer_userchat_holder.get_result() userchat.last_updated = self.now peer_userchat.last_updated = self.now db.put([ message_entity, peer_unreadchat, peer_userchat, userchat ]) peer_update_id = memcache.incr( config.MEMCACHE_USER_UPDATE_ID(peer_id), initial_value=0) memcache.set( config.MEMCACHE_USER_NOTIFICATION( peer_id, peer_update_id), { 'username': models.User.get_username(self.user_key), 'chat_id': chat_id, 'message': message, 'link': '/chat/%s' % models.User.get_username(self.user_key), 'timestamp': message_entity.date_time, }, time=config.NOTIFICATION_DURATION, ) else: db.put(message_entity) if self.chat_update_id: new_messages = db.Query( models.Message).ancestor(chat_key).filter( 'date_time >', chat_timestamp).order('-date_time').fetch(10) try: self.client_update['chat_timestamp'] = str( new_messages[0].date_time) except: pass new_messages.reverse() template_values = { "username": models.User.get_username(self.user_key), "messages": [{ 'message_string': msg.message_string, 'username': models.User.get_username( common.get_ref_key(msg, 'sender').parent()) } for msg in new_messages], } path = os.path.join(os.path.dirname(__file__), '_messages.html') self.client_update['messages_html'] = template.render( path, template_values).decode('utf-8') elif message_entity is not None: self.client_update['chat_timestamp'] = str( message_entity.date_time) template_values = { "username": models.User.get_username(self.user_key), "messages": [{ 'message_string': message_entity.message_string, 'username': models.User.get_username( common.get_ref_key(message_entity, 'sender').parent()) }], } path = os.path.join(os.path.dirname(__file__), '_messages.html') self.client_update['messages_html'] = template.render( path, template_values).decode('utf-8') if chat_id and message_entity: self.chat_update_id = memcache.incr( config.MEMCACHE_CHAT_UPDATE_ID(chat_id), delta=1, initial_value=0) self.client_update[ 'status_class'] = "offline" if peer_status.get_result( ) is None else "online" self._get_client_update() return self.client_update