Export MessageData model

fbchat/__init__.py

@@ -27,7 +27,7 @@ from ._thread import ThreadLocation, ThreadABC, Thread
 from ._user import User, UserData, ActiveStatus
 from ._group import Group, GroupData
 from ._page import Page, PageData
-from ._message import EmojiSize, Mention, Message
+from ._message import EmojiSize, Mention, Message, MessageData
 from ._attachment import Attachment, UnsentMessage, ShareAttachment
 from ._sticker import Sticker
 from ._location import LocationAttachment, LiveLocationAttachment

tests/test_client_payload.py

@@ -4,6 +4,8 @@ from fbchat import (
     ParseError,
     User,
     Group,
+    Message,
+    MessageData,
     UnknownEvent,
     ReactionEvent,
     UserStatusEvent,
@@ -11,7 +13,6 @@ from fbchat import (
     UnsendEvent,
     MessageReplyEvent,
 )
-from fbchat._message import Message, MessageData
 from fbchat._client_payload import parse_client_delta, parse_client_payloads
 
 

tests/test_delta_class.py

@@ -4,6 +4,8 @@ from fbchat import (
     ParseError,
     User,
     Group,
+    Message,
+    MessageData,
     ThreadLocation,
     UnknownEvent,
     PeopleAdded,
@@ -15,7 +17,6 @@ from fbchat import (
     MessageEvent,
     ThreadFolder,
 )
-from fbchat._message import Message, MessageData
 from fbchat._delta_class import parse_delta
 
 

tests/test_delta_type.py

@@ -5,6 +5,8 @@ from fbchat import (
     ParseError,
     User,
     Group,
+    Message,
+    MessageData,
     Poll,
     PollOption,
     PlanData,
@@ -27,7 +29,6 @@ from fbchat import (
     PlanDeleted,
     PlanResponded,
 )
-from fbchat._message import Message, MessageData
 from fbchat._delta_type import parse_delta
 
 

tests/test_message.py

@@ -1,12 +1,7 @@
 import pytest
 import fbchat
-from fbchat._message import (
-    EmojiSize,
-    Mention,
-    Message,
-    MessageData,
-    graphql_to_extensible_attachment,
-)
+from fbchat import EmojiSize, Mention, Message, MessageData
+from fbchat._message import graphql_to_extensible_attachment
 
 
 @pytest.mark.parametrize(