From cb7dcf0397377606f507a7bf0ebdc00a16a00ecc Mon Sep 17 00:00:00 2001 From: ikaroskun Date: Tue, 29 Oct 2024 15:53:59 +0800 Subject: [PATCH] feat(user): :sparkles: add new fields for user model --- pytwitter/models/user.py | 10 ++++++++++ 1 file changed, 10 insertions(+) diff --git a/pytwitter/models/user.py b/pytwitter/models/user.py index ccecdab..79496f1 100644 --- 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)