Merge pull request #173 from MerleLiuKun/feat-user-field

feat(user): ✨ add new fields for user model
sns-sdks · Oct 29, 2024 · 3ae9a54 · 3ae9a54
2 parents eadf16f + cb7dcf0
commit 3ae9a54
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/pytwitter/models/user.py b/pytwitter/models/user.py
@@ -68,6 +68,14 @@ class UserWithheld(BaseModel):
     country_codes: Optional[List[str]] = field(default=None, repr=False)
 
 
+@dataclass
+class UserAffiliation(BaseModel):
+    user_id: Optional[str] = field(default=None)
+    url: Optional[str] = field(default=None)
+    badge_url: Optional[str] = field(default=None, repr=False, compare=False)
+    description: Optional[str] = field(default=None, repr=False, compare=False)
+
+
 @dataclass
 class User(BaseModel):
     """
@@ -82,6 +90,7 @@ class User(BaseModel):
     location: Optional[str] = field(default=None, repr=False, compare=False)
     pinned_tweet_id: Optional[str] = field(default=None, repr=False, compare=False)
     profile_image_url: Optional[str] = field(default=None, repr=False, compare=False)
+    profile_banner_url: Optional[str] = field(default=None, repr=False, compare=False)
     protected: Optional[bool] = field(default=None, repr=False, compare=False)
     url: Optional[str] = field(default=None, repr=False, compare=False)
     verified: Optional[bool] = field(default=None, repr=False, compare=False)
@@ -97,3 +106,4 @@ class User(BaseModel):
         default=None, repr=False, compare=False
     )
     most_recent_tweet_id: Optional[str] = field(default=None, repr=False, compare=False)
+    affiliation: Optional[UserAffiliation] = field(default=None, repr=False, compare=False)