a fid%@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)selffinalizeclientrequestrL/home/ec2-user/.local/lib/python3.9/site-packages/telethon/client/account.py__init__sz_TakeoutClient.__init__cCs|jSrr)rrrrsuccesssz_TakeoutClient.successcCs ||_dSrr)rvaluerrrr"scs@|j}|jjdur*||jIdHj|j_n|jdurs   z_TakeoutClient.__call__cs(|dr|t|jvrtt|S)N__) startswithtype_TakeoutClient__PROXY_INTERFACEAttributeErrorsuper__getattribute__)rname __class__rrr6Psz_TakeoutClient.__getattribute__cCs0t|j|}t|r,tt|jj||S|Sr)getattrrinspectismethod functoolspartialr9rr7rrrr __getattr__]s   z_TakeoutClient.__getattr__cs8|dt|jdr*t||St|j||S)Nz_{}___) r1formatr2__name__lstripr5 __setattr__setattrrr?r8rrrEgsz_TakeoutClient.__setattr__)F)rC __module__ __qualname____doc__r3rpropertyrsetterr rrZ _sync_enterr Z _sync_exitr r/r6r@rE __classcell__rrr8rr s        r c @sReZdZd dddddddddeeeeeeeedd ddZdeedd d ZdS) AccountMethodsTN)contactsuserschats megagroupschannelsfiles max_file_sizer ) rrrNrOrPrQrRrSrTreturnc Cs`t|||||||d} dd| D} |jjdus`. 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') )rNZ message_usersZ message_chatsZmessage_megagroupsZmessage_channelsrSZ file_max_sizecss|]}|duVqdSrr).0argrrr z)AccountMethods.takeout..N) dictvaluesrranyrr"ZInitTakeoutSessionRequestr ) rrrNrOrPrQrRrSrTZrequest_kwargsZ arg_specifiedrrrrtakeoutos \  zAccountMethods.takeout)rrrUc sdzJtd|d4IdH}||_WdIdHqH1IdHs>0YWnty^YdS0dS)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 4 zAccountMethods.end_takeout)T)rCrGrHboolr]r^rrrrrMns* orM)r=r;typingrOrrrtlrr TYPE_CHECKINGZtelegramclientr r rMrrrrs  _