3 .(]# @sddlmZddlZddlZddlmZmZmZy,ddlZ ddl Z ej 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 CsHxB|jD]6\}}ytjj|||<Wq ttfk r>Yq Xq W|S)N)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) availabilityzavailability-messageenabledcCstjj|d|_t|_tj|_tj |_ d|_ t j j|jd}|j|_t jj|_|jjd|j|jt jj|_|jjd|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$zavailability-messager%)namer,r-r0AssertionError)r4Zpspecrrrdo_get_propertyms   z LivepatchService.do_get_propertycsJfdd}tj_d_jdjdjdkrFtjd|_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_ n6x4|D]\}}|s\t j_ |_ Pq\Wt j_ d_ jdjdd _d S) NcsjjtjjkS)N)r) get_statusr SnapStatusZUNKNOWNr)r4rrsz[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$zavailability-messagerF)_rr r r1ZpropsZ connectivityrZNetworkConnectivityZFULLrr!r,r-rr notifyr()Zrulesfuncmessage)r4rr_update_availabilitys,    zILivepatchService.trigger_availability_check.._update_availabilityNr$zavailability-messagerr)rr"r,r-r<r(rZtimeout_add_seconds)r4r?r)r4rtrigger_availability_checkxs !   z+LivepatchService.trigger_availability_checkcCs^|j|krdS|s|jS|jjtjjkr6|j|S|jj\}}|rR|j|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)FrA) r0_disable_servicer)r8rr9ZACTIVE_enable_serviceZenable_or_install)r4r%tokenZsuccessmsgrrr set_enableds  zLivepatchService.set_enabledcCsby(ddd}|jj|j|d}|jtdStk r\}ztjdjt |dSd}~XnXdS)zSynchronously retrieve the status of Canonical Livepatch. Returns: str: The status. A valid string for success, None otherwise. rjson) verbosityformat)params)Z object_hookz"Failed to get Livepatch status: {}N) r+getSTATUS_ENDPOINTrGr ExceptionloggingdebugrIstr)r4rJrerrrr8s  zLivepatchService.get_statuscCs@y |j|Stk r:}zd|jjt|fSd}~XnXdS)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_retryrMENABLE_ERROR_MSGrIrP)r4rDrRrrrrCs  z LivepatchService._enable_servicecCs:d|i}|jj|j|d}|j |jr*dn |jj|jfS)Nz auth-token)rJrA)r+putENABLE_ENDPOINTokrTrItext)r4rDrJrQrrrrSsz+LivepatchService._enable_service_with_retrycCs>y|jStk r8}zd|jjt|fSd}~XnXdS)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_retryrMDISABLE_ERROR_MSGrIrP)r4rRrrrrBsz!LivepatchService._disable_servicecCs.|jj|j}|j |jrdn |jj|jfS)NrA)r+rUDISABLE_ENDPOINTrWrZrIrX)r4rQrrrrYsz,LivepatchService._disable_service_with_retrycCs |jdS)N)r@)r4ZmonitorZnetwork_availablerrrr2sz$LivepatchService._network_changed_cbcCs&|j}|j|kr"||_|jddS)Nr%)r/r0r<)r4ZfmfileZ other_fileZ event_typer%rrrr3s z.LivepatchService._livepatch_enabled_changed_cb) rrrrLrVr[r.r;rTrZintrrr"rZPARAM_READABLErPboolZ__gproperties__r'r7r@rFr8rCr rMrSrBrYr2r3rrrrr#>s0 0 r#)rr;rNZgiZ gi.repositoryrrrZdateutil.parserrr*Zrequire_versionr ImportErrorrZsoftwareproperties.gtk.utilsrr r r Z softwareproperties.LivepatchSnapr rrr#rrrrs