Improved typing

Weverse/models/__init__.py

@@ -1,5 +1,5 @@
-from .artist import Artist
 from .comment import Comment
+from .artist import Artist
 from .community import Community
 from .notification import Notification
 from .photo import Photo

Weverse/models/artist.py

@@ -1,3 +1,9 @@
+from typing import Optional, TYPE_CHECKING
+
+if TYPE_CHECKING:
+    from Weverse.models import Community
+
+
 class Artist:
     r"""An Artist object that represents a Weverse Artist that belongs in a community.
 
@@ -111,5 +117,5 @@ def __init__(self, **kwargs):
         self.to_fan_last_created_at = kwargs.get('to_fan_last_created_at')
         self.to_fan_last_expire_in = kwargs.get('to_fan_last_expire_in')
         self.birthday_img_url = kwargs.get('birthday_img_url')
-        self.community = None
+        self.community: Optional[Community] = None
         self.posts = []

Weverse/models/comment.py

@@ -1,3 +1,8 @@
+from typing import Optional, TYPE_CHECKING
+
+if TYPE_CHECKING:
+    from Weverse.models import Post
+
 class Comment:
     r"""A Comment object that represents a Weverse Comment that belongs to an Artist.
 
@@ -60,4 +65,4 @@ def __init__(self, **kwargs):
         self.post_id = kwargs.get('post_id')
         self.created_at = kwargs.get('created_at')
         self.updated_at = kwargs.get('updated_at')
-        self.post = None
+        self.post: Optional[Post] = None

Weverse/models/community.py

@@ -1,3 +1,9 @@
+from typing import List, TYPE_CHECKING
+
+if TYPE_CHECKING:
+    from Weverse.models import Artist, Tab
+
+
 class Community:
     r"""A Comment object that represents a Weverse Comment that belongs to an Artist.
 
@@ -67,6 +73,6 @@ def __init__(self, **kwargs):
         self.full_name = kwargs.get('full_name')
         self.fc_member = kwargs.get('fc_member')
         self.show_member_count = kwargs.get('show_member_count')
-        self.artists = []
-        self.tabs = []
+        self.artists: List[Artist] = []
+        self.tabs: List[Tab] = []
 

Weverse/models/photo.py

@@ -1,3 +1,9 @@
+from typing import Optional, TYPE_CHECKING
+
+if TYPE_CHECKING:
+    from Weverse.models import Post
+
+
 class Photo:
     r"""A Photo object that represents a Weverse Photo that belongs to media or a post.
 
@@ -63,5 +69,5 @@ def __init__(self, **kwargs):
         self.original_img_width = kwargs.get('original_img_width')
         self.original_img_height = kwargs.get('original_img_height')
         self.file_name = kwargs.get('file_name')
-        self.post = None
+        self.post: Optional[Post] = None
 

Weverse/models/video.py

@@ -1,3 +1,9 @@
+from typing import Optional, TYPE_CHECKING
+
+if TYPE_CHECKING:
+    from Weverse.models import Post
+
+
 class Video:
     r"""A Video object that represents a Weverse Video that belongs to media or a post.
 
@@ -42,5 +48,5 @@ def __init__(self, **kwargs):
         self.thumbnail_width = kwargs.get('thumbnail_width')
         self.thumbnail_height = kwargs.get('thumbnail_height')
         self.length = kwargs.get('length')
-        self.post = None
+        self.post: Optional[Post] = None