File: //lib/python3/dist-packages/uaclient/__pycache__/exceptions.cpython-310.pyc
o
S��ce3 � @ s$ 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:dS )b� N)�Any�Dict�Optional)�error)�messages)�PRINT_WRAP_WIDTHc
@ sD e Zd ZdZdZ d
dedee deeeef ddfdd �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 �* e Zd Zdededdf� fdd� Z� ZS )�APTInstallError�name� service_msgr
Nc � t � jtjjtjjd� d S �N�r
r )�superr r �APT_INSTALL_FAILEDr
r )r r r �� __class__r r r �
�zAPTInstallError.__init__�r r r r r �
__classcell__r r r" r r � "r c � e Zd Z� fdd�Z� ZS )�APTProcessConflictErrorc r r )r r r �APT_PROCESS_CONFLICTr
r �r r" r r r ( r$ z APTProcessConflictError.__init__�r r r r r& r r r" r r) ' � r) c �& e Zd Zdeddf� fdd� Z� ZS )�APTInvalidRepoError� error_msgr
Nc s t � j|d� d S )N�r
)r r )r r0 r" r r r 0 s z APTInvalidRepoError.__init__r% r r r" r r/ / � r/ c � e Zd Zd� fdd� Z� ZS )� APTUpdateProcessConflictErrorr
Nc r r )r r r �APT_UPDATE_PROCESS_CONFLICTr
r r+ r" r r r 5 r$ z&APTUpdateProcessConflictError.__init__�r
Nr, r r r" r r4 4 � r4 c r. )�APTUpdateInvalidRepoError�repo_msgr
Nc �& t jj|d�}t� j|j|jd� d S )N)r9 r )r �APT_UPDATE_INVALID_REPO�formatr r r
r )r r9 r
r" r r r = � z"APTUpdateInvalidRepoError.__init__r% r r r" r r8 < r2 r8 c s, e Zd Zddee ddf� fdd�
Z� ZS )� APTInstallProcessConflictErrorN�
header_msgr
c s2 |r|d7 }t jj|d�}t� j|j|jd� d S )N�.
)r? r )r � APT_INSTALL_PROCESS_CONFLICTr< r r r
r )r r? r
r" r r r C s �z'APTInstallProcessConflictError.__init__r )r r r r r r r& r r r" r r> B s $r> c s2 e Zd Z ddedee ddf� fdd�
Z� ZS )�APTInstallInvalidRepoErrorNr9 r? r
c s4 |r|d7 }t jj||d�}t� j|j|jd� d S )Nr@ )r? r9 r )r �APT_INSTALL_INVALID_REPOr< r r r
r )r r9 r? r
r" r r r O s �z#APTInstallInvalidRepoError.__init__r )r r r r r r r& r r r" r rB N s ����rB c r )� SnapdNotProperlyInstalledError�snap_cmd�servicer
Nc �( t jj||d�}t� j|j|jd� d S )N)rE rF r )r � SNAPD_NOT_PROPERLY_INSTALLEDr< r r r
r )r rE rF r
r" r r r \ s �z'SnapdNotProperlyInstalledError.__init__r% r r r" r rD [ r' rD c r3 )�CannotInstallSnapdErrorr
Nc � t j}t� j|j|jd� d S r )r �CANNOT_INSTALL_SNAPDr r r
r �r r
r" r r r e � z CannotInstallSnapdError.__init__r6 r, r r r" r rI d r7 rI c r. )�ErrorInstallingLivepatchr0 r
Nc r: )N)r0 r )r �ERROR_INSTALLING_LIVEPATCHr< r r r
r �r r0 r
r" r r r k r= z!ErrorInstallingLivepatch.__init__r% r r r" r rN j r2 rN c s. e Zd Zdedededdf� fdd� Z� ZS )� InvalidServiceToDisableError� operation�invalid_servicer r
Nc s* t jj|||d�}t� j|j|jd� d S )N)rR rS r r )r �INVALID_SERVICE_OP_FAILUREr< r r r
r )r rR rS r r
r" r r r q s �z%InvalidServiceToDisableError.__init__r% r r r" r rQ p s ����rQ c �" e Zd Zdef� fdd� Z� ZS )�ProxyNotWorkingError�proxyc �$ t � jtjj|d�jtjjd� d S �N)rW r )r r r � NOT_SETTING_PROXY_NOT_WORKINGr< r
r �r rW r" r r r } �
�z ProxyNotWorkingError.__init__r% r r r" r rV | � rV c rU )�ProxyInvalidUrlrW c rX rY )r r r � NOT_SETTING_PROXY_INVALID_URLr< r
r r[ r" r r r � r\ zProxyInvalidUrl.__init__r% r r r" r r^ � r] r^ 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 r` � s r` c s* e Zd ZdZdZdef� fdd� Z� ZS )�AlreadyAttachedErrorzDAn exception to be raised when a command needs an unattached system.� � account_namec r: )N)rc r )r �ALREADY_ATTACHEDr< r r r
r )r rc r
r" r r r � r= z AlreadyAttachedError.__init__)r r r r r r r r& r r r" r ra � s ra c s$ e Zd ZdZdZ� fdd�Z� ZS )� AttachErrorz@An exception to be raised when we detect a generic attach error.r c rJ r )r �ATTACH_FAILUREr r r
r rL r" r r r � rM zAttachError.__init__)r r r r r r r& r r r" r re � s re c r )� AttachInvalidConfigFileError� config_namer r
Nc s2 t jj||d�}t� jtj|jtd�|j d� d S )N)rh r )�widthr )
r �ATTACH_CONFIG_READ_ERRORr< r r �textwrap�fillr
r r )r rh r r
r" r r r � s �
�z%AttachInvalidConfigFileError.__init__r% r r r" r rg � r' rg c r( )�AttachInvalidTokenErrorc r r )r r r �ATTACH_INVALID_TOKENr
r r+ r" r r r � r$ z AttachInvalidTokenError.__init__r, r r r" r rm � r- rm c r( )�ConnectivityErrorc r r )r r r �CONNECTIVITY_ERRORr
r r+ r" r r r � r$ zConnectivityError.__init__r, r r r" r ro � r- ro c r( )� MagicAttachTokenAlreadyActivatedc rJ r )r �$MAGIC_ATTACH_TOKEN_ALREADY_ACTIVATEDr r r
r rL r" r r r � �
�z)MagicAttachTokenAlreadyActivated.__init__r, r r r" r rq � r- rq c r( )�MagicAttachTokenErrorc rJ r )r �MAGIC_ATTACH_TOKEN_ERRORr r r
r rL r" r r r � rs z MagicAttachTokenError.__init__r, r r r" r rt � r- rt c rU )�MagicAttachInvalidEmail�emailc r: )N)rw r )r �MAGIC_ATTACH_INVALID_EMAILr< r r r
r )r rw r
r" r r r � s
�z MagicAttachInvalidEmail.__init__r% r r r" r rv � r] rv c r( )�MagicAttachUnavailablec rJ r )r �MAGIC_ATTACH_UNAVAILABLEr r r
r rL r" r r r � rs z MagicAttachUnavailable.__init__r, r r r" r ry � r- ry c r( )�MagicAttachInvalidParamc rG )N)�param�valuer )r �MAGIC_ATTACH_INVALID_PARAMr< r r r
r )r r| r} r
r" r r r � � �z MagicAttachInvalidParam.__init__r, r r r" r r{ � r- r{ c s. e Zd ZdZdededef� 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 )N)r� r� r� r ) r� r� r �LOCK_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 r� � s "r� c s e Zd ZdZ� fdd�Z� ZS )�MissingAptURLDirectivez@An exception for when the contract server doesn't include aptURLc r: )N)�entitlement_namer )r �MISSING_APT_URL_DIRECTIVEr< r r r
r �r r� 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 ZdZd� fdd� Z� ZS )�NonRootUserErrorz7An exception to be raised when a user needs to be root.r
Nc r r )r r r � NONROOT_USERr
r r+ r" r r r s
�zNonRootUserError.__init__r6 r� r r r" r r� s r� c s2 e Zd ZdZejfdejddf� fdd�
Z� ZS )�UnattachedErrorz>An exception to be raised when a machine needs to be attached.r
r
Nc s t � j|j|jd� d S r )r r r
r rL r" r r r s zUnattachedError.__init__) r r r r r �
UNATTACHED� NamedMessager r& r r r" r r� s ���r� c s. e Zd ZdZdededdf� fdd� Z� ZS )�SecurityAPIMetadataErrorzDAn exception raised with Security API metadata returns invalid data.r
�issue_idr
Nc s& t � �d| d tjj|d� � d S )NzError: �
)�issue)r r r �SECURITY_ISSUE_NOT_RESOLVEDr< )r r
r� r" r r r s �� ��z!SecurityAPIMetadataError.__init__)r r r r r r r& r r r" r r� s "r� c rU )�InvalidProImager0 c s, || _ tjj|d�}t� j|j|jd� d S )Nr1 r )�contract_server_msgr �INVALID_PRO_IMAGEr< r r r
r rP r" r r r ( s zInvalidProImage.__init__r% r r r" r r� ' r] r� c s8 e Zd ZdZd dedee f� fdd�
Zdd� Z� ZS )
�GCPProAccountErrorz?An exception raised when GCP Pro service account is not enabledNr
r c s || _ t� �||� d S r )�coder r )r r
r r� r" r r r 1 s zGCPProAccountError.__init__c C s dj | j| jd�S )Nz(GCPProServiceAccount Error {code}: {msg})r� r
)r< r� r
r+ r r r �__str__5 s �zGCPProAccountError.__str__r ) r r r r r r r r� r& r r r" r r� . s r� c @ s" e Zd Zdee ddfdd�ZdS )�CloudFactoryError�
cloud_typer
Nc C s
|| _ d S r )r� )r r� r r r r < s
zCloudFactoryError.__init__)r r r r r r r r r r r� ; r] r� c @ � e Zd ZdS )�CloudFactoryNoCloudErrorN�r r r r r r r r� @ � r� c @ r� )�!CloudFactoryUnsupportedCloudErrorNr� r r r r r� D r� r� c @ r� )� CloudFactoryNonViableCloudErrorNr� r r r r r� H r� r� c rU )�EntitlementNotFoundErrorr� c r: )N)r r )r �ENTITLEMENT_NOT_FOUNDr< r r r
r r� r" r r r M r= z!EntitlementNotFoundError.__init__r% r r r" r r� L r] r� c
sL e Zd Z ddejdee deeeef dee f� fdd�
Z � Z
S ) �UrlErrorN�causer� �headers�urlc sR t |dd �r t|j�}nt|�}t� �|� || _|| _| jd u r$i | _|| _d S )N�reason)�getattrr r� r r r� r� r� )r r� r� r� r� � cause_errorr" r r r S s
zUrlError.__init__)NNN) r r r r �URLErrorr r� r r r r& r r r" r r� R s �����r� c s> e Zd Z d
dedee dededdf
� fdd �
Z� ZS ) �ProcessExecutionErrorN� �cmdr �stdout�stderrr
c s< || _ || _|| _|sd}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� � message_tmplr" r r r g s ��z ProcessExecutionError.__init__)Nr� r� )r r r r r r� r r& r r r" r r� f s ������r� c �: e Zd Z� fdd�Zdd� Zd
dd�Z� fdd �Z� ZS ) �ContractAPIErrorc s. t � �||j|j|j� |d u ri }|| _d S r )r r r� r� r� � api_error�r �e�error_responser" r r r ~ s
zContractAPIError.__init__c C s0 || j �d�kr
dS | j �dd��|�rdS dS )Nr� 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 )Nr� 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 r� r r� r� r� )r �prefixr" r r r� � s
���� ��zContractAPIError.__str__r �r r r r r� r� r� r& r r r" r r� } s
r� c r� ) �SecurityAPIErrorc s* t � �||j|j|j� |�dd�| _d S )Nr� r� )r r r� r� r� r� r� r� r" r r r � s zSecurityAPIError.__init__c C s t || jv �S r )�boolr� r� r r r r� � s z SecurityAPIError.__contains__Nc C s || j v r| j S |S r )r� )r � error_strr� r r r r� � s
zSecurityAPIError.__get__c sL t � �� }| jg}|r|d t| j� d d�|� S |d t| j� d S )Nr� z] r� r� )r r� r� r r� �join)r r� �detailsr" r r r� � s
zSecurityAPIError.__str__r r� r r r" r r� � s
r� c @ r� )� InPlaceUpgradeNotSupportedErrorNr� r r r r r� � r� r� c @ r� )�IsProLicensePresentErrorNr� r r r r r� � r� r� c @ r� )�CancelProLicensePollingNr� r r r r r� � r� r� c @ r� )�DelayProLicensePollingNr� r r r r r� � r� r� c r )�InvalidFileFormatError� file_name� file_formatr
Nc rG )N)r� r� r )r �INVALID_FILE_FORMATr< r r r
r )r r� r� r
r" r r r � r z InvalidFileFormatError.__init__r% r r r" r r� � r' r� );rk �typingr r r �urllibr �uaclientr �uaclient.defaultsr � Exceptionr r r) r/ r4 r8 r> rB rD rI rN rQ rV r^ r` ra re rg rm ro rq rt rv ry r{ r� r� r� r� r� r� r� r� r� r� r� r� �IOErrorr� r� r� r� r� r� r� r� r� r r r r �<module> sf
"