o Xzjd%@svddlZddlZddlZddlmZddlmZmZddlm Z m Z ej r+ddl m Z Gdd d ZGd d d ZdS) N)_NOT_A_REQUEST)helpersutils) functions TLRequest)TelegramClientcseZdZdZdZddZeddZejddZdd Z d d Z e j Z e jZdd dZfddZddZfddZZS)_TakeoutClientz' Proxy object over the client. ) __enter____exit__ __aenter__ __aexit__cCs||_||_||_d|_dSN)_TakeoutClient__finalize_TakeoutClient__client_TakeoutClient__request_TakeoutClient__success)selffinalizeclientrequestrR/var/www/kodo/tsheker/venv/lib/python3.10/site-packages/telethon/client/account.py__init__s z_TakeoutClient.__init__cCs|jSrr)rrrrsuccesssz_TakeoutClient.successcCs ||_dSrr)rvaluerrrr"s csD|j}|jjdur||jIdHj|j_|S|jdur td|S)NzgCan't send a takeout request while another takeout for the current session still not been finished yet.)rsession takeout_idrid ValueError)rrrrrr &s  z_TakeoutClient.__aenter__csZ|jdur|jr|du|_|jdur+|tj|jIdH}|s%tdd|j_dSdS)NzFailed to finish the takeout.)rrraccountFinishTakeoutSessionRequestr!rr)rexc_type exc_value tracebackresultrrrr0s    z_TakeoutClient.__aexit__Fcs|jjj}|durtdt| }|r|fn|}g}|D]}t|ts)t| |tIdH| t ||q|j|rD|dn||dIdHS)NzJTakeout mode has not been initialized (are you calling outside of "with"?)r)ordered) rrrr!r is_list_like isinstancerrresolveappendrInvokeWithTakeoutRequest)rrr(rsinglerequestswrappedrrrr__call__>s    z_TakeoutClient.__call__cs(|dr|t|jvrtt|S)N__) startswithtype_TakeoutClient__PROXY_INTERFACEAttributeErrorsuper__getattribute__)rname __class__rrr9Ps z_TakeoutClient.__getattribute__cCs0t|j|}t|rtt|jj||S|Sr)getattrrinspectismethod functoolspartialr<rr:rrrr __getattr__]s  z_TakeoutClient.__getattr__cs8|dt|jdrt||St|j||S)Nz_{}___) r4formatr5__name__lstripr8 __setattr__setattrrrBr;rrrHgsz_TakeoutClient.__setattr__)F)rF __module__ __qualname____doc__r6rpropertyrsetterr rr _sync_enterr _sync_exitr r2r9rCrH __classcell__rrr;rr s         r c@sjeZdZ ddddddddddddededed ed ed ed ed eddfddZdddedefddZdS)AccountMethodsTN)contactsuserschats megagroupschannelsfiles max_file_sizerr rrSrTrUrVrWrXrYreturnc Cs`t|||||||d} dd| D} |jjdust| r(tjjdi| } nd} t||| S)a~ Returns a :ref:`telethon-client` which calls methods behind a takeout session. It does so by creating a proxy object over the current client through which making requests will use :tl:`InvokeWithTakeoutRequest` to wrap them. In other words, returns the current client modified so that requests are done as a takeout: Some of the calls made through the takeout session will have lower flood limits. This is useful if you want to export the data from conversations or mass-download media, since the rate limits will be lower. Only some requests will be affected, and you will need to adjust the `wait_time` of methods like `client.iter_messages `. By default, all parameters are `None`, and you need to enable those you plan to use by setting them to either `True` or `False`. You should ``except errors.TakeoutInitDelayError as e``, since this exception will raise depending on the condition of the session. You can then access ``e.seconds`` to know how long you should wait for before calling the method again. There's also a `success` property available in the takeout proxy object, so from the `with` body you can set the boolean result that will be sent back to Telegram. But if it's left `None` as by default, then the action is based on the `finalize` parameter. If it's `True` then the takeout will be finished, and if no exception occurred during it, then `True` will be considered as a result. Otherwise, the takeout will not be finished and its ID will be preserved for future usage as `client.session.takeout_id `. Arguments finalize (`bool`): Whether the takeout session should be finalized upon exit or not. contacts (`bool`): Set to `True` if you plan on downloading contacts. users (`bool`): Set to `True` if you plan on downloading information from users and their private conversations with you. chats (`bool`): Set to `True` if you plan on downloading information from small group chats, such as messages and media. megagroups (`bool`): Set to `True` if you plan on downloading information from megagroups (channels), such as messages and media. channels (`bool`): Set to `True` if you plan on downloading information from broadcast channels, such as messages and media. files (`bool`): Set to `True` if you plan on downloading media and you don't only wish to export messages. max_file_size (`int`): The maximum file size, in bytes, that you plan to download for each message with media. Example .. code-block:: python from telethon import errors try: async with client.takeout() as takeout: await client.get_messages('me') # normal call await takeout.get_messages('me') # wrapped through takeout (less limits) async for message in takeout.iter_messages(chat, wait_time=0): ... # Do something with the message except errors.TakeoutInitDelayError as e: print('Must wait', e.seconds, 'before takeout') )rS message_users message_chatsmessage_megagroupsmessage_channelsrX file_max_sizecss|]}|duVqdSrr).0argrrr sz)AccountMethods.takeout..Nr) dictvaluesrranyrr"InitTakeoutSessionRequestr ) rrrSrTrUrVrWrXrYrequest_kwargs arg_specifiedrrrrtakeoutos \   zAccountMethods.takeoutrc shz(td|d4IdH}||_WdIdHWdS1IdHs"wYWdSty3YdSw)ap Finishes the current takeout session. Arguments success (`bool`): Whether the takeout completed successfully or not. Returns `True` if the operation was successful, `False` otherwise. Example .. code-block:: python await client.end_takeout(success=False) TNF)r rr!)rrrirrr end_takeouts zAccountMethods.end_takeout)T)rFrJrKboolrirjrrrrrRns@   orR)r@r>typingrTrrrtlrr TYPE_CHECKINGtelegramclientr r rRrrrrs  _