gqddlZddlZddlmZddlmZddlmZdZdZiaddZ ddZ dd Z dd Z dd Z dd Zdd ZddZdZddZdZdZdZy)N)database)helpers)loggerz$3b4b666a-d557-4b92-acdf-e2c8c4b95357zonesignal-disabledc|tvrD|rBtt|tjrt|j dt|<y|tvrB|r@tt|tjrt|j t|=y|tvra|r^d}tj|t |dgt|<t|j tjj|yyy)NTi,args) TEMP_DEVICE_TOKENS isinstance threadingTimercancelset_temp_device_tokenstartr_BLACKLIST_WORDSadd)tokenremoversuccessinvalidate_times "/opt/Tautulli/plexpy/mobile_app.pyrr s ""w (/ A u % , , .$(5! $ $ (/ A u % , , . u % ( (S $-OOOEZbgiman$o5!5!'')##E* .1 (c,tj|SN)r get)rs rget_temp_device_tokenr4s  ! !% ((rc(dx}x}}g}|s|rVd}|r|dz }|j||rd}|j||dj||fDcgc]}|s| c}z }tj}|j d|z|}|Scc}w)NzWHERE z device_id = ?zdevice_token = ?z AND zSELECT * FROM mobile_devices %sr)appendjoinrMonitorDatabaseselect) device_id device_tokenwherewhere_id where_tokenrwdbresults rget_mobile_devicesr*8s%''E'H{ DL   'H KK " ,K KK % 8[*AGQQqGHH  ! ! #B YY85@tY LF M Hs B B c |syt|S)Nr#)r*r,s rget_mobile_device_by_tokenr-Ls  < 88rctj}d|i}|||||d} |r|| d< |jd|| } t| dk(rt jd|znt jd |zt| tjt||g jy #t$r"} t j d| zYd} ~ yd} ~ wwxYw) Nr") device_namer#platformversion onesignal_id friendly_namemobile_devices table_namekey_dict value_dictzKTautulli MobileApp :: Failed to register mobile device in the database: %s.insertzDTautulli MobileApp :: Registered mobile device '%s' in the database.zGTautulli MobileApp :: Re-registered mobile device '%s' in the database.r,)targetrT) rr upsertblacklist_logger Exceptionrwarninfo set_last_seenr Thread set_officialr) r"r/r#r0r1r3r2r(keysvaluesr)es radd_mobile_devicerFSs  ! ! #B  #D(*" * ,F "/&6RXY  Z]hhi ]`kkl|, L CC ct|jr t|}ntjd|zyt j }|jd|g}|dtk(rd|d<|S)Nz[Tautulli MobileApp :: Unable to retrieve mobile device config: invalid mobile_device_id %s.z)SELECT * FROM mobile_devices WHERE id = ?rr2r) strisdigitintrerrorrr select_single_ONESIGNAL_DISABLED)mobile_device_idr(r)s rget_mobile_device_configrOrs $$&/0 rvFF G  ! ! #B   I$4#57Fn!44!#~ Mrc t|jr t|}ntjd|zyd|i}d|j ddi}t j} |jd||tjd|zty #t$r"}tjd |zYd}~yd}~wwxYw) NzYTautulli MobileApp :: Unable to set exisiting mobile device: invalid mobile_device_id %s.Fidr3rr4r5zGTautulli MobileApp :: Updated mobile device agent: mobile_device_id %s.Tz9Tautulli MobileApp :: Unable to update mobile device: %s.) rHrIrJrrKrrr r;r?r<r=r>)rNkwargsrCrDr(rEs rset_mobile_device_configrSs $$&/0 ptDD E " #Dvzz/2> ?F  ! ! #B - P ]`ppq  ORSSTs,6B## C,C  Cctj}|r-tjd|z|j d|g}y|r-tjd|z|j d|g}yy)NzETautulli MobileApp :: Deleting mobile_device_id %s from the database.z'DELETE FROM mobile_devices WHERE id = ?rTz>Tautulli MobileApp :: Deleting device_id %s from the database.z.DELETE FROM mobile_devices WHERE device_id = ?F)rr rdebugaction)rNr"r(r)s rdelete_mobile_devicerWss  ! ! #B \_oopDL\K]^  UXaabKS\R]^rctj}t|}|dkDrdnd} |jd|||g}y#t$r"}t j d|zYd}~yd}~wwxYw)N)r2randroidz\UPDATE mobile_devices SET official = ?, platform = coalesce(platform, ?) WHERE device_id = ?rzATautulli MobileApp :: Failed to set official flag for device: %s.)rr validate_onesignal_idrVr=rr>)r"r2r(officialr0r)rEs rrBrBsy  ! ! #B$,?H$qLydH1"*8Y ?A  WZ[[\sA A- A((A-ctj}tj} |j d||g}y#t $r"}t jd|zYd}~yd}~wwxYw)Nz>UPDATE mobile_devices SET last_seen = ? WHERE device_token = ?rzBTautulli MobileApp :: Failed to set last_seen time for device: %s.)rr r timestamprVr=rr>)r#r( last_seenr)rEs rr@r@sg  ! ! #B!!#I[!*L 9;  X[\\]sA A+ A&&A+cP|y|tk(ryddi}tjd tjdt d|d| }|j }tjd |t|d k(S#t$r"}tjd |zYd}~y d}~wwxYw)Nrz Content-Typezapplication/jsonz-Tautulli MobileApp :: Validating OneSignal IDzhttps://api.onesignal.com/apps/z/subscriptions/z/user/identity)headerszETautulli MobileApp :: OneSignal ID validation returned status code %sz:Tautulli MobileApp :: Failed to validate OneSignal ID: %s.) rMrr?requestsr_ONESIGNAL_APP_ID status_coderJr=r>)r2rarrfrEs rrZrZs , ,12G KK?@ LL:;L:M_]i\jjxyDK Lmm  []hi;#%&&  PSTTUsAA:: B%B  B%cFtD]}t|d|dy)Nr"r2)r*rB)devices rrevalidate_onesignal_idsrjs*$&BVK(&*@ABrcPt}|D]}tj|yr)r*rblacklist_config)devicesds rr<r<s' "G #"#r)NFFFr)NN)NNNNNNN)rdr plexpyrrrrerMr rrr*r-rFrOrSrWrBr@rZrjr<rrrqsv$;*+()(9FJTX>"*   &B #r