U hZ2@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_dueZ create_task deliver_event)selfclienteventr9/tmp/pip-unpacked-wheel-c81u5j2r/telethon/events/album.py__init__#s zAlbumHack.__init__cCs.|}|r*|jj||jt|_dSr )rrmessagesextendrrrr)rrrrrrr,szAlbumHack.extendcsZ|}|dkrdS|j|j}|dkrB||jIdHdS~t|IdHqdS)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!') NFblacklist_chatsfunccstj|||ddS)Nr&)superr)rZchatsr'r( __class__rrr^szAlbum.__init__cs|g}t|tjtjfrt|jtjs*dS|jjdkr>dStt |drRdSt t tt krfddt DD] }t|=q||D]}||k rtt |<q|fdd|DSdS)Ncs g|]\}}|tkr|qSr)_IGNORE_MAX_AGE).0it)nowrr |s zAlbum.build..cs>g|]6}t|tjtjfrt|jtjr|jjkr|jqSr) isinstancerUpdateNewMessageUpdateNewChannelMessagemessageMessage grouped_id)r-u)grouprrr1s  )r2rr3r4r5r6r7 _IGNORE_DICTpopidrlen_IGNORE_MAX_SIZEitemsEvent)clsupdateZothersself_idr.r8r)r9r0rbuildbs*  z Album.buildcst|jdkrt|SdSNr)r=rr)filter)rrr*rrrFsz 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)Z chat_peerZmsg_id broadcast) r)rZpeer_idr<boolpostr sender_idr)rrr5r*rrrs zAlbum.Event.__init__cst|t|j|j|j\|_|_|j D]}| ||jdq.t |j dkr|j |j}|dkr|t|||j |j<n ||j dSrE)r) _set_clientrZ_get_entity_pairrJZ _entitiesZ_mb_entity_cacheZ_senderZ _input_senderrZ _finish_initr=Z_albumsgetr7r r)rrmsgZhackr*rrrKs   zAlbum.Event._set_clientcCs |jdjS)zM The shared ``grouped_id`` between all the messages. r)rr7rrrrr7szAlbum.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|jVqdSr )textr-mrrr sz#Album.Event.text..nextrrNrrrrOszAlbum.Event.textcCstdd|jDdS)zv The raw message text of the first photo with a caption, ignoring any formatting. css|]}|jr|jVqdSr )raw_textrPrrrrRsz'Album.Event.raw_text..rSrTrNrrrrVszAlbum.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_replyrNrrrrWs zAlbum.Event.is_replycCs |jdjS)z The `Forward ` information for the first message in the album if it was forwarded. r)rforwardrNrrrrXszAlbum.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_messagerNrrrrYszAlbum.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)rrespondrargskwargsrrrrZszAlbum.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)rreplyr[rrrr^szAlbum.Event.replycs:|jr6|j|d<|IdH|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_messagesr[rrr forward_tos zAlbum.Event.forward_tocs@|jD] }|jr|j||IdHSq|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)rrVedit)rr\r]rMrrrras zAlbum.Event.editcs2|jr.|jj|IdH|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_messagesr_rr[rrrdelete s zAlbum.Event.deletecs2|jr.|jj|IdH|jdjdIdHdS)z Marks the entire album as read. Shorthand for `client.send_read_acknowledge() ` with both ``entity`` and ``message`` already set. N)Zmax_id)rZsend_read_acknowledger_rr<rNrrr 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. rreN)rpin)rrfrrrrg9szAlbum.Event.pincCs t|jS)zx Return the amount of messages in the album. Equivalent to ``len(self.messages)``. )r=rrNrrr__len__AszAlbum.Event.__len__cCs t|jS)zu Iterate over the messages in the album. Equivalent to ``iter(self.messages)``. )iterrrNrrr__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$rrKpropertyr7rOrVrWrXrYrZr^r`rarbrdrgrhrjrl __classcell__rrr*rr@s0          r@)N)NN) r!r"r#r$r classmethodrDrFrr r@rnrrr*rr%Cs ) r%)rrrcommonrrrrSrtlrZtl.custom.sendergetterr r>r,r:rr r%rrrrs   -