U Lwa# @sddlmZddlZddlZddlmZmZmZz,ddlZ ddl Z e ddddlm Z Wne efk rtYnXddlmZmZmZmZddlmZd d ZGd d d ZGd ddejZdS))gettextN)GioGLibGObjectSnapd1)r)has_gnome_online_accountsis_current_distro_ltsis_current_distro_supportedretry) LivepatchSnapc CsD|D]6\}}ztj|||<Wqttfk r<YqXq|SN)itemsdateutilparserparse ValueError TypeError)Z json_dictkeyvaluerE/usr/lib/python3/dist-packages/softwareproperties/LivepatchService.pydatetime_parser/s rc@seZdZdZdZdZdZdS)LivepatchAvailabilityrN)__name__ __module__ __qualname__FALSETRUENO_CONNECTIVITYCHECKINGrrrrr7src@seZdZdZdZdZdZedZedZ e dde j e j e j ejfedddejfedddejfd Zd d Zd d ZddZddZddZddZeeddZddZeeddZddZddZdS) LivepatchServicezRhttp+unix://%2Fvar%2Fsnap%2Fcanonical-livepatch%2Fcurrent%2Flivepatchd.sock/statuszWhttp+unix://%2Fvar%2Fsnap%2Fcanonical-livepatch%2Fcurrent%2Flivepatchd-priv.sock/enablezXhttp+unix://%2Fvar%2Fsnap%2Fcanonical-livepatch%2Fcurrent%2Flivepatchd-priv.sock/disablez2/var/snap/canonical-livepatch/common/machine-tokenzFailed to enable Livepatch: {}zFailed to disable Livepatch: {}NF) availabilityavailability-messageenabledcCstj|d|_t|_t|_tj |_ d|_ t j j|jd}||_t j|_|jd|j|t jj|_|jd|jdS)Nr)pathznotify::connectivityZchanged)r__init__ _timeout_idr _snaprequests_unixsocketZSession_sessionrr _availability_availability_messagerZFileZ new_for_pathLIVEPATCH_RUNNING_FILE query_exists_enabledZNetworkMonitorZ get_default_nmZconnect_network_changed_cbZ monitor_fileZFileMonitorFlagsZNONEZ _lp_monitor_livepatch_enabled_changed_cb)selfZlp_filerrrr)Ws    zLivepatchService.__init__cCs8|jdkr|jS|jdkr |jS|jdkr0|jStdS)Nr%r&r')namer.r/r2AssertionError)r6Zpspecrrrdo_get_propertyms   z LivepatchService.do_get_propertycsJfdd}tj_d_ddjdkrFtd|_dS)aTrigger a Livepatch availability check to be executed after a short timeout. Multiple triggers will result in a single request. A notify::availability will be emitted when the check starts, and another one when the check ends. csfddtdfttdfttdfttdfg}jjjtjj krVt j _ d_ n4|D] \}}|sZt j_ |_ qqZt j_ d_ ddd _d S) NcsjtjjkSr )r+ get_statusr SnapStatusZUNKNOWNrr6rrz[LivepatchService.trigger_availability_check.._update_availability..z*Canonical Livepatch snap is not available.z6Gnome Online Accounts is required to enable Livepatch.z,Livepatch is not available for this release.z+The current release is no longer supported.r%r&rF)_rr r r3ZpropsZ connectivityrZNetworkConnectivityZFULLrr"r.r/r r!notifyr*)Zrulesfuncmessager<rr_update_availabilitys6     zILivepatchService.trigger_availability_check.._update_availabilityNr%r&rr)rr#r.r/r@r*rZtimeout_add_seconds)r6rCrr<rtrigger_availability_checkxs !   z+LivepatchService.trigger_availability_checkcCs^|j|krdS|s|S|jtjjkr6||S|j\}}|rR||Sd|fSdS)aEnable or disable Canonical Livepatch in the current system. This function will return once the operation succeeded or failed. Args: enabled(bool): wheater to enable or disable the service. token(str): the authentication token to be used to enable Canonical Livepatch service. Returns: (False, '') if successful, (True, error_message) otherwise. )FTN) r2_disable_servicer+r:rr;ZACTIVE_enable_serviceZenable_or_install)r6r'tokenZsuccessmsgrrr set_enableds  zLivepatchService.set_enabledc Cslz*ddd}|jj|j|d}|jtdWStk rf}ztdt |WYdSd}~XYnXdS)zSynchronously retrieve the status of Canonical Livepatch. Returns: str: The status. A valid string for success, None otherwise. rjson) verbosityformatparams)Z object_hookz"Failed to get Livepatch status: {}N) r-getSTATUS_ENDPOINTrKr ExceptionloggingdebugrMstr)r6rOrerrrr:s zLivepatchService.get_statusc CsLz ||WStk rF}zd|jt|fWYSd}~XYnXdS)a`Enable Canonical Livepatch in the current system. This function will return once the operation succeeded or failed. Args: token(str): the authentication token to be used to enable Canonical Livepatch service. Returns: (False, '') if successful, (True, error_message) otherwise. TN)_enable_service_with_retryrRENABLE_ERROR_MSGrMrU)r6rHrWrrrrGs  z LivepatchService._enable_servicecCs:d|i}|jj|j|d}|j |jr*dn |j|jfS)Nz auth-tokenrNrE)r-putENABLE_ENDPOINTokrYrMtext)r6rHrOrVrrrrXsz+LivepatchService._enable_service_with_retryc CsJz |WStk rD}zd|jt|fWYSd}~XYnXdS)zDisable Canonical Livepatch in the current system. This function will return once the operation succeeded or failed. Returns: (False, '') if successful, (True, error_message) otherwise. TN)_disable_service_with_retryrRDISABLE_ERROR_MSGrMrU)r6rWrrrrFs z!LivepatchService._disable_servicecCs.|j|j}|j |jrdn |j|jfS)NrE)r-rZDISABLE_ENDPOINTr\r_rMr])r6rVrrrr^sz,LivepatchService._disable_service_with_retrycCs |dSr )rD)r6ZmonitorZnetwork_availablerrrr4sz$LivepatchService._network_changed_cbcCs&|}|j|kr"||_|ddS)Nr')r1r2r@)r6ZfmfileZ other_fileZ event_typer'rrrr5s z.LivepatchService._livepatch_enabled_changed_cb) rrrrQr[r`r0r?rYr_intrr r#rZPARAM_READABLErUboolZ__gproperties__r)r9rDrJr:rGr rRrXrFr^r4r5rrrrr$>sP  0   r$)rr?rSZgiZ gi.repositoryrrrZdateutil.parserrr,Zrequire_versionr ImportErrorrZsoftwareproperties.gtk.utilsrr r r Z softwareproperties.LivepatchSnapr rrr$rrrrs