EjQdZddlZddlZddlmZddlmZmZm Z ddl m Z ddl Z ddl Z ddlZddlmZddlZddlZddlmZmZmZddlmZddlZdd lmZmZmZmZm Z m!Z!m"Z"m#Z#id d d d ddddddddddddddddddd d!d"d#d$d%d&d'd(d)d*d+d,d-d.Z$e j%e&Z'd/Z(Gd0d1Z)e)Z*Gd2d3e+Z,d4ej-j.d5e/fd6Z0d4ej-j.d7e1d5e/fd8Z2d4ej-j.d5e/fd9Z3d4ej-j.d5e/fd:Z4d;e!d5ej5dzfd<Z6d=e7d5e/fd>Z8d?e#d5dfd@Z9GdAdBZ:dS)CzCommunication with APIs.N)urljoin)ConnectionError HTTPError ReadTimeout)RemoteDisconnected) defaultdict)Timersecondssec_str)cast)UserProfileTypeREQUESTS_PAYLOAD_TYPEGameTypePublicDataType OnlineType ChallengeTypeTOKEN_TESTS_TYPEBackoffDetailsprofilez /api/accountplayingz/api/account/playingstreamz/api/bot/game/stream/{} stream_eventz/api/stream/eventmovez/api/bot/game/{}/move/{}takebackz/api/bot/game/{}/takeback/{}chatz/api/bot/game/{}/chatabortz/api/bot/game/{}/abortacceptz/api/challenge/{}/acceptdeclinez/api/challenge/{}/declineupgradez/api/bot/account/upgraderesignz/api/bot/game/{}/resignexportz/game/export/{} online_botsz/api/bot/online challengez/api/challenge/{}cancelz/api/challenge/{}/cancelstatusz/api/users/statusz /api/user/{}z/api/token/test) public_data token_testceZdZdZddZdS)StopzClass to stop the bot.returnNc0d|_d|_d|_dS)zInitialize the Stop class.FTN) terminated force_quitrestartselfs &/app/lichess-bot-master/lib/lichess.py__init__z Stop.__init__3s r+N)__name__ __module__ __qualname____doc__r3r4r2r*r*0s.  r4r*c:eZdZdZdedejddffd ZxZS)RateLimitedErrorz)r1r=r> __class__s r2r3zRateLimitedError.__init__@s& !!! r4) r6r7r8r9strdatetime timedeltar3 __classcell__)rAs@r2r<r<=s[FFh.@Tr4r<responser+c|jdkS)zFCheck if the status code is 429, which means that we are rate limited.) status_coderFs r2is_new_rate_limitrKFs  3 &&r4rate_limit_status_codec|j|krdS |}d|vo-|didddkS#tjj$rYdSwxYw)zRCheck if response to challenge is a rate limit, either of the bot or the opponent.Ferror ratelimitkey bot.vsBot.day)rIjsongetrequests exceptionsJSONDecodeError)rFrLbodys r2is_daily_game_rate_limitrYJs555u}}$^488K#<#<#@#@#K#K#^^   .uusAAA-,A-c"t|dS)zNCheck if response to a challenge is 400, which means opponent is rate limited.irYrJs r2is_opponent_rate_limitr\V #Hc 2 22r4c"t|dS)zMCheck if response to a challenge is 429, which means the bot is rate limited.rHr[rJs r2is_bot_rate_limitr_[r]r4challenge_responsec|di}|dd}|dkr"tt|dSdS)z_Return the timeout in a challenge response if the bot or the opponent cannot play another game.rOrPrQrRr N)rTr float)r` rate_limitrPs r2get_challenge_timeoutrd`sU#'' R88J .. # #C ouZ 233444 4r4 exceptioncvt|tr|jdur|jjdkp tjS)z/If `is_final` returns True then we won't retry.Ni) isinstancerrFrIstopr.)res r2is_finalriis= y) , , x1C41O xT]TfTruxTx  r4detailsc|d}|d}d|vrd|d<tdjd i|tdtjdS) z;Log exceptions inside functions with the backoff decorator.argskwargsr'zdatazrBacking off {wait:0.1f} seconds after {tries} tries calling function {target} with args {args} and kwargs {kwargs}z Exception: Nr:)loggerdebugformat traceback format_exc)rjrlrms r2backoff_handlerrtos 6?D X Ft+v LLYRRXdd[bddeee LL7y3557788888r4ceZdZdZdedededededdf d Zejej e e e e fd d eeejej dd dddededeeefdzdededejf dZdddededeeefdzdeezeeeefzfdZdddededeeefdzdeefdZdddededeeefdzdefdZejej e e e e fd d eeejej dddddddededeeeefzdzdeeefdzdeeefdzdedzdedee zdzfd Z!dedefd!Z"d"ej#jdefd#Z$d$ed%e%j&ddfd&Z'd$edefd'Z(d$ede%j&fd(Z)dKd)Z*d*ed+e+j,j-ddfd,Z.d*ed-edefd.Z/d*ed/ed0eddfd1Z0d*eddfd2Z1dej#jfd3Z2d*edej#jfd4Z3d5eddfd6Z4dLd5ed8eddfd9Z5defd:Z6deedzfd;Z7d*eddfd<Z8d=eddfd>Z9d*edefd?Z:deefd@Z;d=ededefdAZf dFZ?dGedefdHZ@dIedefdJZAdS)NLichesszBCommunication with lichess.org (and chessdb.cn for getting moves).tokenurlversion logging_level max_retriesr+Ncz||_dd|i|_||_tj|_|jj|jtj|_| d||_ ||_ tt|_tt |d|}||}|st'd|dd}d |d vrt'd |d d S)a Communication with lichess.org (and chessdb.cn for getting moves). :param token: The bot's token. :param url: The base url (lichess.org). :param version: The lichess-bot version running. :param logging_level: The logging level (logging.INFO or logging.DEBUG). :param max_retries: The maximum amount of retries for online moves (e.g. chessdb's opening book). AuthorizationzBearer ?r'rnzThere was an error in retrieving information about the bot's token. Please check that it was copied correctly into your configuration file and try again.scopesrQzbot:play,zPlease use an API access token for your bot that has the scope "Play games with the bot API (bot:play)". The current token has: .N)ryheaderbaseUrlrUSessionsessionheadersupdate other_sessionset_user_agentrzr{rr rate_limit_timersr rapi_postrT RuntimeErrorsplit) r1rwrxryrzr{token_response token_infors r2r3zLichess.__init__~sT .u..   '))  ##DK000%-// C   *&:Ee:L:L. lQV 0W0WXX#''..  1 011 1"-- V\\#.. . . C9? C C CDD D / .r4<皙?)max_timeintervalgiveup on_backoffbackoff_log_levelgiveup_log_levelF)paramsrr> endpoint_name template_argsrrr>ctjd|j||}t |j|j|}|j ||||}t|r-t|dkrdnd} | || | d|_|S)a Send a GET to lichess.org. :param endpoint_name: The name of the endpoint. :param template_args: The values that go in the url (e.g. the challenge id if `endpoint_name` is `accept`). :param params: Parameters sent to lichess.org. :param stream: Whether the data returned from lichess.org should be streamed. :param timeout: The amount of time in seconds to wait for a response. :return: lichess.org's response. backoff)rr>rrrzutf-8)logging getLoggersetLevelrzget_path_templaterrrqrrTrKr set_rate_limit_delayraise_for_statusencoding) r1rrrr>r path_templaterxrFdelays r2api_getzLichess.api_gets* )$$--d.@AAA..}== dl$8M$8-$HII<##CPV#WW X & & <&!8!8AAbAAE  % %mU ; ; ;!!####r4rcP|j|g|Rd|i}|}|S)ad Send a GET to the lichess.org endpoints that return a JSON. :param endpoint_name: The name of the endpoint. :param template_args: The values that go in the url (e.g. the challenge id if `endpoint_name` is `accept`). :param params: Parameters sent to lichess.org. :return: lichess.org's response in a dict. rrrSr1rrrrF json_responses r2 api_get_jsonzLichess.api_get_jsons< 4< M MMMfMMV^VcVcVeVe r4cP|j|g|Rd|i}|}|S)a} Send a GET to the lichess.org endpoints that return a list containing JSON. :param endpoint_name: The name of the endpoint. :param template_args: The values that go in the url (e.g. the challenge id if `endpoint_name` is `accept`). :param params: Parameters sent to lichess.org. :return: lichess.org's response in a list of dicts. rrrs r2 api_get_listzLichess.api_get_lists8 4< M MMMfMM/7}} r4c2|j|g|Rd|i}|jS)ae Send a GET to lichess.org that returns plain text (UTF-8). :param endpoint_name: The name of the endpoint. :param template_args: The values that go in the url (e.g. the challenge id if `endpoint_name` is `accept`). :param params: Parameters sent to lichess.org. :return: The text of lichess.org's response. r)rtext)r1rrrrFs r2 api_get_rawzLichess.api_get_raws- 4< M MMMfMM}r4T)rnrrpayloadrrnrrrctjd|j||}t |j|j|} |j | ||||d} |dkr| | St| r#| |td|r| | } | S)aD Send a POST to lichess.org. :param endpoint_name: The name of the endpoint. :param template_args: The values that go in the url (e.g. the challenge id if `endpoint_name` is `accept`). :param data: Data sent to lichess.org. :param headers: The headers for the request. :param params: Parameters sent to lichess.org. :param payload: Payload sent to lichess.org. :param raise_for_status: Whether to raise an exception if the response contains an error code. :return: lichess.org's response in a dict. rr)rnrrrSr>r#r)rrrrzrrrrqrposthandle_challengerKrr rrS) r1rrnrrrrrrrxrFrs r2rzLichess.api_posts8 )$$--d.@AAA..}== dl$8M$8-$HII<$$StWVZakl$mm K ' '((22 2 X & & B  % %mWR[[ A A A  (  % % ' ' 'AI r4ct|}||r8||}t|dt |d||S)z Get the path template given the endpoint name. Will raise an exception if the path template is rate limited. :param endpoint_name: The name of the endpoint. :return: The path template. z is rate-limited. Will retry in z seconds.) ENDPOINTSis_rate_limitedrate_limit_time_leftr<r )r1rr time_lefts r2rzLichess.get_path_templatesr"-0    . . l11-@@I" __')BTBT___ajll lr4rFc$t|}t|}|}|s|rYttjt |}|r!|td|||d<||d<||d<|S)zMHandle the response to a challenge and, if necessary, the daily game timeout.r#bot_is_rate_limitedopponent_is_rate_limitedrate_limit_timeout) r_r\rSr rCrDrdrr)r1rFrrr`rs r2rzLichess.handle_challenge+s/99#9(#C#C ,4MMOO  =": =+-BCU-V-VWWE" I)))K*@%HHH8K 4 5=U 9 :7< 3 4!!r4r delay_timectd|dt|dt||j|<dS)z Set a delay to a path template if it was rate limited. :param path_template: The path template. :param delay_time: How long we won't call this endpoint. z Endpoint z is rate limited. Waiting z seconds until next request.N)rowarningr r r)r1rrs r2rzLichess.set_rate_limit_delay:sN }=}}GT^L_L_}}}~~~05j0A0A}---r4cB|j| S)z)Check if a path template is rate limited.)r is_expiredr1rs r2rzLichess.is_rate_limitedDs)-8CCEEEEr4c@|j|S)zBHow much time is left until we can use the path template normally.)rtime_until_expirationrs r2rzLichess.rate_limit_time_leftHs%m4JJLLLr4c0|ddS)z%Upgrade the account to a BOT account.rNrr0s r2upgrade_to_bot_accountzLichess.upgrade_to_bot_accountLs i     r4game_idrc |d|t|jdt|jidS)zr Make a move. :param game_id: The id of the game. :param move: The move to make. r offeringDrawrN)rrBr draw_offeredlower)r1rrs r2 make_movezLichess.make_movePs[ fgs49~~,c$2C.D.D.J.J.L.LM  O O O O Or4rc |d||rdnd|rtdntd|S#t$rYdSwxYw)z+Answer an opponent's move takeback request.ryesnoz!Opponent took back previous move.z%Refused opponent's take back request.F)rroinfo Exception)r1rrs r2accept_takebackzLichess.accept_takebackZs  MM*g/HuuD I I I E ?@@@@ CDDDM   55 sAA A$#A$roomrct|tkrPtdt|dtdtd|||d}|d||dS) z Send a message to the chat. :param game_id: The id of the game. :param room: The room (either chat or spectator room). :param text: The text to send. zThis chat message is z1 characters, which is longer than the maximum of z. It will not be sent.z Message: )rrrrN)lenMAX_CHAT_MESSAGE_LENrorr)r1rrrrns r2rz Lichess.chatfs t99+ + + NN_3t99__2F___ ` ` ` NN-t-- . . .d++ fgD 11111r4c2|d|dS)zAborts a game.rNrr1rs r2rz Lichess.abortvs gw'''''r4c2|dddS)z7Get a stream of the events (e.g. challenge, gameStart).rTrr>rr0s r2get_event_streamzLichess.get_event_streamzs||N4|DDDr4c4|d|ddS)z?Get stream of the in-game events (e.g. moves by the opponent).rTrrrrs r2get_game_streamzLichess.get_game_stream~s||HgdB|GGGr4 challenge_idc2|d|dS)zAccept a challenge.rNrr1rs r2accept_challengezLichess.accept_challenges h -----r4genericreasonctjt5|d|d|ddidddddS#1swxYwYdS)zDecline a challenge.rzreason=z Content-Typez!application/x-www-form-urlencodedF)rnrrN) contextlibsuppressrr)r1rrs r2decline_challengezLichess.decline_challenges   + + 2 2 MM)\111#13V"W+0  2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2s AA A ctt|d}||d|S)z&Get the bot's profile (e.g. username).rusername)r r rr)r1rs r2 get_profilezLichess.get_profiles=(9(9)(D(DEE GJ/000r4ctjt5ttt t tf|d}|dcdddS#1swxYwYdS)zw Get the bot's ongoing games. If an error occurs when retrieving the games, None is returned. r nowPlayingN) rrrr dictrBlistrr)r1rFs r2get_ongoing_gameszLichess.get_ongoing_gamess   + + * *Dd8n!45t7H7H7S7STTHL) * * * * * * * * * * * * * * * *tsA A44A8;A8c2|d|dS)zResign a game.r Nrrs r2r zLichess.resigns h(((((r4rc|jdd|jd|i|jj|jdS)z6Set the user agent for communication with lichess.org.z User-Agentz lichess-bot/z user:N)rrryrr)r1rs r2rzLichess.set_user_agentsQ L*W*W*WX*W*WXYYY ##DK00000r4cR |d|S#t$rYdSwxYw)z6Get the PGN (Portable Game Notation) record of a game.r!rQ)rrrs r2 get_game_pgnzLichess.get_game_pgns? ##Hg66 6   22 s  &&c  |d}ttt|d}tt t j|S#t$rgcYSwxYw)z#Get a list of bots that are online.r" ) rrfilterboolrmaprSloadsr)r1online_bots_strr"s r2get_online_botszLichess.get_online_botss} "..}==OvdO,A,A$,G,GHHIIKDJ 4455 5   III sA0A33 BBc Ztt|d||dS)zCreate a challenge.r#F)rr)r rr)r1rrs r2r#zLichess.challenges1MMM+x[`Maacc cr4c6|d|ddS)zCancel a challenge.r$F)rNrrs r2r$zLichess.cancels  h u EEEEEr4)r authenticatedpathrc tjtjttt t fdjdtttj tj  dtffd }|S)zBGet an external move from online sources (chessdb or lichess.org).rr)r max_triesrrrrrr+c~rjnj}|d}|S)Nr)r>rr)rrrTrS)rrrrrr1rs r2online_book_getz0Lichess.online_book_get..online_book_getsD'4Kdll9KG(/ D!F[a (b(b(g(g(i(iM r4) r on_exceptionconstantrRequestsConnectionErrorrrr{rirtrDEBUGr)r1rrrrr s````` r2r zLichess.online_book_gets  g.13JIWbc')(,(8'*%-)807 /6} > > > ! ! ! ! ! ! ! ! ! ! > > !   r4user_idc|dd|i}t|o|ddS)z5Check if lichess.org thinks the bot is online or not.r%idsrronline)rrrT)r1rusers r2 is_onlinezLichess.is_onlinesB  5'2B CCD2T!W[[22333r4 user_namecTtt|d|S)zGet the public data of a bot.r&)r rr)r1rs r2get_public_datazLichess.get_public_datas"ND$5$5mY$O$OPPPr4r5)r)N)Br6r7r8r9rBintr3rr r rrrrrirtrrrrrUResponserrr rrrrrrrrrrmodelsrrCrDrrrrchessengine PlayResultrrrrrrrrrrr rrrr#r$rr rrr:r4r2rvrv{s$LL$Dc$D$Dc$D#$D\_$Ddh$D$D$D$DLW'*-/F S^_#%#&!)%4,3M+2=:::15$QS#S#X-/2;C;L::66:   # s !#s(^d2 (?:T#tH~BU=VV     6:   # s !#s(^d2 >B?>S    59    c  cNT1 =@    W'*-/F S^_#%#&!)%4,3M+2=:::6:26159=*.### ##!$#T#s(^+d2#sCx.4/ # c3h$. # 0$6 #$(#4ACS3SVZ3Z###::#J s s     ")A "m " " " "B#B8CUBZ^BBBBFSFTFFFFM#M(:LMMMM!!!!OOEL,COOOOO s D T    2C2s2#2$2222 (S(T((((E(/":EEEEHsHx/GHHHH.S.T....22c232t2222_ 4>D#8    )c)d))))1s1t1111 CCo!6c#c0Ec-cccc F3F4FFFF!',E!!!C!c39n1E1L! $!=A!NX!!!!$444444 QQQQQQQQr4rv);r9rSrU urllib.parserrequests.exceptionsrrrr http.clientrrrrr collectionsrrCr lib.timerr r r typingr chess.enginerlib.lichess_typesr rrrrrrrrrr6rorr*rhrr<rrrrKrrYr\r_rDrdrrirtrvr:r4r2r's!  bbbbbbbbbb******######----------HHHHHHHHHHHHHHHHHHHH  ~  %  ' '    &   .   #  %  ( * )  '   $ $   (! " !# $"#'    .  8 $ $ tvv|' 8'T'''' x'? Y\ ae    3X_%=3$3333 3 83T3333 m@RUY@Y  d    9^99999_Q_Q_Q_Q_Q_Q_Q_Q_Q_Qr4