Your IP : 216.73.217.13
U
8�-d96 � @ sT d dl Z d dlmZmZmZ d dlmZ d dlmZ d dl m
Z
G dd� de �Z G dd � d e �Z
G d
d � d e �ZG d d
� d
e �ZG dd� de �ZG dd� de �ZG dd� de �ZG dd� de �ZG dd� de �ZG dd� de �ZG dd� de �ZG d d � d e �ZG d d � d e �ZG d d!� d!e �ZG d"d#� d#e �ZG d$d%� d%e �ZG d&d'� d'e �Z G d(d)� d)e �Z G d*d+� d+e �Z G d,d-� d-e �Z G d.d/� d/e �Z G d0d1� d1e �Z!G d2d3� d3e �Z"G d4d5� d5e �Z#G d6d7� d7e �Z$G d8d9� d9e �Z%G d:d;� d;e �Z&G d<d=� d=e �Z'G d>d?� d?e �Z(G d@dA� dAe �Z)G dBdC� dCe �Z*G dDdE� dEe �Z+G dFdG� dGe �Z,G dHdI� dIe,�Z-G dJdK� dKe,�Z.G dLdM� dMe,�Z/G dNdO� dOe �Z0G dPdQ� dQe1�Z2G dRdS� dSe1�Z3G dTdU� dUe2�Z4G dVdW� dWe2�Z5G dXdY� dYe �Z6G dZd[� d[e �Z7G d\d]� d]e7�Z8G d^d_� d_e7�Z9G d`da� dae �Z:G dbdc� dce �Z;G ddde� dee �Z<G dfdg� dge �Z=dS )h� N)�Any�Dict�Optional)�error)�messages)�PRINT_WRAP_WIDTHc @ s: e Zd ZdZdZdeee eeeef dd�dd�Z dS )�UserFacingErrorz�
An exception to be raised when an execution-ending error is encountered.
:param msg:
Takes a single parameter, which is the user-facing error message that
should be emitted before exiting non-zero.
� N)�msg�msg_code�additional_info�returnc C s || _ || _|| _d S �N)r
r r )�selfr
r r � r �5/usr/lib/python3/dist-packages/uaclient/exceptions.py�__init__ s zUserFacingError.__init__)NN)
�__name__�
__module__� __qualname__�__doc__� exit_code�strr r r r r r r r r s ��r c s& e Zd Zeedd�� fdd� Z� ZS )�APTInstallErrorN)�name� service_msgr
c s t � jtjjtjjd� d S �N�r
r )�superr r ZAPT_INSTALL_FAILEDr
r )r r r �� __class__r r r s �zAPTInstallError.__init__�r r r r r �
__classcell__r r r r r s r c s e Zd Z� fdd�Z� ZS )�APTProcessConflictErrorc s t � jtjjtjjd� d S r )r r r ZAPT_PROCESS_CONFLICTr
r �r r r r r ( s �z APTProcessConflictError.__init__�r r r r r" r r r r r# ' s r# c s$ e Zd Zedd�� fdd� Z� ZS )�APTInvalidRepoErrorN�� error_msgr
c s t � j|d� d S )N�r
)r r )r r( r r r r 0 s z APTInvalidRepoError.__init__r! r r r r r& / s r& c s" e Zd Zdd�� fdd� Z� ZS )� APTUpdateProcessConflictErrorN�r
c s t � jtjjtjjd� d S r )r r r ZAPT_UPDATE_PROCESS_CONFLICTr
r r$ r r r r 5 s �z&APTUpdateProcessConflictError.__init__r% r r r r r* 4 s r* c s$ e Zd Zedd�� fdd� Z� ZS )�APTUpdateInvalidRepoErrorN)�repo_msgr
c s& t jj|d�}t� j|j|jd� d S )N)r- r )r ZAPT_UPDATE_INVALID_REPO�formatr r r
r )r r- r
r r r r = s z"APTUpdateInvalidRepoError.__init__r! r r r r r, < s r, c s* e Zd Zdee dd�� fdd�
Z� ZS )� APTInstallProcessConflictErrorN)�
header_msgr
c s2 |r |d7 }t jj|d�}t� j|j|jd� d S )N�.
)r0 r )r Z APT_INSTALL_PROCESS_CONFLICTr. r r r
r )r r0 r
r r r r C s �z'APTInstallProcessConflictError.__init__)N)r r r r r r r" r r r r r/ B s r/ c s, e Zd Zdeee dd�� fdd�
Z� ZS )�APTInstallInvalidRepoErrorN)r- r0 r
c s4 |r |d7 }t jj||d�}t� j|j|jd� d S )Nr1 )r0 r- r )r ZAPT_INSTALL_INVALID_REPOr. r r r
r )r r- r0 r
r r r r O s �z#APTInstallInvalidRepoError.__init__)N)r r r r r r r" r r r r r2 N s � �r2 c s& e Zd Zeedd�� fdd� Z� ZS )� SnapdNotProperlyInstalledErrorN)�snap_cmd�servicer
c s( t jj||d�}t� j|j|jd� d S )N)r4 r5 r )r Z SNAPD_NOT_PROPERLY_INSTALLEDr. r r r
r )r r4 r5 r
r r r r \ s
�z'SnapdNotProperlyInstalledError.__init__r! r r r r r3 [ s r3 c s" e Zd Zdd�� fdd� Z� ZS )�CannotInstallSnapdErrorNr+ c s t j}t� j|j|jd� d S r )r ZCANNOT_INSTALL_SNAPDr r r
r �r r
r r r r e s z CannotInstallSnapdError.__init__r% r r r r r6 d s r6 c s$ e Zd Zedd�� fdd� Z� ZS )�ErrorInstallingLivepatchNr' c s& t jj|d�}t� j|j|jd� d S )N�r( r )r ZERROR_INSTALLING_LIVEPATCHr. r r r
r �r r( r
r r r r k s z!ErrorInstallingLivepatch.__init__r! r r r r r8 j s r8 c s( e Zd Zeeedd�� fdd� Z� ZS )� InvalidServiceToDisableErrorN)� operation�invalid_servicer r
c s* t jj|||d�}t� j|j|jd� d S )N)r< r= r r )r ZINVALID_SERVICE_OP_FAILUREr. r r r
r )r r< r= r r
r r r r q s �z%InvalidServiceToDisableError.__init__r! r r r r r; p s
�r; c s" e Zd Zed�� fdd� Z� ZS )�ProxyNotWorkingError��proxyc s$ t � jtjj|d�jtjjd� d S �Nr? r )r r r Z NOT_SETTING_PROXY_NOT_WORKINGr. r
r �r r@ r r r r } s �z ProxyNotWorkingError.__init__r! r r r r r> | s r> c s" e Zd Zed�� fdd� Z� ZS )�ProxyInvalidUrlr? c s$ t � jtjj|d�jtjjd� d S rA )r r r Z NOT_SETTING_PROXY_INVALID_URLr. r
r rB r r r r � s �zProxyInvalidUrl.__init__r! r r r r rC � s rC c @ s e Zd ZdZdZdS )�NonAutoAttachImageErrorz>Raised when machine isn't running an auto-attach enabled imager N)r r r r r r r r r rD � s rD c s* e Zd ZdZdZed�� fdd� Z� ZS )�AlreadyAttachedErrorzDAn exception to be raised when a command needs an unattached system.� �� account_namec s& t jj|d�}t� j|j|jd� d S )NrG r )r ZALREADY_ATTACHEDr. r r r
r )r rH r
r r r r � s z AlreadyAttachedError.__init__)r r r r r r r r" r r r r rE � s rE c s$ e Zd ZdZdZ� fdd�Z� ZS )� AttachErrorz@An exception to be raised when we detect a generic attach error.r c s t j}t� j|j|jd� d S r )r ZATTACH_FAILUREr r r
r r7 r r r r � s zAttachError.__init__)r r r r r r r" r r r r rI � s rI c s& e Zd Zeedd�� fdd� Z� ZS )� AttachInvalidConfigFileErrorN)� config_namer r
c s2 t jj||d�}t� jtj|jtd�|j d� d S )N)rK r )�widthr )
r ZATTACH_CONFIG_READ_ERRORr. r r �textwrapZfillr
r r )r rK r r
r r r r � s ��z%AttachInvalidConfigFileError.__init__r! r r r r rJ � s rJ c s e Zd Z� fdd�Z� ZS )�AttachInvalidTokenErrorc s t � jtjjtjjd� d S r )r r r ZATTACH_INVALID_TOKENr
r r$ r r r r � s �z AttachInvalidTokenError.__init__r% r r r r rN � s rN c s e Zd Z� fdd�Z� ZS )�ConnectivityErrorc s t � jtjjtjjd� d S r )r r r ZCONNECTIVITY_ERRORr
r r$ r r r r � s �zConnectivityError.__init__r% r r r r rO � s rO c s e Zd Z� fdd�Z� ZS )� MagicAttachTokenAlreadyActivatedc s t j}t� j|j|jd� d S r )r Z$MAGIC_ATTACH_TOKEN_ALREADY_ACTIVATEDr r r
r r7 r r r r � s
�z)MagicAttachTokenAlreadyActivated.__init__r% r r r r rP � s rP c s e Zd Z� fdd�Z� ZS )�MagicAttachTokenErrorc s t j}t� j|j|jd� d S r )r ZMAGIC_ATTACH_TOKEN_ERRORr r r
r r7 r r r r � s
�z MagicAttachTokenError.__init__r% r r r r rQ � s rQ c s" e Zd Zed�� fdd� Z� ZS )�MagicAttachInvalidEmail��emailc s& t jj|d�}t� j|j|jd� d S )NrS r )r ZMAGIC_ATTACH_INVALID_EMAILr. r r r
r )r rT r
r r r r � s
�z MagicAttachInvalidEmail.__init__r! r r r r rR � s rR c s e Zd Z� fdd�Z� ZS )�MagicAttachUnavailablec s t j}t� j|j|jd� d S r )r ZMAGIC_ATTACH_UNAVAILABLEr r r
r r7 r r r r � s
�z MagicAttachUnavailable.__init__r% r r r r rU � s rU c s e Zd Z� fdd�Z� ZS )�MagicAttachInvalidParamc s( t jj||d�}t� j|j|jd� d S )N)�param�valuer )r ZMAGIC_ATTACH_INVALID_PARAMr. r r r
r )r rW rX r
r r r r � s
�z MagicAttachInvalidParam.__init__r% r r r r rV � s rV c s* e Zd ZdZeeed�� fdd� Z� ZS )�
LockHeldErrora An exception for when another pro operation is in progress
:param lock_request: String of the command requesting the lock
:param lock_holder: String of the command that currently holds the lock
:param pid: Integer of the process id of the lock_holder
�� lock_request� lock_holder�pidc s6 || _ || _tjj|||d�}t� j|j|jd� d S )NrZ r ) r\ r] r ZLOCK_HELD_ERRORr. r r r
r )r r[ r\ r] r
r r r r � s �zLockHeldError.__init__)r r r r r �intr r" r r r r rY � s rY c s e Zd ZdZ� fdd�Z� ZS )�MissingAptURLDirectivez@An exception for when the contract server doesn't include aptURLc s& t jj|d�}t� j|j|jd� d S )N��entitlement_namer )r ZMISSING_APT_URL_DIRECTIVEr. r r r
r �r ra r
r r r r s �z MissingAptURLDirective.__init__�r r r r r r" r r r r r_ � s r_ c s& e Zd ZdZdd�� fdd� Z� ZS )�NonRootUserErrorz7An exception to be raised when a user needs to be root.Nr+ c s t � jtjjtjjd� d S r )r r r Z NONROOT_USERr
r r$ r r r r s �zNonRootUserError.__init__rc r r r r rd s rd c s0 e Zd ZdZejfejdd�� fdd�
Z� ZS )�UnattachedErrorz>An exception to be raised when a machine needs to be attached.N)r
r
c s t � j|j|jd� d S r )r r r
r r7 r r r r s zUnattachedError.__init__) r r r r r Z
UNATTACHEDZ NamedMessager r" r r r r re s ��re c s* e Zd ZdZeedd�� fdd� Z� ZS )�SecurityAPIMetadataErrorzDAn exception raised with Security API metadata returns invalid data.N)r
�issue_idr
c s& t � �d| d tjj|d� � d S )NzError: �
)Zissue)r r r ZSECURITY_ISSUE_NOT_RESOLVEDr. )r r
rg r r r r s �� ��z!SecurityAPIMetadataError.__init__)r r r r r r r" r r r r rf s rf c s" e Zd Zed�� fdd� Z� ZS )�InvalidProImager9 c s, || _ tjj|d�}t� j|j|jd� d S )Nr) r )Zcontract_server_msgr ZINVALID_PRO_IMAGEr. r r r
r r: r r r r ( s zInvalidProImage.__init__r! r r r r ri ' s ri c s6 e Zd ZdZdeee d�� fdd�
Zdd� Z� ZS ) �GCPProAccountErrorz?An exception raised when GCP Pro service account is not enabledNr c s || _ t� �||� d S r )�coder r )r r
r rk r r r r 1 s zGCPProAccountError.__init__c C s dj | j| jd�S )Nz(GCPProServiceAccount Error {code}: {msg})rk r
)r. rk r
r$ r r r �__str__5 s �zGCPProAccountError.__str__)N) r r r r r r r rl r" r r r r rj . s rj c @ s e Zd Zee dd�dd�ZdS )�CloudFactoryErrorN)�
cloud_typer
c C s
|| _ d S r )rn )r rn r r r r < s zCloudFactoryError.__init__)r r r r r r r r r r rm ; s rm c @ s e Zd ZdS )�CloudFactoryNoCloudErrorN�r r r r r r r ro @ s ro c @ s e Zd ZdS )�!CloudFactoryUnsupportedCloudErrorNrp r r r r rq D s rq c @ s e Zd ZdS )� CloudFactoryNonViableCloudErrorNrp r r r r rr H s rr c s" e Zd Zed�� fdd� Z� ZS )�EntitlementNotFoundErrorr` c s& t jj|d�}t� j|j|jd� d S )N)r r )r ZENTITLEMENT_NOT_FOUNDr. r r r
r rb r r r r M s z!EntitlementNotFoundError.__init__r! r r r r rs L s rs c s@ e Zd Zdejee eeeef ee d�� fdd�
Z � Z
S )�UrlErrorN)�causerk �headers�urlc sR t |dd �rt|j�}nt|�}t� �|� || _|| _| jd krHi | _|| _d S )N�reason)�getattrr rx r r rk rv rw )r ru rk rv rw Z cause_errorr r r r S s
zUrlError.__init__)NNN) r r r r ZURLErrorr r^ r r r r" r r r r rt R s ��rt c s0 e Zd Zdeee eedd�� fdd�
Z� ZS )�ProcessExecutionErrorN� )�cmdr �stdout�stderrr
c s< || _ || _|| _|s d}nd}t� �|j|||d�� d S )Nz"Invalid command specified '{cmd}'.zEFailed running command '{cmd}' [exit({exit_code})]. Message: {stderr})r| r~ r )r} r~ r r r r. )r r| r r} r~ Z message_tmplr r r r g s ��z ProcessExecutionError.__init__)Nr{ r{ )r r r r r r^ r r" r r r r rz f s ��rz c s: e Zd Z� fdd�Zdd� Zd
dd�Z� fdd �Z� ZS ) �ContractAPIErrorc s. t � �||j|j|j� |d kr$i }|| _d S r )r r rk rv rw � api_error�r �eZerror_responser r r r ~ s zContractAPIError.__init__c C s0 || j �d�krdS | j �dd��|�r,dS dS )Nrk T�messager{ F)r� �get�
startswith�r �
error_coder r r � __contains__� s
z ContractAPIError.__contains__Nc C s | j �d�|kr | j �d�S |S )Nrk r� )r� r� )r r� �defaultr r r �__get__� s zContractAPIError.__get__c s2 t � �� }|d t| j� d d | j�dd� S )N�: [�]�, r� r{ )r rl r rw r� r� )r �prefixr r r rl � s
���� ��zContractAPIError.__str__)N�r r r r r� r� rl r" r r r r r } s
r c s: e Zd Z� fdd�Zdd� Zd
dd�Z� fdd �Z� ZS ) �SecurityAPIErrorc s* t � �||j|j|j� |�dd�| _d S )Nr� r{ )r r rk rv rw r� r� r� r r r r � s zSecurityAPIError.__init__c C s t || jk�S r )�boolr� r� r r r r� � s z SecurityAPIError.__contains__Nc C s || j kr| j S |S r )r� )r Z error_strr� r r r r� � s
zSecurityAPIError.__get__c sL t � �� }| jg}|r6|d t| j� d d�|� S |d t| j� d S )Nr� z] r� r� )r rl r� r rw �join)r r� Zdetailsr r r rl � s
zSecurityAPIError.__str__)Nr� r r r r r� � s
r� c @ s e Zd ZdS )� InPlaceUpgradeNotSupportedErrorNrp r r r r r� � s r� c @ s e Zd ZdS )�IsProLicensePresentErrorNrp r r r r r� � s r� c @ s e Zd ZdS )�CancelProLicensePollingNrp r r r r r� � s r� c @ s e Zd ZdS )�DelayProLicensePollingNrp r r r r r� � s r� c s& e Zd Zeedd�� fdd� Z� ZS )�InvalidFileFormatErrorN)� file_name� file_formatr
c s( t jj||d�}t� j|j|jd� d S )N)r� r� r )r ZINVALID_FILE_FORMATr. r r r
r )r r� r� r
r r r r � s
�z InvalidFileFormatError.__init__r! r r r r r� � s r� c s$ e Zd Zeed�� fdd� Z� ZS )�ParsingErrorOnOSReleaseFile��orig_ver�mod_verc s( t jj||d�}t� j|j|jd� d S )Nr� r )r Z ERROR_PARSING_VERSION_OS_RELEASEr. r r r
r )r r� r� r
r r r r � s
�z$ParsingErrorOnOSReleaseFile.__init__r! r r r r r� � s r� c s e Zd Z� fdd�Z� ZS )� MissingSeriesOnOSReleaseFilec s& t jj|d�}t� j|j|jd� d S )N)�versionr )r Z MISSING_SERIES_ON_OS_RELEASEr. r r r
r )r r� r
r r r r � s z%MissingSeriesOnOSReleaseFile.__init__r% r r r r r� � s r� c s e Zd Z� fdd�Z� ZS )�InvalidLockFilec s& t jj|d�}t� j|j|jd� d S )N)�lock_file_pathr )r ZINVALID_LOCK_FILEr. r r r
r )r r� r
r r r r � s zInvalidLockFile.__init__r% r r r r r� � s r� )>rM �typingr r r Zurllibr Zuaclientr Zuaclient.defaultsr � Exceptionr r r# r& r* r, r/ r2 r3 r6 r8 r; r> rC rD rE rI rJ rN rO rP rQ rR rU rV rY r_ rd re rf ri rj rm ro rq rr rs �IOErrorrt rz r r� r� r� r� r� r� r� r� r� r r r r �<module> sj
"