def does_room_exist(self, room_name):
all_rooms = CryptChatRoom.query()
room_exists = False
for room in all_rooms:
if room.room_name == room_name:
room_exists = True
return room_exists
def post(self):
room_name = self.request.get("room")
new_user_email = self.request.get("user_email").lower()
room = CryptChatRoom.query(CryptChatRoom.room_name == room_name)
our_room = room.fetch()[0]
our_room.allowed_emails.append(new_user_email)
our_room.put()
self.redirect("/room/" + room_name)
def get(self):
room_name = self.request.get("room_name")
room = CryptChatRoom.query(CryptChatRoom.room_name == room_name).fetch()
users = room[0].allowed_emails
users_list = []
for user in users:
users_list.append(user)
self.response.write(json.dumps(users_list))
def is_user_allowed_in_room(self, user_email, room_name):
our_room = CryptChatRoom.query(CryptChatRoom.room_name == room_name)
if our_room.fetch()[0].room_creator_email == user_email:
return True
allowed_users = our_room.fetch()[0].allowed_emails
for user in allowed_users:
if user == user_email:
return True
return False
def get(self):
### should only return rooms that are open to the user
user_email = user_help.get_user_email()
all_rooms = CryptChatRoom.query()
all_rooms_allowed = []
for room in all_rooms:
if user_email in room.allowed_emails:
all_rooms_allowed.append(room.room_name)
self.response.write(json.dumps(all_rooms_allowed))
def post(self):
room_name = self.request.get("room_name")
room_name = re.sub(r'\W+', '', room_name)
room_description = self.request.get("room_description")
if room_name:
room = CryptChatRoom()
room.room_creator_email = user_help.get_user_email()
room.room_name = room_name
room.room_description = room_description
room.allowed_emails.append(user_help.get_user_email())
room.put()
time.sleep(2) # required to give ndb time to create model
return self.redirect("/room/"+room_name)
