o XzjZ2@sddlZddlZddlZddlmZmZmZddlmZddl m Z ddl m Z dZ d ZiZd ZGd d d ZeGd ddeZdS)N) EventBuilder EventCommonname_inner_event)utils)types) SenderGetterdg?c@s(eZdZdZddZddZddZdS) AlbumHacka When receiving an album from a different data-center, they will come in separate `Updates`, so we need to temporarily remember them for a while and only after produce the event. Of course events are not designed for this kind of wizardy, so this is a dirty hack that gets the job done. When cleaning up the code base we may want to figure out a better way to do this, or just leave the album problem to the users; the update handling code is bad enough as it is. cCs6t||_||_|jt|_|j| dSN) weakrefref_client_eventlooptime _HACK_DELAY_due create_task deliver_event)selfclienteventrP/var/www/kodo/tsheker/venv/lib/python3.10/site-packages/telethon/events/album.py__init__#s zAlbumHack.__init__cCs2|}|r|jj||jt|_dSdSr )rrmessagesextendrrrr)rrrrrrr,s zAlbumHack.extendcsZ |}|dur dS|j|j}|dkr#||jIdHdS~t|IdHq)NTr)rrrr_dispatch_eventrasynciosleep)rrdiffrrrr2szAlbumHack.deliver_eventN)__name__ __module__ __qualname____doc__rrrrrrrr s   r csXeZdZdZ d dddfdd ZedddZfd d ZGd d d ee Z Z S)Albuma Occurs whenever you receive an album. This event only exists to ease dealing with an unknown amount of messages that belong to the same album. Example .. code-block:: python from telethon import events @client.on(events.Album) async def handler(event): # Counting how many photos or videos the album has print('Got an album with', len(event), 'items') # Forwarding the album as a whole to some chat event.forward_to(chat) # Printing the caption print(event.text) # Replying to the fifth item in the album await event.messages[4].reply('Cool!') NFblacklist_chatsfunccstj|||ddS)Nr))superr)rchatsr*r+ __class__rrr^szAlbum.__init__cs|g}t|tjtjfr_t|jtjsdS|jjdurdStt |dr)dSt t tt krDfddt DD]}t|=q>|D] }||urRtt |<qF|fdd|DSdS)Ncs g|] \}}|tkr|qSr)_IGNORE_MAX_AGE).0it)nowrr |s zAlbum.build..cs>g|]}t|tjtjfrt|jtjr|jjkr|jqSr) isinstancerUpdateNewMessageUpdateNewChannelMessagemessageMessage grouped_id)r1u)grouprrr5s  )r6rr7r8r9r:r; _IGNORE_DICTpopidrlen_IGNORE_MAX_SIZEitemsEvent)clsupdateothersself_idr2r<r)r=r4rbuildbs.   z Album.buildcst|jdkr t|SdSNr)rArr,filter)rrr.rrrKs z Album.filtercseZdZdZfddZfddZeddZedd Zed d Z ed d Z eddZ ddZ ddZ ddZddZddZddZddZddd d!Zd"d#Zd$d%Zd&d'ZZS)(z Album.Eventz Represents the event of a new album. Members: messages (Sequence[`Message `]): The list of messages belonging to the same album. cs<|d}tj|j|jt|jdt||j||_dS)Nr) chat_peermsg_id broadcast) r,rpeer_idr@boolpostr sender_idr)rrr9r.rrrs    zAlbum.Event.__init__cst|t|j|j|j\|_|_|j D] }| ||jdqt |j dkrG|j |j}|dur?t|||j |j<dS||j dSdSrJ)r, _set_clientr_get_entity_pairrR _entities_mb_entity_cache_sender _input_senderr _finish_initrA_albumsgetr;r r)rrmsghackr.rrrSs   zAlbum.Event._set_clientcC |jdjS)zM The shared ``grouped_id`` between all the messages. r)rr;rrrrr;s zAlbum.Event.grouped_idcCtdd|jDdS)z The message text of the first photo with a caption, formatted using the client's default parse mode. cs|] }|jr|jVqdSr )textr1mrrr z#Album.Event.text..nextrr_rrrrbzAlbum.Event.textcCr`)zv The raw message text of the first photo with a caption, ignoring any formatting. csrar )raw_textrcrrrrerfz'Album.Event.raw_text..rgrhr_rrrrkrjzAlbum.Event.raw_textcCr^)a `True` if the album is a reply to some other message. Remember that you can access the ID of the message this one is replying to through `reply_to_msg_id`, and the `Message` object with `get_reply_message()`. r)ris_replyr_rrrrls zAlbum.Event.is_replycCr^)z The `Forward ` information for the first message in the album if it was forwarded. r)rforwardr_rrrrms zAlbum.Event.forwardcs|jdIdHS)z The `Message ` that this album is replying to, or `None`. The result will be cached after its first use. rN)rget_reply_messager_rrrrnszAlbum.Event.get_reply_messagec|jdj|i|IdHS)z Responds to the album (not as a reply). Shorthand for `telethon.client.messages.MessageMethods.send_message` with ``entity`` already set. rN)rrespondrargskwargsrrrrpzAlbum.Event.respondcro)z Replies to the first photo in the album (as a reply). Shorthand for `telethon.client.messages.MessageMethods.send_message` with both ``entity`` and ``reply_to`` already set. rN)rreplyrqrrrrurtzAlbum.Event.replycs@|jr|j|d<|IdH|d<|jj|i|IdHSdS)z Forwards the entire album. Shorthand for `telethon.client.messages.MessageMethods.forward_messages` with both ``messages`` and ``from_peer`` already set. rN from_peer)rrget_input_chatforward_messagesrqrrr forward_tos  zAlbum.Event.forward_tocsJ|jD]}|jr|j|i|IdHSq|jdj|i|IdHS)a Edits the first caption or the message, or the first messages' caption if no caption is set, iff it's outgoing. Shorthand for `telethon.client.messages.MessageMethods.edit_message` with both ``entity`` and ``message`` already set. Returns `None` if the message was incoming, or the edited `Message` otherwise. .. note:: This is different from `client.edit_message ` and **will respect** the previous state of the message. For example, if the message didn't have a link preview, the edit won't add one by default, and you should force it by setting it to `True` if you want it. This is generally the most desired and convenient behaviour, and will work for link previews and message buttons. Nr)rrkedit)rrrrsr\rrrrzs  zAlbum.Event.editcs:|jr|jj|IdH|jg|Ri|IdHSdS)aH Deletes the entire album. You're responsible for checking whether you have the permission to do so, or to except the error otherwise. Shorthand for `telethon.client.messages.MessageMethods.delete_messages` with ``entity`` and ``message_ids`` already set. N)rdelete_messagesrwrrqrrrdelete s zAlbum.Event.deletecs8|jr|jj|IdH|jdjdIdHdSdS)z Marks the entire album as read. Shorthand for `client.send_read_acknowledge() ` with both ``entity`` and ``message`` already set. N)max_id)rsend_read_acknowledgerwrr@r_rrr mark_read.s zAlbum.Event.mark_readFnotifycs|jdj|dIdHS)z Pins the first photo in the album. Shorthand for `telethon.client.messages.MessageMethods.pin_message` with both ``entity`` and ``message`` already set. rrN)rpin)rrrrrr9szAlbum.Event.pincC t|jS)zx Return the amount of messages in the album. Equivalent to ``len(self.messages)``. )rArr_rrr__len__A zAlbum.Event.__len__cCr)zu Iterate over the messages in the album. Equivalent to ``iter(self.messages)``. )iterrr_rrr__iter__IrzAlbum.Event.__iter__cCs |j|S)zq Access the n'th message in the album. Equivalent to ``event.messages[n]``. )r)rnrrr __getitem__QrzAlbum.Event.__getitem__)r$r%r&r'rrSpropertyr;rbrkrlrmrnrpruryrzr|rrrrr __classcell__rrr.rrDs2           rDr )NN) r$r%r&r'r classmethodrIrKrr rDrrrr.rr(Cs  )r()r!rrcommonrrrrgrtlrtl.custom.sendergetterr rBr0r>rr r(rrrrs   -