iYdZddlmZddlZddlZddlZddlZddlmZddl m Z ddl m Z ddl Tdd lmZmZmZmZmZdd lmZdd lmZmZmZmZmZmZmZmZer4dd lm Z dd l m!Z!ddl"m#Z#ddl$m%Z%ddl&m'Z'ddl m(Z(Gdde(dZ)dZ*ej+e,Z-GddZ.GddZ/GddZ0GddZ1Gdd Z2Gd!d"e Z3dS)#a: The MIT License (MIT) Copyright (c) 2015-present Rapptz Permission is hereby granted, free of charge, to any person obtaining a copy of this software and associated documentation files (the "Software"), to deal in the Software without restriction, including without limitation the rights to use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of the Software, and to permit persons to whom the Software is furnished to do so, subject to the following conditions: The above copyright notice and this permission notice shall be included in all copies or substantial portions of the Software. THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE. ) annotationsN)AutoShardedConnectionState)Client)ExponentialBackoff)*)ClientException HTTPExceptionGatewayNotFoundConnectionClosedPrivilegedIntentsRequired)Status) TYPE_CHECKINGAnyCallableTupleTypeOptionalListDict)Unpack)DiscordWebSocket) BaseActivity)Intents)SessionStartLimit)_ClientOptionsc$eZdZUded<ded<dS)_AutoShardedClientOptionsz List[int] shard_idszOptional[float]shard_connect_timeoutN)__name__ __module__ __qualname____annotations__T/home/ubuntu/.hermes/hermes-agent/venv/lib/python3.11/site-packages/discord/shard.pyrr9s*......r&rF)total)AutoShardedClient ShardInfoSessionStartLimitsc&eZdZdZdZdZdZdZdZdS) EventTyperrN) r!r"r#close reconnectresumeidentify terminate clean_closer%r&r'r-r-Gs, EI FHIKKKr&r-c2eZdZdZdd ZddZddZddZdS) EventItemtypesharderroretypeintr<Optional['Shard']r=Optional[Exception]returnNonec0||_||_||_dSNr:)selfr>r<r=s r'__init__zEventItem.__init__Ss (- */ r&otherobjectboolcZt|tstS|j|jkSrE isinstancer9NotImplementedr;rFrHs r'__lt__zEventItem.__lt__Xs)%++ "! !y5:%%r&cZt|tstS|j|jkSrErLrOs r'__eq__zEventItem.__eq__]s)%++ "! !yEJ&&r&c*t|jSrE)hashr;rFs r'__hash__zEventItem.__hash__bsDIr&N)r>r?r<r@r=rArBrC)rHrIrBrJrBr?)r!r"r# __slots__rGrPrRrVr%r&r'r9r9Psi*I0000 &&&& '''' r&r9cneZdZdd Zedd Zdd Zdd ZddZddZ ddZ ddZ ddZ ddZ dS)Shardwsrclientr) queue_putCallable[[EventItem], None]rBrCc ||_||_|j|_||_d|_|j|_t|_d|_ tttttjt jf|_dS)NF)r[_clientdispatch _dispatch _queue_put _disconnect _reconnectr_backoff_taskOSErrorr r r aiohttp ClientErrorasyncio TimeoutError_handled_exceptions)rFr[r\r]s r'rGzShard.__init__gso$&% .4o7@!& +,>,@,@ -1        A    r&r?c|jjSrE)r[shard_idrUs r'idzShard.idyswr&cr|jj||_dSrE)r`loop create_taskworkerrgrUs r'launchz Shard.launch~s(\&224;;==AA r&c|j4|js|jdSdSdSrE)rgdonecancelrUs r' _cancel_taskzShard._cancel_tasksC : !$*//*;*; ! J        " ! ! !r&ctK||jdd{VdS)N)code)ryr[r2rUs r'r2z Shard.closesF gmmm&&&&&&&&&&&r&ctK|d{V|d|jdS)Nshard_disconnect)r2rbrprUs r' disconnectzShard.disconnectsCjjll )4733333r&e Exceptionc K|d|d|j|js0|t t j||dS|jrdSt|trO|j dvrFt|jd}|t t j ||dSt|tr|jdkrB|t t j|t#|jdS|jdkr0|t t j||dS|j}t(d|j|| t-j|d{V|t t j||dS) Nrr~)6iF'T)r4r{z/Attempting a reconnect for shard ID %s in %.2fs)exc_info)rbrprercr9r-r2r` is_closedrMrherrnoReconnectWebSocketr4r r|r6r rfdelay_logr=rksleepr3)rFrexcretrys r'_handle_disconnectzShard._handle_disconnects |$$$ )47333  OOIiotQ?? @ @ @ F < ! ! # #  F a ! ! ag&<&<$TWT:::C OOIi&6cBB C C C F a) * * v~~ )*=tE^_c_fEgEg h hiiiv~~ )/4 C CDDD ##%% Ddgu_` aaamE"""""""""  )"5tQ??@@@@@r&cnK|js |jd{Vn#t$rN}|jr t jn t j}|t|||Yd}~dSd}~w|j $r&}| |d{VYd}~dSd}~wtj $rYdSt$r9}|tt j||Yd}~dSd}~wwxYw|jdSdSrE)r`rr[ poll_eventrr4r-r5rcr9rmrrkCancelledErrorrr6)rFrr>s r'rtz Shard.workers,((**  g((**********%   ,-HL (():L %q 9 9:::+   --a000000000)       )*=tQ G GHHH ,((**     s4> DAB DB??D D.DDrrcK||d|d|jtd|j|j t j|j|j |j sdn |j j |j|j j |j j }tj|dd{V|_ |dS#|j$r&}||d{VYd}~dSd}~wt&$rz}td|j|j|j|j r t(j n t(j}|t/|||Yd}~dSd}~wtj$rYdSt2$r9}|t/t(j||Yd}~dSd}~wwxYw)Nrr~z1Got a request to %s the websocket at Shard ID %s.)r4gatewayrosessionsequenceN@timeoutz:Somehow got a signal to %s while trying to %s shard ID %s.)ryrbrprdebugopr from_clientr`r4r[r session_idrrkwait_forrurmrrr-r5rcr9rrr6)rFrcororrs r' reidentifyzShard.reidentifys0  |$$$ )47333 FPTPWXXX #/ z$'JCDGO*) D$,T4@@@@@@@@@DG KKMMMMM' - - -))!,, , , , , , , , , , , , , ,! 4 4 4 JJSUVUY[^[acgcj k k k%&XE!!93EB OOIb$22 3 3 3 3 3 3 3 3 3%    FF E E E OOIi&94CC D D D D D D D D D Es8.A1C55 G9?D  G9-A/F""G97 G9.G44G9cK| tj|j|j}t j|dd{V|_|dS#|j $r&}| |d{VYd}~dSd}~wt j $rYdSt$r9}| ttj||Yd}~dSd}~wwxYw)Nrorr)ryrrr`rprkrr[rurmrrrrcr9r-r6)rFrrs r'r3zShard.reconnects7  #/ twOOOD#,T4@@@@@@@@@DG KKMMMMM' - - -))!,, , , , , , , , , , , , , ,%    FF E E E OOIi&94CC D D D D D D D D D Es*AA// C19BC1/ C18.C,,C1N)r[rr\r)r]r^rBrCrWrBrC)rrrBrC)rrrBrC)r!r"r#rGpropertyrpruryr2rrrtrr3r%r&r'rZrZfs    $   X BBBB    ''''4444AAAA:"8      r&rZc^eZdZdZdZdd Zdd Zdd Zdd ZddZ e ddZ ddZ dS)r*aA class that gives information and control over a specific shard. You can retrieve this object via :meth:`AutoShardedClient.get_shard` or :attr:`AutoShardedClient.shards`. .. versionadded:: 1.4 Attributes ------------ id: :class:`int` The shard ID for this shard. shard_count: Optional[:class:`int`] The shard count for this cluster. If this is ``None`` then the bot has not started yet. _parentrp shard_countparentrZr Optional[int]rBrCc:||_|j|_||_dSrEr)rFrrs r'rGzShardInfo.__init__s $ y*5r&rJc&|jjj S)z@:class:`bool`: Whether the shard connection is currently closed.)rr[openrUs r'rzShardInfo.is_closeds>    Fl%%'''''''''''r&cK|s|jd{V|jd{VdS)zG|coro| Disconnects and then connects the shard again. N)rrrr3rUs r'r3zShardInfo.reconnectsl ~~ ,,))++ + + + + + + +l$$&&&&&&&&&&&r&ctK|sdS|jd{VdS)z_|coro| Connects a shard. If the shard is already connected this does nothing. N)rrr3rUs r'connectzShardInfo.connectsK ~~  Fl$$&&&&&&&&&&&r&floatc$|jjjS)zc:class:`float`: Measures latency between a HEARTBEAT and a HEARTBEAT_ACK in seconds for this shard.)rr[latencyrUs r'rzShardInfo.latency!s|&&r&c>|jjS)z:class:`bool`: Whether the websocket is currently rate limited. This can be useful to know when deciding whether you should query members using HTTP or via the gateway. .. versionadded:: 1.6 )rr[is_ratelimitedrUs r'is_ws_ratelimitedzShardInfo.is_ws_ratelimited&s|--///r&N)rrZrrrBrCrBrJrrBr) r!r"r#__doc__rXrGrrr3rrrrr%r&r'r*r*s  1I6666 (((( ( ( ( ('''''''''''X'000000r&r*ceZdZdZdZddZdS)r+aA class that holds info about session start limits .. versionadded:: 2.5 Attributes ---------- total: :class:`int` The total number of session starts the current user is allowed remaining: :class:`int` Remaining remaining number of session starts the current user is allowed reset_after: :class:`int` The number of milliseconds until the limit resets max_concurrency: :class:`int` The number of identify requests allowed per 5 seconds r( remaining reset_aftermax_concurrencykwargsUnpack[SessionStartLimit]c n|d|_|d|_|d|_|d|_dS)Nr(rrrr)rFrs r'rGzSessionStartLimits.__init__Ds: / $[1 &} 5$*+<$=r&N)rr)r!r"r#rrXrGr%r&r'r+r+1s: II>>>>>>r&r+ceZdZUdZerded<d6fd Zd7d dd8dZd9dZe d:dZ e d;dZ dd Zd!d"d?d'Zd@d(Zd@fd) Zd*d+dAd-Zd@d.Zd d d d/dBd4ZdCd5ZxZS)Dr)aA client similar to :class:`Client` except it handles the complications of sharding for the user into a more manageable and transparent single process bot. When using this client, you will be able to use it as-if it was a regular :class:`Client` with a single shard when implementation wise internally it is split up into multiple shards. This allows you to not have to deal with IPC or other complicated infrastructure. It is recommended to use this client only if you have surpassed at least 1000 guilds. If no :attr:`.shard_count` is provided, then the library will use the Bot Gateway endpoint call to figure out how many shards to use. If a ``shard_ids`` parameter is given, then those shard IDs will be used to launch the internal shards. Note that :attr:`.shard_count` must be provided if this is used. By default, when omitted, the client will launch shards from 0 to ``shard_count - 1``. .. container:: operations .. describe:: async with x Asynchronously initialises the client and automatically cleans up. .. versionadded:: 2.0 Attributes ------------ shard_ids: Optional[List[:class:`int`]] An optional list of shard_ids to launch the shards with. shard_connect_timeout: Optional[:class:`float`] The maximum number of seconds to wait before timing out when launching a shard. Defaults to 180 seconds. .. versionadded:: 2.4 r _connectionargsrintentsrr!Unpack[_AutoShardedClientOptions]rBrCc|dd|dd_|dd_tj|d|i|jFjt dtjttfst di_ j j _ fdj _ dS) Nrorr gf@rz>When passing manual shard_ids, you must provide a shard_count.z.shard_ids parameter must be a list or a tuple.cSrEr%rUsr'z,AutoShardedClient.__init__..str&)poprr superrGrr rMlisttuple_AutoShardedClient__shards_get_websocketr _get_client)rFrrr __class__s` r'rGzAutoShardedClient.__init__vs :t$$$.4jjd.K.K6 %'%&fgggu >> X%&VWWW *.*=''3|||$$$r&Nrguild_idrrorcD| |dz |jz}|j|jS)N)rrr[)rFrros r'rz AutoShardedClient._get_websockets)   B$*::H}X&))r&optionsc Ltd|j|j|j|jd|S)N)rahandlershookshttpr%)rra _handlers_hooksr)rFrs r' _get_statezAutoShardedClient._get_states<) ]^+       r&rc|jstdStd|jDt |jz S)aY:class:`float`: Measures latency between a HEARTBEAT and a HEARTBEAT_ACK in seconds. This operates similarly to :meth:`Client.latency` except it uses the average latency of every shard's latency. To get a list of shard latency, check the :attr:`latencies` property. Returns ``nan`` if there are no shards ready. nanc3 K|] \}}|V dSrEr%).0_rs r' z,AutoShardedClient.latency..s&<z/AutoShardedClient.latencies..s&ZZZ558+,ZZZr&ritemsrUs r'rzAutoShardedClient.latenciess' [ZDMDWDWDYDYZZZZr&r?Optional[ShardInfo]cj |j|}t||jS#t$rYdSwxYw)aL Gets the shard information at a given shard ID or ``None`` if not found. .. versionchanged:: 2.0 ``shard_id`` parameter is now positional-only. Returns -------- Optional[:class:`ShardInfo`] Information about the shard with given ID. ``None`` if not found. N)rr*rKeyError)rFrors r' get_shardzAutoShardedClient.get_shardsL 7]8,FVT%566 6   44 s $ 22Dict[int, ShardInfo]cNfdjDS)zaMapping[int, :class:`ShardInfo`]: Returns a mapping of shard IDs to their respective info object.cBi|]\}}|t|jSr%)r*r)rrorrFs r' z,AutoShardedClient.shards..s-lllBR(F)FD,<==lllr&rrUs`r'shardszAutoShardedClient.shardss0mlllVZVcViViVkVkllllr&r+cdK|jd{V\}}}tdi|S)a|coro| Get the session start limits. This is not typically needed, and will be handled for you by default. At the point where you are launching multiple instances with manual shard ranges and are considered required to use large bot sharding by Discord, this function when used along IPC and a before_identity_hook can speed up session start. .. versionadded:: 2.5 Returns ------- :class:`SessionStartLimits` A class containing the session start limits Raises ------ GatewayNotFound The gateway was unreachable Nr%)rget_bot_gatewayr+)rFrlimitss r'fetch_session_start_limitsz,AutoShardedClient.fetch_session_start_limitssG0"Y6688888888 1f!++F+++r&Finitialryarl.URLrrJcK tj||||}tj||jd{V}na#t $rTt d|tjdd{V| ||d{VcYSwxYwt|||j j x|j |<}|dS)N)rrrorz/Failed to connect for shard_id: %s. Retrying...g@)rrrkrr rr exceptionr launch_shardrZ_AutoShardedClient__queue put_nowaitrru)rFrrorrr[rets r'rzAutoShardedClient.launch_shards >#/gwaijjjD'd6PQQQQQQQQQBB > > > NNLh W W W-$$ $ $ $ $ $ $ $**7H======== = = = > ).b$ 8O(P(PP h# s9>ABBcK|rdS|j?||jd{V\|_}}t j|}n t j}|j|j_|j pt|j}||j_ |D],}||dk}| |||d{V-dS)Nrr) rrrryarlURLrDEFAULT_GATEWAYrrranger)rF gateway_url_session_start_limitrrrors r' launch_shardszAutoShardedClient.launch_shardss >>    F   #  !HL HaHaHcHcBcBcBcBcBcBc ?D k+?h{++GG&6G'+'7$N=eD,<&=&= %."! H HH)A,.G##GXw#GG G G G G G G G G H Hr&cKtd{Vtj|_dSrE)r_async_setup_hookrk PriorityQueuer)rFrs r'r z#AutoShardedClient._async_setup_hooksCgg''))))))))),.. r&T)r3r3cK||_|d{V|s|jd{V}|jt jkrw|d{Vt|j trA|j j dkrt|j jd|j j dkr|j dS|jt jt jfvr&|j |j d{Vn|jt jkr |j d{VnM|jt jkr!|d{V|j |jt jkrdS|dSdS)Nrr{)rer rrgetr;r-r2rMr=r r|r r<rpr5r4rr3r6r7)rFr3items r'rzAutoShardedClient.connects#  """"""""".."" ))++++++++DyIO++jjll"""""""dj*:;;)z$..7 FFDPz$.."j(y193CDDDj++DJ7777777777i111j**,,,,,,,,,,i111jjll"""""""j i333%..""     r&cKjr jd{VSfd}tj|_jd{VdS)z:|coro| Closes the connection to Discord. NcpKjd{VfdjD}|rt j|d{Vjd{Vjttj dddS)Ncjg|]/}tj|j0S))rr)rk ensure_futurer2rr)rr<rFs r'rz;AutoShardedClient.close.._close..(s4qqqQV-ekkmm$)LLLqqqr&) rr2rvaluesrkwaitrrrr9r-r7)to_closerFs r'_closez'AutoShardedClient.close.._close%s"((** * * * * * * *qqqqZ^ZgZnZnZpZpqqqH -l8,,,,,,,,,)//## # # # # # # # L # #Ii.CT4$P$P Q Q Q Q Qr&) _closing_taskrkrs)rFrs` r'r2zAutoShardedClient.closes   ,+++++++ + R R R R R%0::          r&)activitystatusrorOptional[BaseActivity]rOptional[Status]cK|d}tj}n.|tjurd}tj}n|}t|}K|jD]$}|j||d{V%|jj }nG|j}|j||d{Vfd|jj D}|dn|f}|D]} | j } |  || _ || _ dS)a>|coro| Changes the client's presence. Example: :: game = discord.Game("with the API") await client.change_presence(status=discord.Status.idle, activity=game) .. versionchanged:: 2.0 Removed the ``afk`` keyword-only parameter. .. versionchanged:: 2.0 This function will now raise :exc:`TypeError` instead of ``InvalidArgument``. Parameters ---------- activity: Optional[:class:`BaseActivity`] The activity being done. ``None`` if no currently active activity is done. status: Optional[:class:`Status`] Indicates what status to change to. If ``None``, then :attr:`Status.online` is used. shard_id: Optional[:class:`int`] The shard_id to change the presence to. If not specified or ``None``, then it will change the presence of every shard the bot can see. Raises ------ TypeError If the ``activity`` parameter is not of proper type. Nonline invisible)rrc*g|]}|jk |Sr%r)rgros r'rz5AutoShardedClient.change_presence..ms%SSSAAJ(#L -KK v~ % %&L .KK Kv;;L  --// W Wh...VVVVVVVVVV%,FFM(+E(**H\*RR R R R R R R RSSSS!1!8SSSF#+RR(  $ $EBz'BM#BII $ $r&cbtd|jDS)a}:class:`bool`: Whether the websocket is currently rate limited. This can be useful to know when deciding whether you should query members using HTTP or via the gateway. This implementation checks if any of the shards are rate limited. For more granular control, consider :meth:`ShardInfo.is_ws_ratelimited`. .. versionadded:: 1.6 c3HK|]}|jVdSrE)r[r)rr<s r'rz6AutoShardedClient.is_ws_ratelimited..s0QQ58**,,QQQQQQr&)anyrrrUs r'rz#AutoShardedClient.is_ws_ratelimitedys/QQ$-:N:N:P:PQQQQQQr&)rrrrrrrBrCrE)rrrorrBr)rrrBrr)rBr)ror?rBr)rBr)rBr+)rrror?rrJrBrCr)r3rJrBrC)rrrrrorrBrCr)r!r"r#rrr$rGrrrrrrrrrr r rr2r$r __classcell__)rs@r'r)r)Ks %%N0////444444&*Z^******      R R RX R[[[X[7777(mmmXm,,,,6W\      HHHH(//////260!!!!0,0#'"& E$E$E$E$E$E$N R R R R R R R Rr&r))4r __future__rrkloggingrirstaterr\rbackoffrrerrorsr r r r r enumsrtypingrrrrrrrrtyping_extensionsrrrrflagsr types.gatewayrrr__all__ getLoggerr!rr-r9rZr*r+r)r%r&r'r;s0#""""" ------''''''RRRRRRRRRRRRRRRRRRRR /(((((())))))&&&&&&000000&&&&&&/////N%////   w"",}}}}}}}}@H0H0H0H0H0H0H0H0V>>>>>>>>4yRyRyRyRyRyRyRyRyRyRr&