o Xzj@szddlZddlZddlZddlmZddlmZmZddlm Z ddZ Gdd d ej Z Gd d d e ej Z d d ZdS)N)utils)TLObjecttypes) ChatGetterc s|durdSt|s|f}t}|D]^}t|tr@|dkr%||q|tt |tt |tt |hqt|t rS|j dkrS|t|q||IdH}t|tjrj|jddIdH}|t|q|S)z>Helper util to turn the input chat or chats into a set of IDs.NriVT) input_peer)r is_list_likeset isinstanceintaddupdate get_peer_idrPeerUserPeerChat PeerChannelrSUBCLASS_OF_IDget_input_entity InputPeerSelfget_me)clientchatsresultchatrQ/var/www/kodo/tsheker/venv/lib/python3.10/site-packages/telethon/events/common.py _into_id_set s,    rc@sNeZdZdZddddddZeejdddZd d Z d d Z d dZ dS) EventBuilderaH The common event builder, with builtin support to filter per chat. Args: chats (`entity`, optional): May be one or more entities (username/peer/etc.), preferably IDs. By default, only matching chats will be handled. blacklist_chats (`bool`, optional): Whether to treat the chats as a blacklist instead of as a whitelist (default). This means that every chat will be handled *except* those specified in ``chats`` which will be ignored if ``blacklist_chats=True``. func (`callable`, optional): A callable (async or not) function that should accept the event as input parameter, and return a value indicating whether the event should be dispatched or not (any truthy value will do, it does not need to be a `bool`). It works like a custom filter: .. code-block:: python @client.on(events.NewMessage(func=lambda e: e.is_private)) async def handler(event): pass # code here NF)blacklist_chatsfunccCs&||_t||_d|_||_d|_dS)NF)rboolrresolvedr _resolve_lock)selfrrrrrr__init__Ds   zEventBuilder.__init__cCsdS)aO Builds an event for the given update if possible, or returns None. `others` are the rest of updates that came in the same container as the current `update`. `self_id` should be the current user's ID, since it is required for some events which lack this information but still need it. Nr)clsr othersself_idrrrbuildKszEventBuilder.buildc sx|jrdS|jst|_|j4IdH|js$||IdHd|_WdIdHdS1IdHs5wYdS)zAHelper method to allow event builders to be resolved before usageNT)r!r"asyncioLock_resolver#rrrrresolveYs .zEventBuilder.resolvecst||jIdH|_dSN)rrr,rrrr+fszEventBuilder._resolvecCsB|jsdS|jdur|j|jv}||jkrdS|jsdS||S)z Returns a truthy value if the event passed the filter and should be used, or falsy otherwise. The return value may need to be awaited. The events must have been resolved before this can be called. NT)r!rchat_idrr)r#eventinsiderrrfilteris    zEventBuilder.filterr.)NN) __name__ __module__ __qualname____doc__r$ classmethodabcabstractmethodr(r-r+r2rrrrr)s  rcsReZdZdZdZdfdd ZddZedd Zd d Z d d Z ddZ Z S) EventCommonaR Intermediate class with common things to all events. Remember that this class implements `ChatGetter ` which means you have access to all chat properties and methods. In addition, you can access the `original_update` field which contains the original :tl:`Update`. EventNcs,tj||di|_d|_||_d|_dS)N) broadcast)superr$ _entities_client _message_idoriginal_update)r# chat_peermsg_idr< __class__rrr$s  zEventCommon.__init__cCs<||_|jrt|j|j|j\|_|_dSd|_|_dS)zR Setter so subclasses can act accordingly when the client is set. N) r? _chat_peerr_get_entity_pairr/r>_mb_entity_cache_chat _input_chatr,rrr _set_clients  zEventCommon._set_clientcCs|jS)zH The `telethon.TelegramClient` that created this event. )r?r#rrrrszEventCommon.clientcCst|Sr.r pretty_formatto_dictrLrrr__str__szEventCommon.__str__cCstj|ddS)Nr)indentrMrLrrr stringifyszEventCommon.stringifycCs"dd|jD}|j|d<|S)NcSs"i|] \}}|ddkr||qS)r_r).0kvrrr s"z'EventCommon.to_dict..rS)__dict__items _event_name)r#drrrrOs zEventCommon.to_dict)NNN) r3r4r5r6rZr$rKpropertyrrPrRrO __classcell__rrrDrr:s  r:cCs2t|drd|j|j_|Std||S)z4Decorator to rename cls.Event 'Event' as 'cls.Event'r;z{}.Eventz$Class {} does not have a inner Event)hasattrformatr3r;rZwarningswarn)r%rrrname_inner_events rb)r8r)r`rtlrrtl.custom.chatgetterrrABCrr:rbrrrrs  Y 2