3 uQhd%@svddlZddlZddlZddlmZddlmZmZddlm Z m Z ej rVddl m Z Gdd d ZGd d d ZdS) N)_NOT_A_REQUEST)helpersutils) functions TLRequest)TelegramClientcseZdZdZdZddZedd Zejd d Zd d Z d dZ e j Z e jZdddZfddZddZfddZZS)_TakeoutClientz' Proxy object over the client. __enter____exit__ __aenter__ __aexit__cCs||_||_||_d|_dS)N)_TakeoutClient__finalize_TakeoutClient__client_TakeoutClient__request_TakeoutClient__success)selffinalizeclientrequestr;/tmp/pip-build-2nz6shyl/telethon/telethon/client/account.py__init__sz_TakeoutClient.__init__cCs|jS)N)r)rrrrsuccesssz_TakeoutClient.successcCs ||_dS)N)r)rvaluerrrr"scs@|j}|jjdkr*||jIdHj|j_n|jdk rW|j|r|dn||dIdHS)NzJTakeout mode has not been initialized (are you calling outside of "with"?)r)ordered) rrrrrZ is_list_like isinstancerrresolveappendrZInvokeWithTakeoutRequest)rrr%rZsinglerequestswrappedrrrr__call__>s    z_TakeoutClient.__call__cs(|jdr|t|jkrttj|S)N__) startswithtype_TakeoutClient__PROXY_INTERFACEAttributeErrorsuper__getattribute__)rname) __class__rrr3Psz_TakeoutClient.__getattribute__cCs0t|j|}tj|r,tjt|jj||S|S)N)getattrrinspectismethod functoolspartialr5)rr4rrrr __getattr__]s   z_TakeoutClient.__getattr__cs8|jdjt|jjdr*tj||St|j||S)Nz_{}___) r.formatr/__name__lstripr2 __setattr__setattrr)rr4r)r5rrr@gsz_TakeoutClient.__setattr__)r r r r)F)r> __module__ __qualname____doc__r0rpropertyrsetterr rrZ _sync_enterr Z _sync_exitr r,r3r;r@ __classcell__rr)r5rr s      r c @sReZdZd dddddddddeeeeeeeedd ddZdeedd d ZdS) AccountMethodsTN)contactsuserschats megagroupschannelsfiles max_file_sizer ) rrrIrJrKrLrMrNrOreturnc Cs\t|||||||d} dd| jD} |jjdks`. 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') )rIZ message_usersZ message_chatsZmessage_megagroupsZmessage_channelsrNZ file_max_sizecss|]}|dk VqdS)Nr).0argrrr sz)AccountMethods.takeout..N) dictvaluesrranyrr ZInitTakeoutSessionRequestr ) rrrIrJrKrLrMrNrOZrequest_kwargsZ arg_specifiedrrrrtakeoutos\zAccountMethods.takeout)rrrPcsJy0td|d4IdH }||_WdQIdHRXWntk rDdSXdS)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)rrrWrrr end_takeouts zAccountMethods.end_takeout)T)r>rBrCboolrWrXrrrrrHns&erH)r9r7typingrJrrrtlrrZ TYPE_CHECKINGZtelegramclientr r rHrrrrs  _