3 uQhZ2@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. cCs6tj||_||_|jjt|_|jj|j dS)N) weakrefref_client_eventlooptime _HACK_DELAY_dueZ create_task deliver_event)selfclienteventr9/tmp/pip-build-2nz6shyl/telethon/telethon/events/album.py__init__#s zAlbumHack.__init__cCs.|j}|r*|jjj||jjt|_dS)N)rrmessagesextendrrrr)rrrrrrr,szAlbumHack.extendcs^xX|j}|dkrdS|j|jj}|dkrD|j|jIdHdS~tj|IdHqWdS)Nr)rrrrZ_dispatch_eventrasynciosleep)rrZdiffrrrr2szAlbumHack.deliver_eventN)__name__ __module__ __qualname____doc__rrrrrrrr s  r csVeZdZdZd 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!') NF)blacklist_chatsfunccstj|||ddS)N)r%r&)superr)rZchatsr%r&) __class__rrr^szAlbum.__init__cs|g}t|tjtjfrt|jtjs*dS|jjdkr>dStjt |drRdSt j t tt krx$fddtj DD] }t|=q~Wx |D]}||k rtt |<qW|jfdd|DSdS)Ncs g|]\}}|tkr|qSr)_IGNORE_MAX_AGE).0it)nowrr |szAlbum.build..cs>g|]6}t|tjtjfrt|jtjr|jjkr|jqSr) isinstancerUpdateNewMessageUpdateNewChannelMessagemessageMessage grouped_id)r*u)grouprrr.s)r/rr0r1r2r3r4 _IGNORE_DICTpopidrlen_IGNORE_MAX_SIZEitemsEvent)clsupdateZothersZself_idr+r5r)r6r-rbuildbs&   z Album.buildcst|jdkrtj|SdS)Nr)r:rr'filter)rr)r(rrrAsz 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|jdtj||j||_dS)Nr)Z chat_peerZmsg_id broadcast) r'rZpeer_idr9boolpostr sender_idr)rrr2)r(rrrs  zAlbum.Event.__init__cstj|tj|j|j|j\|_|_x|j D]}|j ||jdq0Wt |j dkr|j j |j}|dkrt|||j |j<n |j|j dS)Nr)r' _set_clientrZ_get_entity_pairrEZ _entitiesZ_mb_entity_cacheZ_senderZ _input_senderrZ _finish_initr:Z_albumsgetr4r r)rrmsgZhack)r(rrrFs  zAlbum.Event._set_clientcCs |jdjS)zM The shared ``grouped_id`` between all the messages. r)rr4)rrrrr4szAlbum.Event.grouped_idcCstdd|jDdS)z The message text of the first photo with a caption, formatted using the client's default parse mode. css|]}|jr|jVqdS)N)text)r*mrrr sz#Album.Event.text..)nextr)rrrrrIszAlbum.Event.textcCstdd|jDdS)zv The raw message text of the first photo with a caption, ignoring any formatting. css|]}|jr|jVqdS)N)raw_text)r*rJrrrrKsz'Album.Event.raw_text..rL)rMr)rrrrrNszAlbum.Event.raw_textcCs |jdjS)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_reply)rrrrrOs zAlbum.Event.is_replycCs |jdjS)z The `Forward ` information for the first message in the album if it was forwarded. r)rforward)rrrrrPszAlbum.Event.forwardcs|jdjIdHS)z The `Message ` that this album is replying to, or `None`. The result will be cached after its first use. rN)rget_reply_message)rrrrrQszAlbum.Event.get_reply_messagecs|jdj||IdHS)z Responds to the album (not as a reply). Shorthand for `telethon.client.messages.MessageMethods.send_message` with ``entity`` already set. rN)rrespond)rargskwargsrrrrRszAlbum.Event.respondcs|jdj||IdHS)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)rreply)rrSrTrrrrUszAlbum.Event.replycs:|jr6|j|d<|jIdH|d<|jj||IdHSdS)z Forwards the entire album. Shorthand for `telethon.client.messages.MessageMethods.forward_messages` with both ``messages`` and ``from_peer`` already set. rNZ from_peer)rrget_input_chatZforward_messages)rrSrTrrr forward_tos zAlbum.Event.forward_tocs@x&|jD]}|jr|j||IdHSqW|jdj||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)rrNedit)rrSrTrHrrrrXs zAlbum.Event.editcs2|jr.|jj|jIdH|jf||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)rZdelete_messagesrVr)rrSrTrrrdelete szAlbum.Event.deletecs2|jr.|jj|jIdH|jdjdIdHdS)z Marks the entire album as read. Shorthand for `client.send_read_acknowledge() ` with both ``entity`` and ``message`` already set. Nr)Zmax_id)rZsend_read_acknowledgerVrr9)rrrr mark_read.szAlbum.Event.mark_readF)notifycs|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. r)r\N)rpin)rr\rrrr]9szAlbum.Event.pincCs t|jS)zx Return the amount of messages in the album. Equivalent to ``len(self.messages)``. )r:r)rrrr__len__AszAlbum.Event.__len__cCs t|jS)zu Iterate over the messages in the album. Equivalent to ``iter(self.messages)``. )iterr)rrrr__iter__IszAlbum.Event.__iter__cCs |j|S)zq Access the n'th message in the album. Equivalent to ``event.messages[n]``. )r)rnrrr __getitem__QszAlbum.Event.__getitem__)r r!r"r#rrFpropertyr4rIrNrOrPrQrRrUrWrXrYr[r]r^r`rb __classcell__rr)r(rr=s&         r=)N)NN) r r!r"r#r classmethodr@rArr r=rdrr)r(rr$Cs  ) r$)rrr commonrrrrLrtlrZtl.custom.sendergetterr r;r)r7rr r$rrrrs   -