bf/dZddlZddlZ ddlmZn#YnxYwddlmZddlmZm Z m Z ddl m Z m Z mZGddeZGd d eZdS) z Handlers for the Streaming API: https://github.com/mastodon/documentation/blob/master/content/en/methods/timelines/streaming.md N) signature)Mastodon)MastodonMalformedEventErrorMastodonNetworkErrorMastodonReadTimeout)ChunkedEncodingError ReadTimeoutConnectionErrorcteZdZdZdZdZdZdZdZdZ dZ d Z d Z d Z d ZddZdZdZdZdZd S)StreamListeneraCallbacks for the streaming API. Create a subclass, override the on_xxx methods for the kinds of events you're interested in, then pass an instance of your subclass to Mastodon.user_stream(), Mastodon.public_stream(), or Mastodon.hashtag_stream().cdS)z^A new status has appeared. `status` is the parsed `status dict` describing the status.Nselfstatuss `/home/feoh/.local/pipx/venvs/mastodon-archive/lib/python3.11/site-packages/mastodon/streaming.py on_updatezStreamListener.on_update  cdS)zAA status has been deleted. `status_id` is the status' integer ID.Nr)r status_ids r on_deletezStreamListener.on_delete rcdS)ziA new notification. `notification` is the parsed `notification dict` describing the notification.Nrr notifications ron_notificationzStreamListener.on_notification!rrcdS)zgFilters have changed. Does not contain a payload, you will have to refetch filters yourself.Nrrs ron_filters_changedz!StreamListener.on_filters_changed&rrcdS)zA direct message (in the direct stream) has been received. `conversation` is the parsed `conversation dict` dictionary describing the conversationNrr conversations ron_conversationzStreamListener.on_conversation+rrcdS)zA new announcement has been published. `announcement` is the parsed `announcement dict` describing the newly posted announcement.Nrr annoucements ron_announcementzStreamListener.on_announcement0rrcdS)zASomeone has reacted to an announcement. TODO: what is payload lolNrrTODOs ron_announcement_reactionz'StreamListener.on_announcement_reaction5rrcdS)zaAn announcement has been deleted. `annoucement_id` is the id of the deleted announcement.Nrrannoucement_ids ron_announcement_deletez%StreamListener.on_announcement_delete9rrcdS)zgA status has been edited. 'status' is the parsed JSON dictionary describing the updated status.Nrrs ron_status_updatezStreamListener.on_status_update>rrcdS)z9An encrypted message has been received. Currently unused.Nrrunclears ron_encrypted_messagez#StreamListener.on_encrypted_messageCrrcdS)aThere was a connection error, read timeout or other error fatal to the streaming connection. The exception object about to be raised is passed to this function for reference. Note that the exception will be raised properly once you return from this function, so if you are using this handler to reconnect, either never return or start a thread and then catch and ignore the exception. Nr)rerrs ron_abortzStreamListener.on_abortGs  rNcdS)zAn unknown mastodon API event has been received. The name contains the event-name and unknown_event contains the content of the unknown event. Nrrname unknown_events ron_unknown_eventzStreamListener.on_unknown_eventR  rcdS)zThe server has sent us a keep-alive message. This callback may be useful to carry out periodic housekeeping tasks, or just to confirm that the connection is still open.Nrrs rhandle_heartbeatzStreamListener.handle_heartbeatXr?rci}t} |dD]}|r|D]}t|g}|dkr |d}nP#t$rC}t d}||t j||Yd}~nd}~wwxYw|dkr||i}n| ||}t}| |،dS#t$rD}td}||t j||Yd}~dSd}~wt$rE}td f}||t j||Yd}~dSd}~wt$rE}td f}||t j||Yd}~dSd}~wwxYw) a Handles a stream of events from the Mastodon server. When each event is received, the corresponding .on_[name]() method is called. When the Mastodon API changes, the on_unknown_event(name, content) function is called. The default behavior is to throw an error. Define a callback handler to intercept unknown events if needed (and avoid errors) response; a requests response object with the open stream for reading. ) chunk_size zutf-8zMalformed UTF-8NzServer ceased communication.z$Timed out while reading from server.z"Requests reports connection error.) bytearray iter_contentdecodeUnicodeDecodeErrorrr9six raise_from _dispatch _parse_lineextendrrr rr ) rresponseevent line_bufferchunk chunk_partliner8 exceptions r handle_streamzStreamListener.handle_stream^skk . !..!.<< ; ;;&+;; %. |%<%< %.. "'2'9'9''B'B#5""",G$5-7-7 $ i 8 8 8 #$-$'!"!"!"!"!"!"!"!" " $rzz $u 5 5 5(*(,(8(8u(E(E*3++KK'..z::::- ; ;.$   ,-KLLI MM) $ $ $ N            +6889I MM) $ $ $ N            ,4667I MM) $ $ $ N           sZ4D AD  B*'9B% D %B**AD G19E G1:F G1,:G,,G1c |dr|nc |dd\}}n+#td}|||xYw||vr||xxd|zz cc<n|||<|S)N:z: rCzMalformed event. ) startswithrAsplitrr9)rrUrQkeyvaluerVs rrNzStreamListener._parse_lines ??3   #  ! ! # # # # !ZZa00 UU 78JKK  i(((e||c dUl* "c  s A(A.c |d}|d} tj|d}n #d}YnxYwtj|tj}n#t$rP}t d|jd|}||tj ||Yd}~nTd}~wt$rD}t d|}||tj ||Yd}~nd}~wwxYwd| d d z}t|||j } tt| j} n%#| jjd| jj} YnxYwd | vr(| |j kr| ||dS| |||dS| |j kr$| |jkr | dS| |dS| ||dS) NrQdatastream) object_hookz Missing fieldrzBad JSONon_._ for_stream)jsonloadsr_Mastodon__json_hooksKeyErrorrargsr9rKrL ValueErrorreplacegetattrr>listr parameters__code__ co_varnames co_argcountr ) rrQr<r`rfpayloadr8rV handler_namehandler handler_argss rrMzStreamListener._dispatchsl >D=D "!Zh88  "! j83QRRRGG   3!e55I MM) $ $ $ N           4JEEI MM) $ $ $ N          t||C555 $ d.CDD W ' 2 2 =>>LL W"+78U9I9U8UVLLL < ' '$///,,,,,gz22222$///d555GIIIIIGG$$$$$g&&&&&sBA.A4%A C>"AB-- C>::C99C>1!E E5N)__name__ __module__ __qualname____doc__rrrr r$r(r,r0r2r6r9r>rArWrNrMrrrr r s""                                          <<<|$4'4'4'4'4'rr ceZdZdZ dfd ZdZdZdZdZdZ d Z d Z d Z d Z d ZddZxZS)CallbackStreamListenera3 Simple callback stream handler class. Can optionally additionally send local update events to a separate handler. Define an unknown_event_handler for new Mastodon API events. This handler is *not* guaranteed to receive these events forever, and should only be used for diagnostics. Nc tt|||_||_||_||_||_||_||_ ||_ | |_ | |_ | |_ | |_dSrx)superr~__init__update_handlerlocal_update_handlerdelete_handlernotification_handlerfilters_changed_handlerconversation_handlerunknown_event_handlerstatus_update_handlerannouncement_handlerannouncement_reaction_handlerannouncement_delete_handlerencryted_message_handler)rrrrrrrrrrrrr __class__s rrzCallbackStreamListener.__init__s $d++44666,$8!,$8!'>$$8!%:"%:"$8!-J*+F((@%%%rc|j|| |j'd|ddvr||dSdSdS#t$r.}tjt d||Yd}~dSd}~wwxYw)N@accountacctzreceived bad update)rr ExceptionrKrLr)rrr8s rrz CallbackStreamListener.on_update s   *    ' ' ' (4SF9DUV\D]=]=]))&1111154=]=]    N+,A6JJ          s,A B#BBcB|j||dSdSrx)r)r deleted_ids rrz CallbackStreamListener.on_deletes-   *    + + + + + + *rcB|j||dSdSrx)rrs rrz&CallbackStreamListener.on_notification-  $ 0  % %l 3 3 3 3 3 1 0rc@|j|dSdSrx)rrs rr z)CallbackStreamListener.on_filters_changeds+  ' 3  ( ( * * * * * 4 3rcB|j||dSdSrx)rr"s rr$z&CallbackStreamListener.on_conversation#rrcB|j||dSdSrx)rr&s rr(z&CallbackStreamListener.on_announcement's-  $ 0  % %k 2 2 2 2 2 1 0rcB|j||dSdSrx)rr*s rr,z/CallbackStreamListener.on_announcement_reaction+s-  - 9  . .t 4 4 4 4 4 : 9rcB|j||dSdSrx)rr.s rr0z-CallbackStreamListener.on_announcement_delete/s-  + 7  , ,^ < < < < < 8 7rcB|j||dSdSrx)rrs rr2z'CallbackStreamListener.on_status_update3s-  % 1  & &v . . . . . 2 1rcB|j||dSdSrx)rr4s rr6z+CallbackStreamListener.on_encrypted_message7s-  ( 4  ) )' 2 2 2 2 2 5 4rcD|j|||dSdSrx)rr;s rr>z'CallbackStreamListener.on_unknown_event;s/  % 1  & &t] ; ; ; ; ; 2 1r) NNNNNNNNNNNNrx)ryrzr{r|rrrrr r$r(r,r0r2r6r> __classcell__)rs@rr~r~s!%&* $&*&*'+'+)-&*/3-1*.AAAAAA8   ,,,444+++444333555===///333<<<<<<<rs    !!!!!!! DddddddddddRRRRRRRRRRP'P'P'P'P'VP'P'P'fX<X<X<X<X<^X<X<X<X<X