o l>jE @sUdZddlZddlZddlZddlmZmZddlmZddlm Z m Z m Z m Z ddl mZddlmZmZmZddlmZdd lmZd Zeed <eeZejed <d Zeed< dde de dej!de"ddf ddZ#GdddeZ$GdddZ%dS)zFind my iPhone service.N)datetime timedelta)MappingProxyType)AnyCallableIteratorOptional)Response)PyiCloudAuthRequiredExceptionPyiCloudNoDevicesExceptionPyiCloudServiceUnavailable) BaseService)PyiCloudSessionz US/Pacific_FMIP_CLIENT_CONTEXT_TIMEZONE_LOGGER_MAX_REFRESH_RETRIESFintervalfunc stop_eventlocatereturnc Cstt|d}|j|dsE|tkr=z||Wnty3}z td|WYd}~nd}~wwtt|d}|j|drdSdS)z:Thread function to monitor the FindMyiPhoneServiceManager.)seconds)timeoutz%FindMyiPhone monitor thread error: %sN)rnowrwait Exceptionrdebug)rrrr next_eventexcr S/home/thesage/.local/lib/python3.10/site-packages/pyicloud/services/findmyiphone.py_monitor_threads   r"c s6eZdZdZ  d,dedededeeefdedBd df fd d Z d-d e d e d dfddZ d e d dfddZ d.ddZ d e d dfddZd/d e d dfddZdeeBd dfddZd efddZd efdd Zd edfd!d"Zd efd#d$Zed e fd%d&Zed0d(d)Zed eeeeffd*d+ZZS)1FindMyiPhoneServiceManagerzThe 'Find my iPhone' iCloud service This connects to iCloud and return phone data including the near-realtime latitude and longitude. FN service_roottoken_endpointsessionparamsrefresh_intervalrcst|||||_|dur|nd|_|d}|d|_|d|_|d|_|d|_|d|_|d |_ |d |_ i|_ g|_ d|_ d|_d|_t|_|jd d dS) z*Initialize the FindMyiPhoneServiceManager.Ngr@z/fmipservice/client/webz /initClientz/refreshClientz /playSoundz /sendMessagez /lostDevicez/remoteWipeWithUserAuthz/fmipWebAuthenticateTr)super__init__ _with_family_refresh_interval_fmip_init_url_fmip_refresh_url_fmip_sound_url_fmip_message_url_fmip_lost_url_fmip_erase_url_erase_token_url_devices_devices_names _server_ctx _user_info_monitor threadingEventr_refresh_client_with_reauth)selfr$r%r&r' with_familyr( fmip_endpoint __class__r r!r+.s&          z#FindMyiPhoneServiceManager.__init__rretrycCs|jz|j|dWn&ty3|durtdd|_|jjj dd|j |ddYdSw|j |d| dS)z Refreshes the FindMyiPhoneService endpoint with re-authentication. This ensures that the location data is up-to-date. r)TzRe-authenticating sessionN) force_refresh)rrB) rset_refresh_clientr rrr7r&service authenticater<_initialize_devices_start_monitor_thread)r=rrBr r r!r<Ps    z6FindMyiPhoneServiceManager._refresh_client_with_reauthcCsd}t}|jrT|jrT|jddrT|tkrTdd|jdiD}|s'n-||kr2td|n"|}t d|j |d |d 7}|jrT|jrT|jddrT|tks|j sZt t d t|j d S) z;Initializes the devices for the FindMyiPhoneServiceManager.r hasMembersFcSs"h|] \}}|ddkr|qS)deviceFetchStatusLOADING)get).0kvr r r! xs zAFindMyiPhoneServiceManager._initialize_devices.. membersInfoz>No progress on LOADING family members after retry %d, stoppingg?r)zNumber of devices found: %dN)rDr,r8rMritemsrrtimesleeprEr5r infolen)r=rretriesprev_loading_keys loading_keysr r r!rHgsD     z.FindMyiPhoneServiceManager._initialize_devicescCsB|jstjt|j|j|jddd|_|j|j dSdS)z=Starts the monitor thread for the FindMyiPhoneServiceManager.)rrrT)targetkwargsdaemonN) is_aliver:Threadr"rEr-rr9clearstartr=r r r!rIs z0FindMyiPhoneServiceManager._start_monitor_threadc Cs@ddddd|jtddi}|jr%|j|d<|r%d |d <|dd d d |jj|jr.|jn|j|j|d }| }| d|_|jrMd|jvrMd|jd<| d|_ d|vr^t ddSt dt|d|dD])}|d}||jvrt||j||j|j|j|j|jd|j|<ql|j||qlt|j|_dS)zx Refreshes the FindMyiPhoneService endpoint, this ensures that the location data is up-to-date. clientContextziCloud Find (Web)z2.0z3.0rSr)appName appVersion apiVersiondeviceListVersionfmlytimezone inactiveTime serverContextTisUpdatingAllLocationsall) shouldLocateselectedDevice)urlr'json theftLossNuserInfocontentzFMIP returned 0 deviceszFMIP returned %d devicesid)manager sound_urllost_url message_url erase_urlerase_token_url)r,rr7updater&postr/r.r'rrrMr8rrrXr5 AppleDevicer0r2r1r3r4listkeysr6)r=rreq_jsonreqresp device_info device_idr r r!rEs`        z*FindMyiPhoneServiceManager._refresh_clientTcCs|j|ddS)z:Public method to refresh the FindMyiPhoneService endpoint.r)N)r<)r=rr r r!refreshsz"FindMyiPhoneServiceManager.refreshkeyrcCs0|js |jddt|tr|j|}|j|S)zGets a device by name or index.Tr))r_r< isinstanceintr6r5r=rr r r! __getitem__s     z&FindMyiPhoneServiceManager.__getitem__cCs|jS)z%String representation of the devices.)r5rcr r r!__str__sz"FindMyiPhoneServiceManager.__str__cCs|S)Representation of the device.r rcr r r!__repr__sz#FindMyiPhoneServiceManager.__repr__cCs |js |jddt|jS)zIterates over the devices.Tr))r_r<iterr5valuesrcr r r!__iter__s z#FindMyiPhoneServiceManager.__iter__cCs|js |jddt|jS)zReturns the number of devices.Tr))r_r<rXr5rcr r r!__len__s  z"FindMyiPhoneServiceManager.__len__cCs|jduo |jS)z"Indicates if the service is alive.N)r9r_rcr r r!r_z#FindMyiPhoneServiceManager.is_alive"MappingProxyType[str, AppleDevice]cCs t|jS)zReturns the devices.)rr5rcr r r!devicess z"FindMyiPhoneServiceManager.devicescCs|jrt|jSdS)zReturns the user info.N)r8rrcr r r! user_info rz$FindMyiPhoneServiceManager.user_info)FNFrN)T)rr)__name__ __module__ __qualname____doc__strrdictrfloatr+boolr<rHrIrErrrrrrrrpropertyr_rrrr __classcell__r r r@r!r#'sB  " (@  $r#c@seZdZdZdeeefdeeefdededededed ed d fd d Ze d e fddZ dEddZ e d e eeeffddZdFde eed eeeffddZdGdEddZ     dH dEddZ  dId ed!ed"ed d fd#d$Zd efd%d&Z  dId!ed"ed d fd'd(Ze d eeeffd)d*Zd efd+d,Zd efd-d.Ze d efd/d0Ze d efd1d2Ze d efd3d4Ze d efd5d6Ze d efd7d8Ze d efd9d:Ze d efd;d<Z e d efd=d>Z!e d efd?d@Z"d efdAdBZ#d efdCdDZ$d S)Jrz Apple device.rur'rwrxryr{r|rzrNc Cs4||_||_||_||_||_||_||_||_dS)zInitialize the Apple device.N)_content_manager_params _sound_url _lost_url _erase_urlr4 _message_url) r=rur'rwrxryr{r|rzr r r!r+s  zAppleDevice.__init__cCs|jjS)zGets the session.)rr&rcr r r!r&)szAppleDevice.sessioncCs ||_dS)zUpdates the device data.N)rr=datar r r!r}.s zAppleDevice.updatecCs|jdurdS|jdS)zUpdates the device location.FNlocation)location_availablerrcr r r!r2s  zAppleDevice.location additionalcCs:gd}|dur ||7}i}|D] }|j|||<q|S)zkReturns status information for device. This returns only a subset of possible properties. ) batteryLeveldeviceDisplayName deviceStatusnameN)rrM)r=rfields propertiesfieldr r r!status9szAppleDevice.statusFind My iPhone AlertcCsB|jdur td|jd|ddid}|jj|j|j|ddS) zSend a request to the device to play a sound. It's possible to pass a custom message by changing the `subject`. Fz&Sound is not available for this devicervriT)devicesubjectrdr'rrN)sound_availabler rr&r~rr)r=rrr r r! play_soundLs zAppleDevice.play_soundThis is a noteFcCsF|jdur td|jd|d||||d}|jj|j|j|ddS)zSend a request to the device to a display a message. It's possible to pass a custom message by changing the `subject`. Fz*Messaging is not available for this devicervT)rruserTexttextsoundvibratestroberN)messaging_availabler rr&r~rr)r=rmessagesoundsrrrr r r!display_message[s  zAppleDevice.display_message*This device has been lost. Please call me.numberr newpasscodecCsF|jdur td|d|dd|jd|d}|jj|j|j|ddS)zSend a request to the device to trigger 'lost mode'. The device will show the message in `text`, and if a number has been passed, then the person holding the device can call the number without entering the passcode. Fz*Lost mode is not available for this deviceTrv)rrownerNbrlostModeEnabledtrackingEnabledrpasscoderN)lost_mode_availabler rr&r~rr)r=rrrrr r r! lost_devicexs  zAppleDevice.lost_devicecCsPd|jjdi}|jj|j|d}d|vsd|dvr"td|ddS)z3Get the erase token for the Find My iPhone service.dsWebAuthToken session_token)rqrrtokensmmeFMIPWebEraseDeviceTokenz(Find My iPhone erase token not available)r&rrMr~r4rrr rr r r!_get_erase_tokens  zAppleDevice._get_erase_tokencCsD|jdur td|||jd|d}|jj|j|j|ddS)z5Send a request to the device to start a remote erase.Fz&Erase is not available for this devicerv) authTokenrrrrN)erase_availabler rrr&r~rr)r=rrrr r r! erase_devices zAppleDevice.erase_devicecCs|jjs |j|jS)zGets the device data.rr_rrrcr r r!rs zAppleDevice.datacCs|jjs |j|j|S)%Gets an attribute of the device data.rrr r r!rs  zAppleDevice.__getitem__cCsB|jjs |j||jvr|j|Stdt|jd|d)r'z' object has no attribute ')rr_rrAttributeErrortyper)r=attrr r r! __getattr__s   zAppleDevice.__getattr__cC|jddS)zGets the device name.rrrrMrcr r r!rzAppleDevice.namecCr)zGets the device model. deviceModelrrrcr r r!modelrzAppleDevice.modelcCr)zGets the device model name.rrrrcr r r! model_namerzAppleDevice.model_namecCr)zGets the device type. deviceClassrrrcr r r! device_typerzAppleDevice.device_typecCr)z3Indicates if lost mode is available for the device.lostModeCapableFrrcr r r!rrzAppleDevice.lost_mode_availablecC|jdiddS)z3Indicates if messaging is available for the device.featuresMSGFrrcr r r!rzAppleDevice.messaging_availablecCr)z/Indicates if sound is available for the device.rSNDFrrcr r r!rrzAppleDevice.sound_availablecCr)z/Indicates if erase is available for the device.rWIPFrrcr r r!rrzAppleDevice.erase_availablecCs.|jdiddod|jvo|jdduS)z2Indicates if location is available for the device.rLOCFrN)rrMrrcr r r!rs  zAppleDevice.location_availablecCs|jd|jS)z$String representation of the device.z: )rrrcr r r!rszAppleDevice.__str__cCs d|dS)rz r rcr r r!rs zAppleDevice.__repr__r)N)r)rrFFF)rr)%rrrrrrrr#r+rrr&r}rrrrrrrrrrrrrrrrrrrrrrrrr r r r!rs     $       rr)&rloggingr:rUrrtypesrtypingrrrrrequestsr pyicloud.exceptionsr r r pyicloud.services.baser pyicloud.sessionrrr__annotations__ getLoggerrrLoggerrrrr;rr"r#rr r r r!s:       j