mirror of
https://github.com/Rapptz/discord.py.git
synced 2025-04-19 15:36:02 +00:00
Inject state into PartialEmoji in Activity
This commit is contained in:
parent
adb69e7157
commit
483ffeacd4
@ -99,6 +99,8 @@ if TYPE_CHECKING:
|
||||
ActivityButton,
|
||||
)
|
||||
|
||||
from .state import ConnectionState
|
||||
|
||||
|
||||
class BaseActivity:
|
||||
"""The base activity that all user-settable activities inherit from.
|
||||
@ -809,16 +811,16 @@ ActivityTypes = Union[Activity, Game, CustomActivity, Streaming, Spotify]
|
||||
|
||||
|
||||
@overload
|
||||
def create_activity(data: ActivityPayload) -> ActivityTypes:
|
||||
def create_activity(data: ActivityPayload, state: ConnectionState) -> ActivityTypes:
|
||||
...
|
||||
|
||||
|
||||
@overload
|
||||
def create_activity(data: None) -> None:
|
||||
def create_activity(data: None, state: ConnectionState) -> None:
|
||||
...
|
||||
|
||||
|
||||
def create_activity(data: Optional[ActivityPayload]) -> Optional[ActivityTypes]:
|
||||
def create_activity(data: Optional[ActivityPayload], state: ConnectionState) -> Optional[ActivityTypes]:
|
||||
if not data:
|
||||
return None
|
||||
|
||||
@ -831,10 +833,10 @@ def create_activity(data: Optional[ActivityPayload]) -> Optional[ActivityTypes]:
|
||||
try:
|
||||
name = data.pop('name')
|
||||
except KeyError:
|
||||
return Activity(**data)
|
||||
ret = Activity(**data)
|
||||
else:
|
||||
# we removed the name key from data already
|
||||
return CustomActivity(name=name, **data) # type: ignore
|
||||
ret = CustomActivity(name=name, **data) # type: ignore
|
||||
elif game_type is ActivityType.streaming:
|
||||
if 'url' in data:
|
||||
# the url won't be None here
|
||||
@ -842,4 +844,9 @@ def create_activity(data: Optional[ActivityPayload]) -> Optional[ActivityTypes]:
|
||||
return Activity(**data)
|
||||
elif game_type is ActivityType.listening and 'sync_id' in data and 'session_id' in data:
|
||||
return Spotify(**data)
|
||||
return Activity(**data)
|
||||
else:
|
||||
ret = Activity(**data)
|
||||
|
||||
if isinstance(ret.emoji, PartialEmoji):
|
||||
ret.emoji._state = state
|
||||
return ret
|
||||
|
@ -717,7 +717,7 @@ class Client:
|
||||
"""Optional[:class:`.BaseActivity`]: The activity being used upon
|
||||
logging in.
|
||||
"""
|
||||
return create_activity(self._connection._activity)
|
||||
return create_activity(self._connection._activity, self._connection)
|
||||
|
||||
@activity.setter
|
||||
def activity(self, value: Optional[ActivityTypes]) -> None:
|
||||
|
@ -448,7 +448,7 @@ class Member(discord.abc.Messageable, _UserTag):
|
||||
self._avatar = data.get('avatar')
|
||||
|
||||
def _presence_update(self, data: PartialPresenceUpdate, user: UserPayload) -> Optional[Tuple[User, User]]:
|
||||
self.activities = tuple(map(create_activity, data['activities']))
|
||||
self.activities = tuple(create_activity(d, self._state) for d in data['activities'])
|
||||
self._client_status._update(data['status'], data['client_status'])
|
||||
|
||||
if len(user) > 1:
|
||||
|
@ -188,7 +188,7 @@ class WidgetMember(BaseUser):
|
||||
except KeyError:
|
||||
activity = None
|
||||
else:
|
||||
activity = create_activity(game)
|
||||
activity = create_activity(game, state)
|
||||
|
||||
self.activity: Optional[Union[BaseActivity, Spotify]] = activity
|
||||
|
||||
|
Loading…
x
Reference in New Issue
Block a user