gVddlmZddlZddlmZddlZddlZddlZddlm Z ddl m Z ddl Z ddl m Z ddl mZddl mZdd l mZdd l mZdd l mZdd l mZdd lmZmZddiZdZdZ d dZdZd!dZd!dZd"dZd!dZd#dZ d$dZ!dZ"dZ#dZ$Gdde%Z&Gdde&Z'y)%) OrderedDictN)groupby)TemplateLookup) exceptions)common)database)helpers) libraries)logger)newsletter_handler) pmsconnect)send_notificationEMAILrecently_addedc"ddtddg}|S)NRecently Addedr)labelnameid) AGENT_IDS)agentss #/opt/Tautulli/plexpy/newsletters.pyavailable_newsletter_agentsr,s&&$,- F Mc dddddddddg}|S) NScheduleon_cronz-Trigger a notification on a certain schedule.Tautulli Newsletterz fa-calendar) newsletter)rr descriptionsubjectbodymessageicon media_types)actionss ravailable_notification_actionsr)8s-#!N0-%. G Nrc t|jr4t|}||||||||| | | d } |dk(r tdi| St di| Sy)N) newsletter_idnewsletter_id_nameconfig email_config start_dateend_dater"r#r$ email_msg_idemail_reply_msg_idrr')strisdigitint RecentlyAdded Newsletter) r+r,agent_idr-r.r/r0r"r#r$r1r2kwargss rget_agent_classr:Gsm 8}x=#0(:"". *&$$".(: < q= *6* *'' 'rcNtdttdDS)Nc3&K|] }|d yw)rNr').0as r z(get_newsletter_agents..bsdq6dsc |dS)Nrr')ks rz'get_newsletter_agents..bs XYZaXbrkey)tuplesortedrr'rrget_newsletter_agentsrGas dF+F+HNb$cd ddrcdx}}g}|r@d}|r|dz }|j||dj|fDcgc]}|s| c}z }tj}|j d|z|}|Scc}w)NrzWHERE znewsletters.id = ?z AND arSELECT newsletters.id, newsletters.agent_id, newsletters.agent_name, newsletters.agent_label, newsletters.friendly_name, newsletters.cron, newsletters.active, MAX(newsletter_log.timestamp) AS last_triggered, newsletter_log.success AS last_success FROM newsletters LEFT OUTER JOIN newsletter_log ON newsletters.id = newsletter_log.newsletter_id %s GROUP BY newsletters.idargs)appendjoinrMonitorDatabaseselect)r+wherewhere_idrJwdbresults rget_newslettersrTesEH D  , ,H KK & 8*:Qq:;;  ! ! #B YY &   F M;s A5 A5 ctj}t|jr-t j d|z|j d|g}yy)NzDTautulli Newsletters :: Deleting newsletter_id %s from the database.z$DELETE FROM newsletters WHERE id = ?rITF)rrMr3r4r debugaction)r+rRrSs rdelete_newsletterrXsS  ! ! #B =!!# [$% &AXrc t|jr t|}ntjd|zyt j }|jd|g}|sy tj|jdd}tj|jdd}|jd}|jd}|jd }t||d |d ||||| } |r$tj| j| _| j |d<| j"|d<| j$|d <| j&|d<| j|d<| j)||d<| j+||d<|S#t$r"} tjd | zYd} ~ yd} ~ wwxYw)NzWTautulli Newsletters :: Unable to retrieve newsletter config: invalid newsletter_id %s.z&SELECT * FROM newsletters WHERE id = ?rInewsletter_config{}r.r"r#r$id_namer8)r+r,r8r-r.r"r#r$zDTautulli Newsletters :: Failed to get newsletter config options: %s.r-mask_passwordsconfig_optionsemail_config_options)r3r4r5r errorrrM select_singlejsonloadspopr: Exceptionr mask_config_passwordsr.r"r#r$r-return_config_optionsreturn_email_config_options) r+r^rRrSr-r.r"r#r$newsletter_agentes rget_newsletter_configrls =!!#M*  n$% &  ! ! #B   Fm_  ]F  FJJ':DABzz&**^T"BC **Y'zz&!**Y'*[abk[l4::4F28|3:wX(/(E(EFVFcFc(d%(00F9%**F6N(00F9'..F8-::F>/EEUcEdF %5%Q%Qao%Q%pF !" M  [^__`s+BF G 'GG c tjr tntjdzyt fdt Dd}|stjdzyt|d}ddi}|d|d|dd d tj|jtj|j|j|j|jd }tj } |j#d || |j%}tj&d |dd|dt)|S#t*$r"}tj,d|zYd}~yd}~wwxYw)NzJTautulli Newsletters :: Unable to add new newsletter: invalid agent_id %s.Fc34K|]}|dk(s |ywrNr'r=r>r8s rr?z(add_newsletter_config..Rag>Q!R zUTautulli Newsletters :: Unable to retrieve new newsletter agent: invalid agent_id %s.r)r8rrr) r8 agent_name agent_labelr\ friendly_namerZr.r"r#r$ newsletters table_namekey_dict value_dictz4Tautulli Newsletters :: Added new newsletter agent:  (newsletter_id ).z;Tautulli Newsletters :: Unable to add newsletter agent: %s.)r3r4r5r ranextrr:rcdumpsr-r.r"r#r$rrMupsertlast_insert_idinfoblacklist_loggerrfwarn) r8r9agent agent_classkeysvaluesrRr+rks ` radd_newsletter_configrsc 8}x= a  ! R8:RTX YE  l  !!5;7K $d|zYd}~yd}~wwxYw)NzOTautulli Newsletters :: Unable to set existing newsletter: invalid agent_id %s.Fc34K|]}|dk(s |ywror'rps rr?z(set_newsletter_config..rqrrzZTautulli Newsletters :: Unable to retrieve existing newsletter agent: invalid agent_id %s.newsletter_config_newsletter_email_passwordz r+r.r"r#r$rr8r-r.r"r#r$rrr\rrucronactive) r8rsrtr\rurZr.r"r#r$rrrvrwz2Tautulli Newsletters :: Updated newsletter agent: r{r|Tz>Tautulli Newsletters :: Unable to update newsletter agent: %s.) r3r4r5r rar}rlistr startswithlenreitemsrlr:getrcr~r-r.r"r#r$rrMrrr schedule_newslettersrrfr)r+r8r9r config_prefixemail_config_prefixrArZr.cfgvalold_newsletter_configr"r#r$rrrrRrks ` rset_newsletter_configrs 8}x= f  ! R8:RTX YE  q  !(M-#'v{{}"5Vm9T3}-./A>VV"&++-0WALLAT4Uc-./0&**Q-?WLW!&&(KS  $9 $V ! 5n Ec JL  Kjj#G ::f Djj#G!5;):*1gOK - D+!&M"7^Ir2%zz/2>#'::k.@.@#A"jj)A)AB$,,!&&$,,jj( 8, F  ! ! #B  ]Tf M W~}6 7//mLUVWR  TWXXYs%6J>*6K*AK K3K..K3c t|}|r*t|d|d|d|||}|jStjdy)Nrr8r-r.rzMTautulli Newsletters :: Notification requested but no newsletter_id received.)rlr:sendr rV)r+r"r#r$newsletter_log_idr9rZrs rsend_newsletterr#sV-MJ)::)F'8'B->~-N(/%)(/ 1 zz| derc tj}|jd}|D]d}tj|dxsd}t j |tj|dxsd}t j |fy)Nz7SELECT newsletter_config, email_config FROM newslettersrZr[r.)rrMrNrcrdr blacklist_config)rR notifiersnr-r.s rrr1s{  ! ! #B STI .A12:d;'zz!N"3";t<  - .rc tjjr0tjdtjj}nt j jttjd}t j jt|tjj}tjjs|jdd}t|gddg} |j|}|jd i|dfS#t!j"jd fcYSxYw) NzCTautulli Newsletters :: Using custom newsletter template directory.zdata/interfaces/.htmlz.internal.htmlunicodeh) directoriesdefault_filtersFTr')plexpyCONFIGNEWSLETTER_CUSTOM_DIRr rospathrLr3PROG_DIRNEWSLETTER_TEMPLATESNEWSLETTER_INLINE_STYLESreplacer get_templaterenderrhtml_error_template) template_namer9 template_dir interface_dir _hplookuptemplates rserve_templater<s }}** YZ}}::  S%9;MN ww||C $6 8Z8Z[ }}55)11';KLML>IWZK[\I?))-8x((%//?--/668$>>s $D%%&E cd}d}tj}|r|r5tjdd}|j d|g}|d}|s2|r5|S)NrrzLSELECT EXISTS(SELECT uuid FROM newsletter_log WHERE uuid = ?) as uuid_exists uuid_exists)rrMr generate_uuidrb)uuidrrRrSs rgenerate_newsletter_uuidrPsf DK  ! ! #Bk##%bq)!! Z]a\bd]+ k Krc eZdZdZdddddddddZej Zded<ded <dZd Z dZ d Z dZ dd Z ddZddZdZdZd dZd dZdZdZdZdZdZdZdZd dZdZd dZy )!r7rrdays) custom_cron time_frametime_frame_units formattedthreaded notifier_idfilename save_onlyr from_namer*View the newsletter here: {newsletter_url}z!newsletter_{newsletter_uuid}.htmlNc :|j||j|_|j||j|_t |_| |_| |_||_ |xsd|_ d|_ d|_ |r- tj|ddjd|_ |jtj"|_ |r- tj|ddj%d|_ |j|jddk(r0|jj'|jd  |_ nq|jdd k(r0|jj'|jd  |_ n/|jj'|jd  |_ |jj)|_|jj)|_|j/|_|xs |j2|_|xs |j6|_| xs |j:|_|jdxs |j>|_ |j@jCds|xj@dz c_ |jE\|_#|_$|_%|jM|_'i|_(d|_)d|_*d|_+y#t $rYXwxYw#t $rYwxYw)Nr-defaultrz YYYY-MM-DDlocal)tzinfodayrmonthsr)rr)r)hoursrrF), set_config_DEFAULT_CONFIGr-_DEFAULT_EMAIL_CONFIGr.rrr1r2r+r,r/r0arrowrceil ValueErrornowfloorshift timestampend_time start_time build_params parameters_DEFAULT_SUBJECTr" _DEFAULT_BODYr#_DEFAULT_MESSAGEr$_DEFAULT_FILENAMErendswith build_textsubject_formattedbody_formattedmessage_formattedbuild_filenamefilename_formatteddatar is_previewtemplate_error) selfr+r,r-r.r/r0r"r#r$r1r2s r__init__zNewsletter.__init__qsooVT=Q=QoR  OO<IcIcOd,. ("4*"4":   % (L Q V VW\ ]  == !IIKDM  "'))J W"U"["[\a"b ?? "{{-.(:"&--"5"5dkk,>W=W"5"X/0F:"&--"5"5DKK N MMO 1DAq!S! ' 3 3FJJq!4D E 1 At$JJq!$!!T*%&CJqM$%JqM & 1a 0 1  1rcyrr'rs r retrieve_datazNewsletter.retrieve_datas rcyNFr'rs r _has_datazNewsletter._has_datasrc |rd|_|jddlm}|jj Dcic]\}}|t ||r t|n|!}}}|j||jdScc}}w)NTr CustomArrow)titlerr) rrplexpy.notification_handlerr rrrr3NAMEr)rpreviewr rArrs rraw_datazNewsletter.raw_datasx "DO  c |rd|_|jtjd|jrdndzt |j |j |j|j|j|j|j|j\}|_ |jr|Sd|vr|jdd}|Sd}tjd d |z|}|S) NTz0Tautulli Newsletters :: Generating newsletter%s.z previewr)rrr"r#r$rrr z'zNewsletter generated by Tautulli.a

The Tautulli newsletter footer was removed from the newsletter template.
Please leave the footer in place as it is unobtrusive and supports Tautulli.
Thank you.

z ()z\1)rrr rr _TEMPLATErrrrrrrrresub)rr newsletter_renderedmsgs rgenerate_newsletterzNewsletter.generate_newsletters "DO  FX\XgXg*mopq3A..**$$**OO 4 0T0   & & 58K K"5"="=9f#  #" C#%&&FY"Z ""rcP|j|_|jr#tjd|j zy|j s#tjd|j zy|j|jdry|jS)NzUTautulli Newsletters :: %s newsletter failed to render template. Newsletter not sent.FzGTautulli Newsletters :: %s newsletter has no data. Newsletter not sent.rT) rr rr rar rr_saver-_sendrs rrzNewsletter.sends224    LLpsws|s|| }~~ KKadhdmdmm n ;;{ #zz|rc |j}tjjxs.tj j tjd}tj j ||}tj j|st j| t|d5}|jjD]*}d|vs|j|dzjd, dddtj d|j"d|dy#1swY0xYw#t$$r2}tj&d |j"d |d |Yd}~yd}~wwxYw) Nrvwbzz zutf-8zTautulli Newsletters :: z newsletter saved to ''z'Tautulli Newsletters :: Failed to save z newsletter to 'z': )rrrNEWSLETTER_DIRrrrLDATA_DIRexistsmakedirsopenr splitlineswriteencoder rr OSErrorra)rnewsletter_filenewsletter_foldernewsletter_file_fpn_filelinerks rrzNewsletter._saves%11"MM88hBGGLLZg.)s)Z!''))Zsz7HTML email support is required to view the newsletter. rr)rr"r# plaintextmsg_id reply_msg_id)r-)r"r#r.r/r0)rr"r#r')r-rLr r!rrNEWSLETTER_SELF_HOSTED HTTP_BASE_URLrformatrr2r.rrr1rnotifyr)rnewsletter_strippedr.r2emails rrzNewsletter._send'sR ;;{ #"$'')ZT__=W=W=Y)Z"Z RI}}33 8S8S6T//66III <@KK   *4;;7I+J  h2 $X%58O8O%O! " tyy# $"//"&"9"9#)==#D#D) .rcddlm}|} |jt|jfi|j } |jt|jfi|j } |jt|jfi|j }|||fS#t $rQ}tjd|zt|jjd i|j }Yd}~d}~wt$rR}tjd|zt|jjd i|j }Yd}~d}~wwxYw#t $rR}tjd|zt|jjd i|j }Yd}~Bd}~wt$rR}tjd|zt|jjd i|j }Yd}~d}~wwxYw#t $rR}tjd|zt|jjd i|j }Yd}~d}~wt$rR}tjd|zt|jjd i|j }Yd}~!d}~wwxYw) NrCustomFormatterzZTautulli Newsletter :: Unable to parse parameter %s in newsletter subject. Using fallback.UTautulli Newsletter :: Unable to parse custom newsletter subject: %s. Using fallback.zWTautulli Newsletter :: Unable to parse parameter %s in newsletter body. Using fallback.zRTautulli Newsletter :: Unable to parse custom newsletter body: %s. Using fallback.zZTautulli Newsletter :: Unable to parse parameter %s in newsletter message. Using fallback.zUTautulli Newsletter :: Unable to parse custom newsletter message: %s. Using fallback.r')r r[r3r3r"r LookupErrorr rarrfr#rr$r)rr[custom_formatterr"rkr#r$s rrzNewsletter.build_textss??*, K-&--c$,,.?S4??SG E*#**3tyy>MT__MD K-&--c$,,.?S4??SGg%%3 K LLuxyy z7c$//077J$//JG K LLpstt u7c$//077J$//JG K  E LLruvv w13t))*11DDOODD E LLmpqq r13t))*11DDOODD E  K LLuxyy z7c$//077J$//JG K LLpstt u7c$//077J$//JG Ksq0B&0E10H& E/AC;; EAEE H%AF22 H>AH  H KAI)) K5AKKcddlm}|} |jt|jfi|j }|S#t $rR}tjd|zt|jjdi|j }Yd}~|Sd}~wt$rR}tjd|zt|jjdi|j }Yd}~|Sd}~wwxYw)NrrZz[Tautulli Newsletter :: Unable to parse parameter %s in newsletter filename. Using fallback.r\r') r r[r3r3rrr]r rarrf)rr[r^rrks rrzNewsletter.build_filenames?*, M.'..s4==/AUT__UH M LLvyzz {9s411299LDOOLH   M LLpstt u9s411299LDOOLH  Ms%0A C5 AB C5#AC00C5cT|j}|rtj||Sr)_return_config_optionsr rg)rr^r_s rrhz Newsletter.return_config_optionss)446   ) ). 9rc g}|Srr'rr_s rraz!Newsletter._return_config_optionss rctt|jj|}|D] }d|dz|d<|S)Nr] newsletter_r)rr.rh)rr^r_rs rriz&Newsletter.return_email_config_optionssHt001GGWeGf 2A%& 1AfI 2r) NNNNNNNNNNNNN)F)__name__ __module__ __qualname__r rrreturn_default_configrrrrrrrrrrrrrrrrrr8rrrhrarir'rrr7r7^s D&'%&+1$%#$&'#%$%'O"G99;)>+&+,-(,@M;I^bRV7;9#vE&  #(#T$<*"H !F&B rr7ceZdZdZdZej jZged<dZdZ dZ dZ dd Z d Z d Zd Zd ZdZdZy)r6z# Recently Added Newsletter rincl_librariesz-Recently Added to {server_name}! ({end_date})rrzrecently_added.htmlNc ddlm}tj}g}d}d}|s~|j t |d|}|dDcgc])}|j tj|dkr|+} }t| dkrd }n|dz }|j| |s~|d vrjg} |D]a} | d |jd vr|j tj| dcxkr|jksNnQ| j| c| }|d k(rg} g} |D]} | d |jd vr| dd k(r| d}n| ddk(r| d}n | ddk(r| d}| vrG|j|d}|j||d }|dDcgc]<}|j tj|dcxkr|jkrnn|>}}|j!d|sg}t#|dD]b\\}}}t%|}||Dcgc]}tj|dc}\}}|j|||t||dd||Dcgc]}tj|dc}\}}||d<t||d<||d<| j|| j|| }|dk(rXg}g}|D]L} | d |jd vr| ddk(r| d}n| ddk(r| d}n | dd k(r| d}|vrG|j|d}|j||!}|dDcgc]<}|j tj|dcxkr|jkrnn|>}}|j!d"|sg}|D]D}|j|dd}tj|d#|d$<|j|Ft||d%<||d<|j||j|O|}|Scc}wcc}wcc}wcc}wcc}w)&Nr)format_group_indexF10)startcount media_typeradded_at T)movie other_video section_idrlshowrr rating_keyseasonparent_rating_keyepisodegrandparent_rating_key) media_info)rrget_grandchildren children_listc2tj|dS)Nparent_media_index)r rxs rrBz3RecentlyAdded._get_recently_added..sW5H5HK_I`5arrCc|d|dfS)Nr parent_titler'rs rrBz3RecentlyAdded._get_recently_added..sqI]G^`abp`qFrr media_index)rr  episode_range episode_countr| season_range season_countartistalbumtrack)rrc |dS)Nrsr'rs rrBz3RecentlyAdded._get_recently_added..+s Qz]rchildren_count track_count album_count)r rnr PmsConnectget_recently_added_detailsr3rr rrextendr-rrKget_metadata_detailsget_item_childrensortrr) rrrrn pms_connectrdonerp recent_itemsifiltered_items movie_listitem shows_listshow_rating_keysshow_rating_key show_metadatachildrenfiltered_childrenseasonsindexr episodesdnumnum00 artists_listartist_rating_keysartist_rating_keyartist_metadataalbumsr>album_metadatas r_get_recently_addedz!RecentlyAdded._get_recently_addedsB ++- &AAE Z^kuAvL)56F)GWA!%73F3Fq}3U!U WNW>"R'   ! !. 1 1 1J& , %T[[9I-JJ??W%8%8j9I%JZT]]Z%%d+  ,(N  J! &+ 9 %T[[9I-JJ %/&*<&8O,'83&*+>&?O,'94&*+C&DO"&66 + @ @]b @ c &88U_sw8x080I%n1(,':M:MaPZm:\(l_c_l_l(l&'%n!%n!&&+a&b(078I%ns .P,AP1P6 $P; AQc zddlm}m}|jds"t j d|j zt}|jD]V}t|d|jdvs!|ddk(r|dd k(r|jd C|j|dXi}|D]}||vs|j|||<|jdg}|jd g}|jd g} | D cgc]} | d D]} | } } } |jd g} |jstjddk(rf||z| z| zD]W}|dd k(rd}d}nd}d}||dd|||d<|dr||ddddddd|d <nd!|d <d!|d"<d!|d#<|d"|d$<Yntjr||z| z| zD]}|dd k(rd}d}nd}d}||d|d%|d&d||'}|jd(xst j"|d"<||d|d%|d&dddddd) }|jd(|d#<d!|d<d!|d <|d"|d$<n,||z| z| zD]}d!|d<d!|d <d!|d"<d!|d#<|d"|d$< ||j$d*<|j$Scc} } w)+Nr) get_img_infoset_hash_image_inforlzUTautulli Newsletters :: Failed to retrieve %s newsletter data: no libraries selected.rw section_typerurcom.plexapp.agents.nonervrxrrT) include_selfz self-hostedrrcoverposterthumb)imgwidthheightfallback thumb_hashartii282828)rrropacity backgroundblurrart_hashr thumb_urlart_url poster_urlryr )rryr rrrimg_url) rryr rrrrrrr)r rrr-r rr set _get_sectionsr3addrrrr get_img_servicerONLINE_POSTER_THUMBr)rrrr&srrrmoviesshowsartistsrr>rrvrrrimg_infos rrzRecentlyAdded.retrieve_data@s{Q{{+, KKorvr{r{{ |e ##% 7A1\?#t{{3C'DD^$/AgJB[4[OOM2OOAn$56  7% RJ/-1-E-Ej-Qz* R ##GR0""62. $$Xr2")C6'?Ca!C!CC$((; ??g554HMY/+= 7 %0 F&H F'H%8W S(&T\";': Ks3 "xa%(QD$(*D$$&[!"$Y%)+%6\") 7, $ $ &/+= 7 %0 F&H F'H'W $|2DDQXMfxA%-LL$;$Yv?Y?Y[!'U \0B$w-c2(QR]bd#+,,y"9Y%'\"#%Z %)+%6\"- 72/+= 7%'\"#%Z $&[!"$Y%)+%6\"  7'5 "#yy}Ds J7c|jjd}|rE|jds3|jds"|jds|jdryy)NrrurxrrvTF)rr)rrs rrzRecentlyAdded._has_datasV'78 ""7+""6*""8,""=1rcFtjjSr)r Libraries get_sectionsrs rrzRecentlyAdded._get_sectionss""$1133rc i}|jD]T}|ddk7s |ddk(r |ddk(rd}n|d}|j|g}|j|d|dd |||<Vtd |jdfd |jd fd |jdfd|jdffDcgc] \}}|||f c}}}|Scc}}w)Nrphotorurrrvrw section_name)valuetextzMovie LibrarieszTV Show LibrariesrxzMusic LibrariesrzOther Video Libraries)rrrKr)rsectionsr library_typegrouprArgroupss r_get_sections_optionsz#RecentlyAdded._get_sections_optionss##% /A G+^$/AgJB[4[#0L#$^#4L \26 q&'&79:).& /  W 5 6 (,,v"6 7  X 6 7 $hll=&A B 2 A ] !!f s<C cf|j}g}|jD]4}t|d|jdvs!|j |d6dj t ||d<tjj|d<tjj|d<|S)Nrwrlrz, newsletter_librariespms_identifier pms_web_url) r8rr3r-rKrLrFrrPMS_IDENTIFIER PMS_WEB_URL)rrrrs rrzRecentlyAdded.build_paramss'') !##% ?A1\?#t{{3C'DD$++An,=> ?.2YYv>R7S-T )*'-}}'C'C #$$*MM$=$= =!rcRd|jdddd|jddg}|S)NzIncluded Librariesrlz2Select the libraries to include in the newsletter. newsletter_config_incl_libraries selectizeT)rrr!r input_typeselect_options select_all)r-rrcs rraz$RecentlyAdded._return_config_optionss>*kk"23P7&#99;    rr)rgrhri__doc__r r7rcopyrrrrrrrrrrrar'rrr6r6sj D 00557O(*O$%F@M%I{zTl 4,  rr6) NNNNNNNNNNNNrrrf)NNNNN)( collectionsrrc itertoolsrrrr mako.lookuprmakorrrrr r r r r plexpy.notifiersrrrrr)r:rGrTrXrlrrrrrrobjectr7r6r'rrrs$$  & %5a   koUY:>4e6 'T(V?D f.?( VVr ]J]r