U hd%@svddlZddlZddlZddlmZddlmZmZddlm Z m Z ej rVddl 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)selffinalizeclientrequestr;/tmp/pip-unpacked-wheel-c81u5j2r/telethon/client/account.py__init__sz_TakeoutClient.__init__cCs|jSrr)rrrrsuccesssz_TakeoutClient.successcCs ||_dSrr)rvaluerrrr"scs@|j}|jjdkr*||jIdHj|j_n|jdk rs   z_TakeoutClient.__call__cs(|dr|t|jkrtt|S)N__) startswithtype_TakeoutClient__PROXY_INTERFACEAttributeErrorsuper__getattribute__)rname __class__rrr5Psz_TakeoutClient.__getattribute__cCs0t|j|}t|r,tt|jj||S|Sr)getattrrinspectismethod functoolspartialr8rr6rrrr __getattr__]s   z_TakeoutClient.__getattr__cs8|dt|jdr*t||St|j||S)Nz_{}___) r0formatr1__name__lstripr4 __setattr__setattrrr>r7rrrDgsz_TakeoutClient.__setattr__)F)rB __module__ __qualname____doc__r2rpropertyrsetterr rrZ _sync_enterr Z _sync_exitr r.r5r?rD __classcell__rrr7rr s        r c @sReZdZd dddddddddeeeeeeeedd ddZdeedd d ZdS) AccountMethodsTN)contactsuserschats megagroupschannelsfiles max_file_sizer ) rrrMrNrOrPrQrRrSreturnc Cs\t|||||||d} dd| D} |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') )rMZ message_usersZ message_chatsZmessage_megagroupsZmessage_channelsrRZ file_max_sizecss|]}|dk VqdSrr).0argrrr sz)AccountMethods.takeout..N) dictvaluesrranyrr"ZInitTakeoutSessionRequestr ) rrrMrNrOrPrQrRrSZrequest_kwargsZ arg_specifiedrrrrtakeoutos \ zAccountMethods.takeout)rrrTc sLz0td|d4IdH }||_W5QIdHRXWntk rFYdSXdS)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!)rrr[rrr end_takeouts zAccountMethods.end_takeout)T)rBrFrGboolr[r\rrrrrLns, orL)r<r:typingrNrrrtlrr TYPE_CHECKINGZtelegramclientr r rLrrrrs  _