def validate_user(self, user: UserObj):
user_q = session.query(User).get(user.get_username())
if user_q is None:
return False
elif (user_q.username == user.get_username()) and (user.check_password(
user_q.password_hash)):
return True
else:
return False
def write_message(self, message: MessageObj, user: UserObj):
if (self.__user_repo.is_username_exist(message.get_sender())) and \
(self.__user_repo.is_username_exist(message.get_receiver())) and \
(user.get_username() == message.get_sender()):
self.__message_repo.add_message(message=message)
return True
else:
return False
def register_user(self, user: UserObj):
if not self.check_username(username=user.get_username()):
return self.__user_repo.add_user(user=user)
else:
return False
def delete_message(self, message: MessageObj, user: UserObj):
return self.__message_repo.delete_message(
user_name=user.get_username(), message=message)
def get_all_unread_messages(self, user: UserObj):
return self.__message_repo.get_all_unread_messages(
user_name=user.get_username())
def read_message(self, user: UserObj):
ans: MessageObj = self.__message_repo.get_message(user.get_username())
if ans is not None:
self.__message_repo.edit_message_is_read(ans.id)
return ans
def add_user(self, user: UserObj):
user_data = User(username=user.get_username(),
password_hash=user.set_password_hash())
session.add(user_data)
session.commit()
return True