Expose _ActivityTag as BaseActivity to easily refer to.

This commit is contained in:
Rapptz
2020-01-14 20:54:31 -05:00
parent a841efa087
commit 42a084028c
7 changed files with 39 additions and 21 deletions

View File

@ -38,7 +38,7 @@ import zlib
import websockets
from . import utils
from .activity import _ActivityTag
from .activity import BaseActivity
from .enums import SpeakingState
from .errors import ConnectionClosed, InvalidArgument
@ -490,8 +490,8 @@ class DiscordWebSocket(websockets.client.WebSocketClientProtocol):
async def change_presence(self, *, activity=None, status=None, afk=False, since=0.0):
if activity is not None:
if not isinstance(activity, _ActivityTag):
raise InvalidArgument('activity must be one of Game, Streaming, or Activity.')
if not isinstance(activity, BaseActivity):
raise InvalidArgument('activity must derive from BaseActivity.')
activity = activity.to_dict()
if status == 'idle':