g^ ddlZddlZddlZddlZddlZddlZddlZddlZddlZddl Z ddl Z ddl Z ddl m Z ddlmZmZddlmZddlmZddlmZmZddlmZddlZddlmZdd lmZmZmZ dd l m Z ejDd Z#id d ddddddddddddddddddd d!d"d#d$d%d&d'd(d)d*d+d,d-d.d/d0Z$e$jKDcic]\}}|| c}}Z&id1dd2d d,dd3dd4dd5dd6dd7dd8dd9dd:d!d;d#dd?d@dAdBdCdDidEdFdGdHdIdJdKdLdMdNdOdPdQdRdSdTdUdVdWdXdYdZd[d\d]d^d_d`dadbdcdddedfdgdhdidjZ'e'jKDcic]\}}|| c}}Z(iZ)GdkdlejTZ+dmZ,dnZ-doZ.dpZ/dqZ0ddrZ1dsZ2dtZ3duZ4dvZ5dwZ6ddxZ7dyZ8ddzZ9dd{Z: dd|Z; dd}ZddZ?dZ@dZAdZBddZCddZDdZEdZFdZGgdZHejdk\reHjgdeHDcgc]\}}eK|deK|c}}ZLejddjeLdZOdZPy#e!$rdZ YwxYwcc}}wcc}}wcc}}w)N)deque)datetime timedelta)getpass)sha1)EventThread)quote)_codes) BadRequestNotFound Unauthorized)tqdmplexapimovieshowseasonepisodetrailercomicpersonartistalbum track picture clip photo photoalbumplaylistplaylistFolder collection*i)optimizedVersionuserPlaylistItemtaggenredirectorwriterroleproducercountrychapterreviewlabelmarkermediaProcessingTargetmakemodelapertureexposureisolensdeviceautotagmoodi,stylei-formati.similari1concerti2banneri7posteri8arti9guidi: ratingImagei<themei=studioi>networki?iBiCi) showOrdering clearLogoplacec$eZdZdZddZdZdZy) SecretsFilterz! Logging filter to hide secrets. Nc*|xs t|_yN)setsecrets)selfrgs "/opt/Tautulli/lib/plexapi/utils.py__init__zSecretsFilter.__init__js'#% cJ| |dk7r|jj||SN)rgadd)rhsecrets ri add_secretzSecretsFilter.add_secretms%  &B, LL  V $ rkct|j}tt|D]?}t ||t s|j D]}||j|d||<At||_y)NzT) listargsrangelen isinstancestrrgreplacetuple)rhrecord cleanargsirps rifilterzSecretsFilter.filterrsx% s9~& LA)A,,"llLF#,Q<#7#7 #KIaLL LI& rkre)__name__ __module__ __qualname____doc__rjrqr~rkrircrcgs+( rkrcc lt|dt|d|j}|r|jd|n |j}t|ddr|d}nt|ddr|d}|tvr? ?K Jrkctj|}||Sd|vr"|jddd}t||Stj|S)z Return the PlexObject class for the specified ehash. This recursively looks up the class with the highest specificity, falling back to the default class if not found. rrr)default)rgetrsplit getPlexObject)rrrs rirrsT //% C   e| S!$Q'UG44 ??7 ##rkc||S|tk(r|dvry|dvryt||ttfvr ||S||S#t$rtdcYSwxYw)aB Cast the specified value to the specified type (returned by func). Currently this only support str, int, float, bool. Should be extended if needed. Parameters: func (func): Callback function to used cast to type (int, bool, float). value (any): value to be cast and returned. )rT1trueT)rF0falseFnan)bool ValueErrorintfloat)funcvalues ricastrsz }  t| * * , , U| ;  ; <  sAAAc |syg}t|dD]1}t||}|j|dt|d3ddj |S)z Returns a query string (uses for HTTP URLs) where only the value is URL encoded. Example return value: '?genre=action&type=1337'. Parameters: args (dict): Arguments to include in query string. rnc"|jSrelowerxs rizjoinArgs..s !'')rk)key=)safe?&)sortedrxappendr join)rtarglistrrs rijoinArgsrsk Gd 349DI#ae" 56789sxx ! ""rkc4|dj|ddzS)Nrrrss ri lowerFirstrs Q4:: for each set of `*args` in listargs. Each call to is called concurrently in their own separate threads. Parameters: callback (func): Callback function to apply to each set of `*args`. listargs (list): List of lists; `*args` to pass each thread. N)job_is_done_event)targetrtkwargsTc3>K|]}|j ywre)is_alive).0ts ri zthreaded..>s1A1::<1sg?) rrvrr rdaemonstartis_setalltimesleep)callbacklistargsthreadsresultsrrtrs rithreadedr-s2WG #g,''tvXDXiAjkl!     &&( 11 1  4 &&(  0!!-A 00 0s CCc|;|r tj||S t |} tj |S|S#t$rtj d||YywxYw#t$rtj d|YywxYw#tttf$rL tj dt|zcYS#t$rtj d|YYywxYwwxYw)z Returns a datetime object from the specified value. Parameters: value (str): value to return as a datetime format (str): Format to pass strftime (optional; if value is a str). NzCFailed to parse "%s" to datetime as format "%s", defaulting to NonezAFailed to parse "%s" to datetime as timestamp, defaulting to Noner)secondszQFailed to parse "%s" to datetime as timestamp (out-of-bounds), defaulting to None) rstrptimerloginfor fromtimestampOSError OverflowErrorr)rrTs ri toDatetimerEs   ((77  E  --e44 L# ^`egmn   \^cd  ]J7  #11!4y7OOO$ HHprwx  sQA A&B A#"A#&BB C+ !CC+C'#C+&C''C+c t|}|dkrdtt|zSt|d\}}t|d\}}t|d\}}t|d\}}|dk(rdn|d|dkDrd ndd |d d |d d |d d |dzS)z Returns human readable time duration [D day[s], ]HH:MM:SS.UUU from milliseconds. Parameters: milliseconds (str, int): time duration in milliseconds. r-i<rnz dayrr, 02d:r03d)rmillisecondToHumanstrabsdivmod) millisecondssecsmsminshoursdayss rirrds |$La*3|+<===lD)HD"b!JD$r"KE4#KD%!)BD6TAXc2-Fb!IPUVY{Z[\`ad[eefgklofppqrtuxqyMz zzrkc|xsd}|xst}|j|Dcgc]}|dk7s ||c}Scc}w)a$ Returns a list of strings from the specified value. Parameters: value (str): comma delimited string to convert to list. itemcast (func): Function to cast each list item to (default str). delim (str): string delimiter (optional; default ','). rn)rxr)ritemcastritems ritoListr ts= KRE3H',{{5'9 HtTRZHTN HH Hs < <cdtjtjtjd|j ddj }djfd|D}|S)Nz-_.()[] NFKDASCIIignorernc3.K|] }|vr|nywrer)rcry whitelists rirz cleanFilename..sZAANq?Zs)string ascii_lettersdigits unicodedata normalizeencodedecoder)filenamerycleaned_filenamers ` @ri cleanFilenamersc6//0@I",,VX>EEgxX__awwZIYZZ rkc i}|jD]}d}|jD]^} |jr |j}| js(d| jd| jj d|j }`|sy|C|j} d|jdd| dttj}|j|||||}t||j|} | |d |d <|S) a Helper to download a bif image or thumb.url from plex.server.sessions. Parameters: filename (str): default to None, height (int): Height of the image. width (int): width of the image. opacity (int): Opacity of the resulting image (possibly deprecated). saturation (int): Saturating of the resulting image. Returns: {'hellowlol': {'filepath': '', 'url': 'http://'}, {'': {filepath, url}}, ... Nz/library/parts/z /indexes//session_transcode_r_)r)filepathurlusername)sessions iterPartsthumbindexesidr viewOffset_prettyfilename usernamesrrtranscodeImagedownload_token) serverrheightwidthopacity saturationrmediar!part prettynamer s ridownloadSessionImagesr6s D" BOO% dD{{kk||'y $,,:L:L:N9OqQVQaQaPbc  d "224 /0B/C1ZLPQRUVZV_V_VaRbQcd''VUGZPCV]]XFH,4SAD  B Krkc |xstj}d|i} |j|| d} | jdvrt j| jd} | j j dd} d| jd | d | jd| } | jd k(r t| | jd k(r t| t| |xstj}tj|d |sV| jjdr;tj d| jjd}|dr|dnd}tj"j%|}tj"j'||}tj"j)|d}|s8| jjd}|rd|vr||j+ddz }|rt,j/d||St,j1d||r:t2r4t5| jjdd}t3dd||}t7|d5}| j9|D]7}|j;||st2sj=t?|9 ddd|rt2rjA|jCdr4|r2tEjF|d5}|jI|ddd|S|S#1swYhxYw#1swY|SxYw)a Helper to download a thumb, videofile or other media item. Returns the local path to the downloaded file. Parameters: url (str): URL where the content be reached. token (str): Plex auth token to include in headers. filename (str): Filename of the downloaded file, default None. savepath (str): Defaults to current working dir. chunksize (int): What chunksize read/write at the time. mocked (bool): Helper to do everything except write the file. unpack (bool): Unpack the zip file. showstatus(bool): Display a progressbar. Example: >>> download(a_episode.getStreamURL(), a_episode.location) /path/to/file z X-Plex-TokenT)headersstream)rCrErKr  (z) z; ii)exist_okzContent-Dispositionzfilename=\"(.+)\"Nrz content-typeimagerrzMocked download %szDownloading: %szcontent-lengthB)unit unit_scaletotaldescwb) chunk_sizezipr)%requestsSessionr status_codecodestextryr!rr r osgetcwdmakedirsr8refindallpathbasenamersplitextrrdebugrrropen iter_contentwriteupdatervcloseendswithzipfileZipFile extractall)r!tokenrsavepathsession chunksizeunpackmocked showstatusr8responsecodenameerrtextmessagefullpath extension contenttyperBbarhandlechunks rir,r,s(+))+Gu%G{{3{=H?299X11215--''c2h**+2hZr(,,q R   3 &w' '  ! !S (7# #W% %&299;HKK4( ((,,-BC::2H4D4D4H4HI^4_`"*1+8A;4ww)Hww||Hh/H  *2.I &&**>: 7k1  ))#.q1 1H &1HH )dH$$(()91=>EI h '**i*@ 'E LL d 3u:& '' d F __Xs + (v   h ' ( O8O'' ( Os$8)L-"L-)L-L9-L69Mcddlm}ddlm}|rN|jrB|j r6t d|jd||j|j S|jd}|jd}|r|rt d|d|||S|jd}|rt d || Std }td }t d|d|||S) aK Helper function tries to get a MyPlex Account instance by checking the the following locations for a username and password. This is useful to create user-friendly command line tools. 1. command-line options (opts). 2. environment variables and config.ini 3. Prompt on the command line. r)CONFIG) MyPlexAccountzAuthenticating with Plex.tv as z..zauth.myplex_usernamezauth.myplex_passwordzauth.server_tokenz&Authenticating with Plex.tv with tokenr^zWhat is your plex.tv username: zWhat is your plex.tv password: ) rrpplexapi.myplexrqr"passwordprintrinputr)optsrprqconfig_usernameconfig_password config_tokenr"rts rigetMyPlexAccountr{s,  $-- / bABT]]DMM::jj!78Ojj!78O? //@CD_o>>::12L 67<0067H89H +H:R 89 8 ,,rkcddlm}d|vr td|d}||}|j||j |j |j |j|S)a Helper function to create a new MyPlexDevice. Returns a new MyPlexDevice instance. Parameters: headers (dict): Provide the X-Plex- headers for the new device. A unique X-Plex-Client-Identifier is required. account (MyPlexAccount): The Plex account to create the device on. timeout (int): Timeout in seconds to wait for device login. r)MyPlexPinLoginX-Plex-Client-Identifier0The X-Plex-Client-Identifier header is required.)r8timeout)clientId)rsr}r runlinkpin waitForLoginrN)r8accountrr}clientIdentifierpinlogins ricreateMyPlexDevicersm.!0KLL9:g.H LLL! LL  >>#3> 44rkc>ddlm}m}d|vr td||d}t dt |j ||j ||j|jrt d ||j St d y ) a Helper function for Plex OAuth login. Returns a new MyPlexAccount instance. Parameters: headers (dict): Provide the X-Plex- headers for the new device. A unique X-Plex-Client-Identifier is required. forwardUrl (str, optional): The url to redirect the client to after login. timeout (int, optional): Timeout in seconds to wait for device login. Default 120 seconds. r)rqr}r~rT)r8oauthz#Login to Plex at the following url:rzLogin successful!rrz Login failed.N) rsrqr}r ruoauthUrlrrr^)r8 forwardUrlrrqr}rs ri plexOAuthr/s=!0KLLgT:H /0 (  J '( LLL! ~~ !"8>>22 orkct|dk(r|dStt|D]5\}}t|r||n t ||}td|d|7t t |dt fddDr'ttdjd}||S|tS#ttf$rYnwxYwn) zm Command line helper to display a list of choices, asking the user to choose one of the options. rrz z: c3&K|]}|v ywrer)rrinps rirzchoose..[s6186s)rz::rcX|jrt|jSdSre)striprrs rirzchoose..\sQWWY3qwwy>Drkr) rvru enumeratecallablerrvanyslicemaprrr IndexError)msgitemsrindexr}nameidxrs @richooserJs  5zQQx Ge$$q"4.tAwga.> 5'D6"#$ G  3%r #C6%566S!PRUR[R[\_R`abSz!SX&J'    s.AC7 CCCcg}|jD]<}|j|j|jg}||vr|jcS||z }>t d|ddj |d)z] Return the full agent identifier from a short identifier, name, or confirm full identifier. zCould not find "z" in agents list (r))agents identifiershortIdentifierrr r)sectionagentrag identifierss rigetAgentIdentifierres| Fnn}}b&8&8"''B K == +  &ug-@6AR@SSTU VVrkchtj|jdjdS)Nutf-8)base64 b64encoderr)rKs ri base64strrps'   DKK0 1 8 8 AArkcfd}|S)NcHtjfd}|S)zThis is a decorator which can be used to mark functions as deprecated. It will result in a warning being emitted when the function is used.cdjdd}tj|ttj ||i|S)Nz'Call to deprecated function or method "z", r)category stacklevel)rwarningswarnDeprecationWarningrwarning)rtrrrrhrs riwrapperz.deprecated..decorator..wrapperysI;DMM?#gYVWXC MM#(:z R KK (( (rk) functoolswraps)rrrhrs` ri decoratorzdeprecated..decoratorus'   )  ) rkr)rhrrs`` ri deprecatedrts  rkc#Kt|g}|rC|j}||j|k(r||jt ||rByyw)z} Iterate through an XML tree using a breadth-first search. If tag is specified, only return nodes with that tag. N)rpopleftr6extendrs)rootr6queuenodes ri iterXMLBFSrsJ 4&ME }} ;$((c/J T$Z s AAAc 8d}tj|fd|i|S)z Convert an object to a JSON string. Parameters: obj (object): The object to convert. **kwargs (dict): Keyword arguments to pass to ``json.dumps()``. ct|tr|jS|jj Dcic]\}}|j dr||c}}Scc}}w)Nr)rwr isoformat__dict__r startswith)rrvs ri serializeztoJson..serializesL c8 $==? "!$!3!3!5OAQ\\#=N1OOOs A!A!r)jsondumps)rrrs ritoJsonrs#P ::c 79 7 77rkct|dr|jSt|d5}|jcdddS#1swYyxYw)Nreadrb)hasattrrrU)filefs ri openOrReadrsBtVyy{ dD Qvvxs AA cRt|jdjS)z! Return the SHA1 hash of a guid. r)rr hexdigest)rZs risha1hashrs  G$ % / / 11rk))rr )r&r()r,)))ii)iii))ii)ii)ii)ii)ii)ii)ii)ii)i i )i i )i i )i i )i i )ii)ii)iir[rn]c.tjd|Srm)_illegal_XML_resubrs ricleanXMLStringrs   r1 %%rk)Nrre)N,)r)Nrdr)NNNiFFF)r$)Nx)r)QrrrloggingrLrOrsysrrrr[ collectionsrrrrhashlibr threadingrr urllib.parser rGrequests.status_codesr rJplexapi.exceptionsr r rr ImportError getLoggerrrrrrrrFilterrcrrrrrrrrrrrrrr rr6r,r{rrrrrrrrrr_illegal_XML_characters maxunicoderchr_illegal_XML_rangescompilerrr)rrlowhighs0000rirs   (#1AAg " Q A aq  q   Q  a a Q Rr B R" b!""#$' *(3'8'8':;tq!ad;&  1&  Q& !&  &   a &   A & & q& q&  b&  R&  b& R&  C&  S&  !& "#& $ 3%& & C'& ( c)& *s+& , C-& . S/& 0 c1& 2s3& 4s5& 6 c7& 8 c9& : 3;& < C=& >3?& @ SA& B cC& DsE& F K& N%-NN$45DAq1a45 GNN*$ $2 #  >@$@"8$8"10> { IDG25BPT49L^->5066WB  ! 82 >>W"" ,/d Cz3t9+"**"''*=">!?qAB&g D6