← Back
Editing: cc_rh_subscription.cpython-38.pyc
U Ӈg�< � @ s� U d Z ddlZddlmZmZ ddlmZ ddlmZ ddl m Z ddl m Z e� e�Zdd d d ge d gd �Ze ed< eeeedd�dd�ZG dd� de�ZG dd� d�Zddd�ZdS )zDRed Hat Subscription: Register Red Hat Enterprise Linux based system� N)�subp�util)�Cloud)�Config)� MetaSchema)� PER_INSTANCEZcc_rh_subscriptionZfedoraZrhelZ openeuler�rh_subscription)�idZdistrosZ frequencyZactivate_by_schema_keys�meta)�name�cfg�cloud�args�returnc C sf t |td�}|�� s$t�d| � d S |�� �sXz�|�� \}}|dk rLt|��|�� }|s`td��|jr�|j d k r�|� � s�td��q�|j �d� n$|jr�|� � s�td��n |j �d� |j d k r�t|j t�s�d }t|��|�|j �} | s�td �|j ���|�� } | �s td ��|�d � W n< tk �rT } z |�t| �� |�d � W 5 d } ~ X Y nX n |�d� d S )N)�logz%s: module not configured.Tz-Registration failed or did not run completelyz Setting of service-level failedz(Completed auto-attach with service levelzSetting auto-attach failedzCompleted auto-attachz"Pools must in the format of a listzUnable to attach pools {0}z Unable to add or remove reposz-rh_subscription plugin completed successfullyz4rh_subscription plugin did not complete successfullyz System is already registered)�SubscriptionManager�LOG� is_configured�debug� is_registered� _verify_keys�SubscriptionError� rhn_register� auto_attach� servicelevel�_set_service_levelr �_set_auto_attach�pools� isinstance�list�addPool�format� update_repos� log_success�log_warn�str) r r r r ZsmZverifyZ verify_msgZcontZ pool_failZ return_stat�e� r' �E/usr/lib/python3/dist-packages/cloudinit/config/cc_rh_subscription.py�handle sN � � r) c @ s e Zd ZdS )r N)�__name__� __module__� __qualname__r' r' r' r( r P s r c @ s� e Zd Zddddddddd d d g Zd'd d�Zdd� Zdd� Zdd� Zdd� Zdd� Z dd� Z dd � Z d d � Z d d � Z d!d"� Zd#d$� Zd%d&� Zd S )(r �org�activation-key�username�password� disable-repo� enable-repo�add-pool� rhsm-baseurl�server-hostname� auto-attach� service-levelNc C s� |d kr t }|| _|| _| j�di �| _| j�d�| _| j�d�| _| j�d�| _| j�d�| _| j�d�| _ | j�d�| _ | j�d�| _ | j�d �| _ | j�d �| _ | j�d �| _| j�d �| _d S ) Nr r4 r5 r3 r. r- r/ r0 r6 r2 r1 r7 )r r r �get�rhel_cfg� rhsm_baseurl�server_hostnamer �activation_keyr- �useridr0 r � enable_repo� disable_repor )�selfr r r' r' r( �__init__c s z SubscriptionManager.__init__c C s | j �|� dS )z>Simple wrapper for logging info messages. Useful for unittestsN)r �info�r@ �msgr' r' r( r# u s z SubscriptionManager.log_successc C s | j �|� dS )zASimple wrapper for logging warning messages. Useful for unittestsN)r ZwarningrC r' r' r( r$ y s z SubscriptionManager.log_warnc C s� | j D ].}|| jkrd�|d�| j��}d|f S q| jdk rdt�| j�sdt�| j�sdd}d|fS | jdk r�| jr�t�t | j��r�d}d|fS dS )zq Checks that the keys in the rh_subscription dict from the user-data are what we expect. z?{0} is not a valid key for rh_subscription. Valid keys are: {1}�, FNz8The key auto-attach must be a boolean value (True/False zqThe service-level key must be used in conjunction with the auto-attach key. Please re-run with auto-attach: True)TN) r9 � valid_rh_keysr! �joinr r Zis_trueZis_falser r% )r@ �kZbad_keyZnot_boolZno_autor' r' r( r } s2 �� � �� ���z SubscriptionManager._verify_keysc C s0 dg}z t |� W n tjk r* Y dS X dS )zg Checks if the system is already registered and returns True if so, else False ZidentityFT)� _sub_man_clir �ProcessExecutionError)r@ �cmdr' r' r( r � s z!SubscriptionManager.is_registeredc C s� | j dk r�| jdk r�dd�| j �d�| j�g}| jdk rJ|�d�| j�� | jdk rf|�d�| j�� zt|dd�d }W nF tjk r� } z&|j d kr�| � d �|j �� W Y �d S d}~X Y nX n�| j dk �r�| j dk �r�dd �| j �d�| j �g}| jdk �r|�d�| j�� | jdk �r2|�d�| j�� zt|dd�d }W nJ tjk �r� } z(|j d k�rz| � d �|j �� W Y �d S d}~X Y nX n| � d� | � d� d S |�d�d �� }| j�d|� dS )z� Registers the system by userid and password or activation key and org. Returns True when successful False when not. N�registerz--activationkey={0}z --org={0}z --baseurl={0}z--serverurl={0}T)� logstring_valr � z Registration failed due to: {0}Fz--username={0}z--password={0}z8Unable to register system due to incomplete information.z9Use either activationkey and org *or* userid and passwordzID: � z"Registered successfully with ID %s)r< r- r! r: �appendr; rI r rJ �stdoutr$ �stderrr= r0 �split�rstripr r )r@ rK � return_outr& Zreg_idr' r' r( r � sX � � � ���z SubscriptionManager.rhn_registerc C s� ddd� | j�g}zt|�d }W nt tjk r� } zT|j�� dkrj|j�d�D ]}|dkrP| �|� qPn| �d� |j � � �� W Y �dS d }~X Y nX |�d�D ]}|dkr�| j � |� q�d S ) N�attach�--autoz--servicelevel={0}r rN � z*Setting the service level failed with: {0}FT) r! r rI r rJ rQ rT rS r$ rR �stripr r �r@ rK rU r& �liner' r' r( r � s* ���z&SubscriptionManager._set_service_levelc C s| ddg}zt |�d }W n: tjk rR } z| �d�|�� W Y �dS d }~X Y nX |�d�D ]}|dkr^| j�|� q^dS ) NrV rW r z Auto-attach failed with: {0}FrX rN T)rI r rJ r$ r! rS r r rZ r' r' r( r s z$SubscriptionManager._set_auto_attachc C sX g }g }dddg}t |�d }|�� �d�}dddg}t |�d }|�� �d�}||fS )zh Gets the list pools for the active subscription and returns them in list form. r z --availablez --pool-onlyr rX z --consumed)rI rT rS )r@ Z available�consumedrK Zresultsr' r' r( � _getPools s z SubscriptionManager._getPoolsc C s� ddg}t |�d }g }|�d�D ]$}d|kr"|�|�d�d �� � q"ddg}t |�d }g }|�d�D ]$}d|krj|�|�d�d �� � qj||fS ) zl Obtains the current list of active yum repositories and returns them in list form. �reposz--list-enabledr rX zRepo ID:�:rO z--list-disabled)rI rS rP rY )r@ rK rU � active_reposZrepo�inactive_reposr' r' r( � _getRepos0 s z SubscriptionManager._getReposc C s� t |�dkr | j�d� dS | �� \}}g }dg}|D ]6}||kr\||kr\|�d�|�� q6| �d�|�� q6t |�dkr�|�|� z*t|� | j�dd� |�� d d �� W dS t j k r� } z | �d �||�� W Y �d S d }~X Y nX d S )zl Takes a list of subscription pools and "attaches" them to the current subscription r zNo pools to attachTrV z --pool={0}zPool {0} is not availablez/Attached the following pools to your system: %srE z--pool=rN z$Unable to attach pool {0} due to {1}FN) �lenr r r] rP r! r$ �extendrI rG �replacer rJ )r@ r Zpool_availableZ pool_consumedZ pool_listrK Zpoolr& r' r' r( r F s0 � �zSubscriptionManager.addPoolc C s8 | j }| j}|dkrg }|dkr$g }t|t�s<| �d� dS t|t�sT| �d� dS t|�dkr|t|�dkr|| j�d� dS | �� \}}g }g }|D ](}||kr�|� d� |�� q�|� |� q�g }g } |D ](}||kr�|� d� |�� q�| � |� q�t|�dk�r8|D ]0} | |k�r$| j�d | � n| �d � | �� �qt| �dk�r`| D ]} | j�d | � �qJd g} t|�dk�r~| � |� t|�dk�r�| � |� z t | � W n< t jk �r� } z| �d � | �� W Y �dS d} ~ X Y nX t|�dk�r | j�dd�|��dd�� t|�dk�r4| j�dd�|��dd�� dS )z� Takes a list of yum repo ids that need to be disabled or enabled; then it verifies if they are already enabled or disabled and finally executes the action to disable or enable Nz&Repo IDs must in the format of a list.Fr z No repo IDs to enable or disableTz --enable={0}z --disable={0}zRepo %s is already enabledz!Repo {0} does not appear to existz.Repo %s not disabled because it is not enabledr^ z Unable to alter repos due to {0}z Enabled the following repos: %srE z --enable=rN z Disabled the following repos: %sz --disable=)r> r? r r r$ rc r r rb rP r! rd rI r rJ rG re ) r@ ZereposZdreposr` ra Z enable_listZenable_list_failZrepoidZ disable_listZdisable_list_failZfailrK r& r' r' r( r" h s| � � ��z SubscriptionManager.update_reposc C s t | jr| jp| j�S )N)�boolr= r0 r< )r@ r' r' r( r � s z!SubscriptionManager.is_configured)N)r* r+ r, rF rA r# r$ r r r r r r] rb r r" r r' r' r' r( r T s2 � #H "Vr Fc C s t j dg| |d�S )z� Uses the preferred cloud-init subprocess def of subp.subp and runs subscription-manager. Breaking this to a separate function for later use in mocking and unittests zsubscription-manager)Z logstring)r )rK rM r' r' r( rI � s rI )F)�__doc__ZloggingZ cloudinitr r Zcloudinit.cloudr Zcloudinit.configr Zcloudinit.config.schemar Zcloudinit.settingsr Z getLoggerr* r r �__annotations__r% r r) � Exceptionr r rI r' r' r' r( �<module> s$ �6 p
Save File
Cancel