U h@szddlZddlZddlZddlmZddlmZmZddlm Z ddZ Gdd d ej Z Gd d d e ej Z d d ZdS)N)utils)TLObjecttypes) ChatGetterc s|dkr dSt|s|f}t}|D]}t|tr~|dkrH||q|tt |tt |tt |hq&t|t r|j dkr|t|q&||IdH}t|tjr|jddIdH}|t|q&|S)z>Helper util to turn the input chat or chats into a set of IDs.NriVT)Z input_peer)rZ is_list_likeset isinstanceintaddupdateZ get_peer_idrZPeerUserZPeerChatZ PeerChannelrZSUBCLASS_OF_IDZget_input_entityZ InputPeerSelfZget_me)clientchatsresultZchatr:/tmp/pip-unpacked-wheel-c81u5j2r/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)r boolrresolvedr _resolve_lock)selfr rrrrr__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 Zothersself_idrrrbuildKszEventBuilder.buildc sZ|jr dS|jst|_|j4IdH"|jsF||IdHd|_W5QIdHRXdS)zAHelper method to allow event builders to be resolved before usageNT)rrasyncioLock_resolverr rrrresolveYs zEventBuilder.resolvecst||jIdH|_dSN)rr r rrrrfszEventBuilder._resolvecCsB|js dS|jdk r.|j|jk}||jkr.dS|js8dS||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)rr chat_idrr)reventZinsiderrrfilteris   zEventBuilder.filter)N)NN) __name__ __module__ __qualname____doc__r classmethodabcabstractmethodrr!rr%rrrrr)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_clientZ _message_idZoriginal_update)rZ chat_peerZmsg_idr/ __class__rrrs zEventCommon.__init__cCs:||_|jr*t|j|j|j\|_|_n d|_|_dS)zR Setter so subclasses can act accordingly when the client is set. N) r2Z _chat_peerrZ_get_entity_pairr#r1Z_mb_entity_cacheZ_chatZ _input_chatr rrr _set_clientszEventCommon._set_clientcCs|jS)zH The `telethon.TelegramClient` that created this event. )r2rrrrr szEventCommon.clientcCst|Sr"rZ pretty_formatto_dictr6rrr__str__szEventCommon.__str__cCstj|ddS)Nr)indentr7r6rrr stringifyszEventCommon.stringifycCs"dd|jD}|j|d<|S)NcSs"i|]\}}|ddkr||qS)r_r).0kvrrr s z'EventCommon.to_dict..r<)__dict__items _event_name)rdrrrr8s zEventCommon.to_dict)NNN) r&r'r(r)rCrr5propertyr r9r;r8 __classcell__rrr3rr-s   r-cCs0t|drd|j|j_ntd||S)z4Decorator to rename cls.Event 'Event' as 'cls.Event'r.z{}.Eventz$Class {} does not have a inner Event)hasattrformatr&r.rCwarningswarn)rrrrname_inner_events rK)r+rrIrtlrrZtl.custom.chatgetterrrABCrr-rKrrrrs  Y2