cjh MXdZddlmZddlZddlZddlmZmZmZm Z ddl Z ddl m Z ddl mZddlmZdd lmZmZmZmZmZdd ldd lmZerdd lmZdd lmZe ddZdZej>e Z!GddZ"GddZ#GddZ$GddZ%GddeZ&y)ae The MIT License (MIT) Copyright (c) 2015-2021 Rapptz Copyright (c) 2021-present Pycord Development 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) TYPE_CHECKINGAnyCallableTypeVar)ExponentialBackoff)Client)Status)ClientExceptionConnectionClosedGatewayNotFound HTTPExceptionPrivilegedIntentsRequired)*)AutoShardedConnectionState) BaseActivity)DiscordWebSocketEI EventItem)bound)AutoShardedClient ShardInfoc$eZdZdZdZdZdZdZdZy) EventTyperrN) __name__ __module__ __qualname__close reconnectresumeidentify terminate clean_close@/var/www/html/venv/lib/python3.12/site-packages/discord/shard.pyrr=s EI FHIKr*rc@eZdZdZ ddZddZddZd dZy) rtypesharderrorc.||_||_||_yNr-)selfetyper/r0s r+__init__zEventItem.__init__Is #( ', r*c`t|tstS|j|jkSr2 isinstancerNotImplementedr.r3others r+__lt__zEventItem.__lt__Ps%%+! !yy5::%%r*c`t|tstS|j|jk(Sr2r7r:s r+__eq__zEventItem.__eq__Us%%+! !yyEJJ&&r*c,t|jSr2)hashr.r3s r+__hash__zEventItem.__hash__ZsDIIr*N)r4intr/z Shard | Noner0zException | NonereturnNone)r3rr;rrDboolrDrC)r r!r" __slots__r5r<r>rBr)r*r+rrFs;*I--!--6F- -& ' r*cveZdZ d dZed dZddZddZddZddZ ddZ ddZ dd Z dd Z y )ShardcT||_||_|j|_||_|jj |_d|_|j|_t|_ d|_ tttttj t"j$f|_y)NF)ws_clientdispatch _dispatch _queue_putloop _disconnect _reconnectr _backoff_taskOSErrorrrr aiohttp ClientErrorasyncio TimeoutError_handled_exceptions)r3rLclient queue_puts r+r5zShard.__init___s %'% .4oo7@/3||/@/@ !& ++,>,@ *.         A  r*c.|jjSr2)rLshard_idrAs r+idzShard.idwswwr*c`|jj|j|_yr2)rQ create_taskworkerrUrAs r+launchz Shard.launch|sYY**4;;=9 r*c|j6|jjs|jjyyyr2)rUdonecancelrAs r+ _cancel_taskzShard._cancel_tasks1 :: !$**//*; JJ   +< !r*cxK|j|jjdd{y7w)N)code)rhrLr#rAs r+r#z Shard.closes* ggmmm&&&s 0:8:cxK|jd{|jd|jy7!w)Nshard_disconnect)r#rOr`rAs r+ disconnectzShard.disconnects-jjl )4773 s :8":c K|jd|jd|j|js+|jt t j ||y|jjryt|trP|jdvrBt|jd}|jt t j||yt|tr|jdk(r>|jt t j |t#|jy|jdk7r+|jt t j ||y|j$j'}t(j+d|j|| t-j.|d{|jt t j0||y7/w) Nrnrm)6iF'T)r%irjz/Attempting a reconnect for shard ID %s in %.2fs)exc_info)rOr`rSrPrrr#rM is_closedr8rVerrnoReconnectWebSocketr%r rkr'rrTdelay_logr0rYsleepr$)r3eexcretrys r+_handle_disconnectzShard._handle_disconnectsr |$ )4773 OOIiootQ? @  << ! ! #  a !agg&<$TWWT:C OOIi&6&6cB C  a) *vv~!++T3LTWW3U vv~ )//4 CD ##% = GG   mmE"""  )"5"5tQ?@ #sG HG>0Hc~K|jjs? |jjd{|jjs>yy7!#t$rR}|j rt j nt j}|jt|||Yd}~yd}~w|j$r$}|j|d{7Yd}~yd}~wtj$rYyt$r4}|jtt j||Yd}~yd}~wwxYwwr2)rMrrrL poll_eventrtr%rr&rPrr[r{rYCancelledError Exceptionr')r3rxr4s r+rcz Shard.workers,,((* gg((***,,((**% ,-HH (():L:L %q 9:++ --a000))   )*=*=tQ GH sD=AAAD=D=A D:(AB50D=5D:C&CC&!D=&D:<D=>D:*D50D=5D::D=cK|j|jd|jd|jtj d|j |j t j|j|j|j|jj|jj}tj|dd{|_ |jy7#|j $r$}|j#|d{7Yd}~yd}~wtj$$rYyt&$r4}|j)t+t,j.||Yd}~yd}~wwxYww)Nrnrmz1Got a request to %s the websocket at Shard ID %s.)r%r_sessionsequenceN@timeout)rhrOr`rvinfoopr from_clientrMr%rL session_idrrYwait_forrdr[r{r~rrPrrr')r3rycororxs r+ reidentifyzShard.reidentifys  |$ )4773 EsvvtwwW #// zz**)) D$,,T4@@DG KKMA'' -))!, , ,%%   E OOIi&9&94C D D EsmA)F,A:D&D' D0FDFD1&D)'D1,F1FF F*F;FFFc$K|j tj|j|jj |j }tj|dd{|_|jy7#|j$r$}|j|d{7Yd}~yd}~wtj$rYyt$r4}|jtt j"||Yd}~yd}~wwxYww)N)gatewayr_rr)rhrrrMrLresume_gateway_urlr`rYrrdr[r{r~rrPrrr')r3rrxs r+r$zShard.reconnects  #// 22D $,,T4@@DG KKMA'' -))!, , ,%%   E OOIi&9&94C D D EskDAB .B / B 8D B D B9.B1/B94D9D DD *DDD  DN)rLrr\rr]zCallable[[EventItem], None]rDrErGrDrE)rxrrDrE)ryrtrDrE)r r!r"r5propertyr`rdrhr#rnr{rcrr$r)r*r+rJrJ^sk  " /    0  : '4$AL".r*rJcVeZdZdZdZd dZd dZd dZd dZd dZ e ddZ d d Z y )raA 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. _parentr` shard_countcB||_|j|_||_yr2r)r3parentrs r+r5zShardInfo.__init__s$ yy'2r*cD|jjj S)z1Whether the shard connection is currently closed.)rrLopenrAs r+rrzShardInfo.is_closeds<>  ll%%''' /979cK|js"|jjd{|jjd{y7'7w)zG|coro| Disconnects and then connects the shard again. N)rrrrnr$rAs r+r$zShardInfo.reconnectsG ~~,,))+ + +ll$$&&& ,&s!.AA!AAAAcvK|jsy|jjd{y7w)z_|coro| Connects a shard. If the shard is already connected this does nothing. N)rrrr$rAs r+connectzShardInfo.connects+ ~~ ll$$&&&rcB|jjjS)zSMeasures latency between a HEARTBEAT and a HEARTBEAT_ACK in seconds for this shard.)rrLlatencyrAs r+rzShardInfo.latency's||&&&r*cJ|jjjS)zWhether 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 )rrLis_ratelimitedrAs r+is_ws_ratelimitedzShardInfo.is_ws_ratelimited,s||--//r*N)rrJr int | NonerDrErDrFrrDfloat) r r!r"__doc__rHr5rrrnr$rrrrr)r*r+rrs@ 1I3 ( (''''0r*rc eZdZUdZerded<dd dfdZ ddd ddZdd Ze dd Z e dd Z dd Z e d d Z dd d!dZd"dZddd#dZd"dZdddd d$dZd%dZxZS)&raMA 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``. Attributes ---------- shard_ids: Optional[List[:class:`int`]] An optional list of shard_ids to launch the shards with. r _connectionNrQc|jdd|jdd_t |d|i|jBj t dt jttfs t di_ jj_ fdj_ tj_y)Nr_ shard_idsrQz>When passing manual shard_ids, you must provide a shard_count.z.shard_ids parameter must be a list or a tuple.cSr2r)rAsr+z,AutoShardedClient.__init__..kstr*)poprsuperr5rr r8listtuple_AutoShardedClient__shards_get_websocketr _get_clientrY PriorityQueue_AutoShardedClient__queue)r3rQargskwargs __class__s` r+r5zAutoShardedClient.__init__Us  :t$+1::k4+H $4T4V4 >> %'%T u >%&VWW *.*=*=''3$,,. r*)r_c\||dz |jz}|j|jS)N)rrrL)r3guild_idr_s r+rz AutoShardedClient._get_websocketns3   B$*:*::H}}X&)))r*c td|j|j|j|j|j d|S)N)rNhandlershookshttprQr))rrN _handlers_hooksrrQ)r3optionss r+ _get_statezAutoShardedClient._get_statevsA) ]]^^++     r*c|js tdStd|jDt |jz S)aIMeasures 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|] \}}| ywr2r)).0_rs r+ z,AutoShardedClient.latency..s 7]]8,FVT%5%56 6  s ' 33c |jjDcic]\}}|t||jc}}Scc}}w)z?Returns a mapping of shard IDs to their respective info object.)rrrrrs r+shardszAutoShardedClient.shardssG %)MM$7$7$9  & i(8(89 9   s!AFinitialcK tj||||}tj|dd{}t|||jjx|j|<}|jy7E#t$rRt j d|tjdd{7|j||d{7cYSwxYww)N)rrr_gf@rz/Failed to connect for shard_id: %s. Retrying...g@)rrrYrrrv exceptionrw launch_shardrJr put_nowaitrrd)r3rr_rrrLrets r+rzAutoShardedClient.launch_shards >#//gwD''e< NNLh W--$ $ $**7H=== = >sPC3A?A=A?AC=A??7C6B97CCCCCCcK|j+|jjd{\|_}n"|jjd{}|j|j_|j xst |j}||j_|D]&}||dk(}|j|||d{(|jjjy777/w)Nrr) rrget_bot_gateway get_gatewayrrrangershards_launchedset)r3rrr_rs r+ launch_shardszAutoShardedClient.launch_shardss    #.2ii.G.G.I(I %D g II1133G'+'7'7$NN=eD,<,<&= %."! HH)A,.G##GXw#G G G H ((,,.)J3 Hs4*DC;*DC=A7DC?,D=D?DT)r$cK||_|jd{|js|jj d{}|j t jk(rX|jd{t|jtr%|jjdk7r |jy|j t jt jfvr.|jj|jd{n|j t j k(r#|jj!d{n_|j t j"k(r$|jd{|j|j t j$k(ry|jsyy777X777Ow)Nrj)rSrrrrgetr.rr#r8r0r rkr&r%r/rr$r'r()r3r$items r+rzAutoShardedClient.connectsV#  """.."))++DyyIOO+jjl""djj*:; SW@W**$y1193C3CDDjj++DJJ777i111jj**,,,i111jjl""jj i333.."" #," 8,"spG&G2G&G4G&GBG&G ?G&G"5G&G$>G&G&G&G& G&"G&$G&cZK|jryd|_|jD]} |jdd{|j j Dcgc]1}tj|j|j3}}|rtj|d{|jjd{|jjtt j"ddy7#t$rYwxYwcc}w7q7Qw)z:|coro| Closes the connection to Discord. NT)forcer)rr_closed voice_clientsrnrrvaluesrY ensure_futurer#rQwaitrrrrrr()r3vcr/to_closes r+r#zAutoShardedClient.closes >>   $$ B mm$m/// --/   ! !%++-dii @   ,,x( ( (iioo  )*?*?t LM0   )sd(D+DDDD+#6D"D+5D'6!D+D)9D+D DD+D D+)D+)activitystatusr_cK|d}tj}n2|tjurd}tj}n |}t|}|[|jj D]'}|j j||d{)|jj}ng|j|}|j j||d{|jjDcgc]}|j|k(s|}}|dn|f} |D]} | j} | | | _ || _ !y77ecc}ww)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. 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 ------ InvalidArgument If the ``activity`` parameter is not of proper type. Nonline invisible)rrr))r rofflinestrrrrLchange_presencerguildsr_me activitiesr) r3rrr_ status_value status_enumr/rgrguildrs r+rz!AutoShardedClient.change_presences;J >#L --K v~~ %&L ..K Kv;L  --/ Whh...VVV W%%,,FMM(+E((**H\*R R R!%!1!1!8!8SAAJJ(D> /E<E>EcVtd|jjDS)anWhether 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 c3PK|]}|jj ywr2)rLr)rr/s r+rz6AutoShardedClient.is_ws_ratelimited..HsQ588**,Qs$&)anyrrrAs r+rz#AutoShardedClient.is_ws_ratelimited=s"Q$--:N:N:PQQQr*)rrrQz asyncio.AbstractEventLoop | NonerrrDrEr2)rrr_rrDr)rrrDrr)rDzlist[tuple[int, float]])r_rCrDzShardInfo | None)rDzdict[int, ShardInfo])rrr_rCrrFrDrEr)r$rFrDrE)rzBaseActivity | Nonerz Status | Noner_rCrDrEr)r r!r"rr__annotations__r5rrrrrrrrrrr#rr __classcell__)rs@r+rr7s94// 26//// /  /4&**EI*"*8B* *  R R  7  ?D&)7; "/"26*N:)- $ B$&B$ B$  B$  B$H Rr*r)'r __future__rrYloggingtypingrrrrrWbackoffr r\r enumsr errorsr r rrrrstaterrrrr__all__ getLoggerr rvrrrJrrr)r*r+rs2#88'-&) [ )B  w"0KK\H0H0VQRQRr*