def as_dto(self) -> MessageDTO:
""" Casts message object to DTO """
dto = MessageDTO()
dto.message_id = self.id
dto.message = self.message
dto.from_username = self.from_user.username
dto.sent_date = self.date
dto.read = self.read
dto.subject = self.subject
return dto
def as_dto(self) -> MessageDTO:
""" Casts message object to DTO """
dto = MessageDTO()
dto.message_id = self.id
dto.message = self.message
dto.sent_date = self.date
dto.read = self.read
dto.subject = self.subject
dto.project_id = self.project_id
dto.task_id = self.task_id
if self.message_type is not None:
dto.message_type = MessageType(self.message_type).name
if self.from_user_id:
dto.from_username = self.from_user.username
return dto
def get_all_messages(user_id: int) -> MessagesDTO:
""" Gets all messages to the user """
user_messages = Message.query.filter(
Message.to_user_id == user_id).all()
if len(user_messages) == 0:
raise NotFound()
messages_dto = MessagesDTO()
for message in user_messages:
dto = MessageDTO()
dto.message_id = message.id
dto.subject = message.subject
dto.sent_date = message.date
dto.read = message.read
dto.from_username = message.from_user.username
messages_dto.user_messages.append(dto)
return messages_dto