o Xzj| @sJddlZddlZddlZddlZddlZddlZddlZddlZddlZddl Z ddl m Z z ddl Z e j Z WneySZzdZ GdddeZ WYdZ[ndZ[wwddlmZmZmZddlmZmZddlmZmZdd lmZmZdd lmZdd l m!Z!ej"rd d l#m$Z$ej%ej&gej&fZ'GdddZ(GdddZ)dS)N)dequec@s eZdZdS)OperationalErrorN)__name__ __module__ __qualname__rrR/var/www/kodo/tsheker/venv/lib/python3.10/site-packages/telethon/client/updates.pyrsr)eventsutilserrors) EventBuilder EventCommon)types functions)GapErrorPrematureEndReason)get_running_loop) __version__)TelegramClientc @seZdZd$ddZd$ddZd$ddZddd efd d Z d%ddd ed efddZ d%ddd ed ede fddZ  d&ddZ d$ddZ d$ddZddZd$ddZd$ddZd$d d!Zd$d"d#Zd S)' UpdateMethodsselfrcsz7z#|tjIdH|jIdH}|jdur|j|WW|IdHSty.YnwW|IdHdS|IdHwN)rupdatesGetStateRequest disconnected_updates_error disconnectKeyboardInterrupt)rresultrrr_run_until_disconnected)s   "z%UpdateMethods._run_until_disconnectedcs*| |_|r|tjIdHdSdS)z Change the value of `receive_updates`. This is an `async` method, because in order for Telegram to start sending updates again, a request must be made. N) _no_updatesrrr)rreceive_updatesrrrset_receive_updates6s z!UpdateMethods.set_receive_updatescCs\|jr |Szz|j|WW|Sty!YnwW|dS|w)a Runs the event loop until the library is disconnected. It also notifies Telegram that we want to receive updates as described in https://core.telegram.org/api/updates. If an unexpected error occurs during update handling, the client will disconnect and said error will be raised. Manual disconnections can be made by calling `disconnect() ` or sending a ``KeyboardInterrupt`` (e.g. by pressing ``Ctrl+C`` on the console window running the script). If a disconnection error occurs (i.e. the library fails to reconnect automatically), said error will be raised through here, so you have a chance to ``except`` it on your own code. If the loop is already running, this method returns a coroutine that you should await on your own code. .. note:: If you want to handle ``KeyboardInterrupt`` in your code, simply run the event loop in your code too in any way, such as ``loop.run_forever()`` or ``await client.disconnected`` (e.g. ``loop.run_until_complete(client.disconnected)``). Example .. code-block:: python # Blocks the current task here until a disconnection occurs. # # You will still receive updates, since this prevents the # script from exiting. await client.run_until_disconnected() N)loop is_runningr!run_until_completerrrrrrrun_until_disconnectedAs %  z$UpdateMethods.run_until_disconnectedeventcsfdd}|S)a7 Decorator used to `add_event_handler` more conveniently. Arguments event (`_EventBuilder` | `type`): The event builder class or instance to be used, for instance ``events.NewMessage``. Example .. code-block:: python from telethon import TelegramClient, events client = TelegramClient(...) # Here we use client.on @client.on(events.NewMessage) async def handler(event): ... cs||Sr)add_event_handler)fr*rrr decorators z#UpdateMethods.on..decoratorr)rr*r.rr-ronpszUpdateMethods.onNcallbackcCsbt|}|dur|D] }|j||fq dSt|tr!|}n|s't}|j||fdS)aS Registers a new event handler callback. The callback will be called when the specified event occurs. Arguments callback (`callable`): The callable function accepting one parameter to be used. Note that if you have used `telethon.events.register` in the callback, ``event`` will be ignored, and instead the events you previously registered will be used. event (`_EventBuilder` | `type`, optional): The event builder class or instance to be used, for instance ``events.NewMessage``. If left unspecified, `telethon.events.raw.Raw` (the :tl:`Update` objects with no further processing) will be passed instead. Example .. code-block:: python from telethon import TelegramClient, events client = TelegramClient(...) async def handler(event): ... client.add_event_handler(handler, events.NewMessage) N)r _get_handlers_event_buildersappend isinstancetypeRaw)rr0r*buildersrrrr+s $ zUpdateMethods.add_event_handlerreturncCsld}|r t|ts t|}t|j}|r4|d8}|j|\}}||kr2|r*t||r2|j|=|d7}|s|S)a Inverse operation of `add_event_handler()`. If no event is given, all events for this callback are removed. Returns how many callbacks were removed. Example .. code-block:: python @client.on(events.Raw) @client.on(events.NewMessage) async def handler(event): ... # Removes only the "Raw" handling # "handler" will still receive "events.NewMessage" client.remove_event_handler(handler, events.Raw) # "handler" will stop receiving anything client.remove_event_handler(handler) rr)r4r5lenr2)rr0r*foundievcbrrrremove_event_handlers z"UpdateMethods.remove_event_handler5typing.Sequence[typing.Tuple[Callback, EventBuilder]]cCsdd|jDS)a Lists all registered event handlers. Returns A list of pairs consisting of ``(callback, event)``. Example .. code-block:: python @client.on(events.NewMessage(pattern='hello')) async def on_greeting(event): '''Greets someone''' await event.reply('Hi') for callback, event in client.list_event_handlers(): print(id(callback), type(event)) cSsg|]\}}||fqSrr).0r*r0rrr z5UpdateMethods.list_event_handlers..)r2r(rrrlist_event_handlerssz!UpdateMethods.list_event_handlerscs|jtIdHdS)a "Catches up" on the missed updates while the client was offline. You should call this method after registering the event handlers so that the updates it loads can by processed by your script. This can also be used to forcibly fetch new updates if there are any. Example .. code-block:: python await client.catch_up() N)_updates_queueputrUpdatesTooLongr(rrrcatch_ups zUpdateMethods.catch_upc sjdup j }d_z~jrIdHt}r|rPjr3 | IdHn|rOj | }j ||j j|s5qtjjkrjtdtjjIdHjfddtjjkrtdjtdtjjj}|rjtdz |IdH}Wntjtjtj t!fy}zjtdt"|jj#WYd}~qd}~wtj$tj%fy}z,jtd t"|jj#|r|_&IdHWYd}~WdSWYd}~qd}~wtj't(fy@}z jt)d |j#|_&IdHWYd}~WdSd}~wtj*ye}zjt)d t+d |j#WYd}~qd}~wt,y}zjtd t"|j|t-.dIdHWYd}~qd}~wwj/|j\}}} |rjtd0||| IdH} |1| qj2j}|r(jtd|j3j4z |IdH}Wn+tj$tj%fy}z4jt)d|j3j4t"|jj5|t6j7j|r |_&IdHWYd}~WdSWYd}~qd}~wtj't(fyN}z(jt)d|j3j4|j5|t6j7j|_&IdHWYd}~WdSd}~wtj8tj9tjtjtj t!fy}zjt)d|j3j4t"|jj5|t6j7jWYd}~qd}~wtj:tj;fyjtd|j3j4j5|t6j<jYqtj*y}z jt)dt+d |j3j4|j5|t6j7jWYd}~qd}~wt,y}zjtd|j3j4t"|j|t-.dIdHWYd}~qd}~wwj=||j\}}} |rjtd|j3j40||| IdH} |1| qj>} | t?@} | dkr[zt-AjBC| IdH}Wnt-jDyZjtdYqwqg} z jE|j| \}} Wn tFyuYqw0| || IdH} |1| s#WdSWdSt-jGyYdStHy}zjtIdt+d|_&IdHWYd}~dSd}~ww)NTzAIn-memory entity cache limit reached (%s/%s), flushing to sessioncs|jjkp |jjvSr)_mb_entity_cacheself_id _message_boxmap)idr(rr.rBz,UpdateMethods._update_loop..z\in-memory entities exceed entity_cache_limit after flushing; consider setting a larger limitz9In-memory entity cache at %s/%s after flushing to sessionz&Getting difference for account updatesz9Cannot get difference since Telegram is having issues: %szsz5UpdateMethods._preprocess_updates..) rHrsr maybe_asyncsessionprocess_entitiesrcontacts ResolvedPeer itertoolschain _entities)rrrrentitiesurrrrrs$ z!UpdateMethods._preprocess_updatesc sdd}|roz tj|jddIdHWqtjy Yntjy*YdSty2Yqw|IdH|j s@qz |j |Wn t tjfyVYdSw| IdHt |jIdH|s dSdS)NcSs tddS)Nll)random randrangerrrrrMs z/UpdateMethods._keepalive_loop..<)timeout)rRrorrrrr_clean_exported_senders_sender_transport_connected_keepalive_pingConnectionErrorr^r rrsave)rrndrrr_keepalive_loops4   zUpdateMethods._keepalive_loopc sd}|jjsz |jddIdHWn tyYnwt|||}|jD]6}|D]1}|tj}|r9| ||tj }|rE| ||tj }|rQ| ||jr\||IdHq+q'|jD]\}}|t|} | snqa|jsy||IdH|| } t| r| IdH} | sqaz || IdHWqatjyt|dt|} |jtd| Yqatjyt|dt|} |jtd| t| jYdSty} z!t | t!j"r|#rt|dt|} |jt$d| WYd} ~ qad} ~ wwdS)NT) input_peerrEEvent handler "%s" already has an open conversation, ignoring new one=Event handler "%s" stopped chain of propagation for event %s.Unhandled exception on %s)%rHrIget_mernEventBuilderDict_conversationsvaluesr NewMessage_on_new_message MessageEdited_on_edit MessageRead_on_read_custom _check_customr2r5resolvedresolvefilterinspect isawaitabler AlreadyInConversationErrorgetattrreprr\rrcStopPropagationrr4rorrRr) rupdateothersbuiltconv_setconvr<builderr0r*rnamerrrrrT,sr               zUpdateMethods._dispatch_updatec sL|jD]\}}t|tjrqt||jsq|js!||IdH||}t |r0|IdH}|s3qz ||IdHWqt j yWt |dt |}|jtd|Yqtjywt |dt |}|jtd|t|jYdSty}z!t|tjr|rt |dt |}|jtd|WYd}~qd}~wwdS)zO Dispatches a single, out-of-order event. Used by `AlbumHack`. Nrrrr)r2r4r r6Eventrrrrrr rrrr\rrcrr5rrorrRr)rr*rr0rrrrrr_dispatch_eventosF        zUpdateMethods._dispatch_eventc sVz |IdHWdSty*}z|jtdt||WYd}~dSd}~ww)Nz:Error executing high-level request after reconnect: %s: %s) rrr\rrlr5r]rGr rmr)rrrrr_handle_auto_reconnects z$UpdateMethods._handle_auto_reconnect)rrr)rrr8r?)rrrr!r$r)r r/Callbackr+intr>rCrGrrrrrTrrrrrrr%sD  / 4 '  q ( C)rc@s"eZdZdZd ddZddZdS) rzI Helper "dictionary" to return events from types and cache them. clientrcCs||_||_||_dSr)rrr)rrrrrrr__init__s zEventBuilderDict.__init__cCsz|j|WSty?||j|j|jj}|j|<t|tr5|j|_ |jj |_ | |j|YS|r;|j|_ |YSwr) __dict__KeyErrorbuildrrr_self_idr4roriginal_updater _set_client_client)rrr*rrr __getitem__s     zEventBuilderDict.__getitem__N)rr)rrr__doc__rrrrrrrs  r)*rorrrsysr tracebacktypingloggingr` collectionsrsqlite3r ImportErrorrrr r r events.commonr rtlrr_updatesrrhelpersrversionr TYPE_CHECKINGtelegramclientrCallableAnyrrrrrrrsF     &