Exemple #1
0
def _message_to_pb(message: Message):
    """
    Turns the given message to a protocol buffer
    """
    if message.is_normal_message:
        return conversations_pb2.Message(
            message_id=message.id,
            author_user_id=message.author_id,
            time=Timestamp_from_datetime(message.time),
            text=conversations_pb2.MessageContentText(text=message.text),
        )
    else:
        return conversations_pb2.Message(
            message_id=message.id,
            author_user_id=message.author_id,
            time=Timestamp_from_datetime(message.time),
            chat_created=conversations_pb2.MessageContentChatCreated()
            if message.message_type == MessageType.chat_created else None,
            chat_edited=conversations_pb2.MessageContentChatEdited()
            if message.message_type == MessageType.chat_edited else None,
            user_invited=conversations_pb2.MessageContentUserInvited(
                target_user_id=message.target_id)
            if message.message_type == MessageType.user_invited else None,
            user_left=conversations_pb2.MessageContentUserLeft()
            if message.message_type == MessageType.user_left else None,
            user_made_admin=conversations_pb2.MessageContentUserMadeAdmin(
                target_user_id=message.target_id)
            if message.message_type == MessageType.user_made_admin else None,
            user_removed_admin=conversations_pb2.
            MessageContentUserRemovedAdmin(target_user_id=message.target_id) if
            message.message_type == MessageType.user_removed_admin else None,
        )
Exemple #2
0
def message_to_pb(message: Message):
    """
    Turns the given message to a protocol buffer
    """
    if message.is_normal_message:
        return conversations_pb2.Message(
            message_id=message.id,
            author_user_id=message.author_id,
            time=Timestamp_from_datetime(message.time),
            text=conversations_pb2.MessageContentText(text=message.text),
        )
    else:
        return conversations_pb2.Message(
            message_id=message.id,
            author_user_id=message.author_id,
            time=Timestamp_from_datetime(message.time),
            chat_created=conversations_pb2.MessageContentChatCreated()
            if message.message_type == MessageType.chat_created else None,
            host_request_status_changed=conversations_pb2.
            MessageContentHostRequestStatusChanged(
                status=hostrequeststatus2api[
                    message.host_request_status_target])
            if message.message_type == MessageType.host_request_status_changed
            else None,
        )