Your IP : 216.73.217.13
U
�Ad0D �
@ s� U d Z ddlmZ ddlmZ ddlmZ ddlmZmZ ddl m
Z
ddl m Z ddl
mZmZ dd lmZ e�e�Zd
Zd d d
eddgeed�ed�ed�gdgd�Zeed< ee�Z ee e
eedd�dd�ZG dd� de�Z G d d � d �Z d!d d �Z dS )"zDRed Hat Subscription: Register Red Hat Enterprise Linux based system� )�Logger)�dedent��log)�subp�util)�Cloud)�Config)�
MetaSchema� get_meta_doc)� PER_INSTANCEa[ Register a Red Hat system either by username and password *or* activation and
org. Following a successful registration, you can:
- auto-attach subscriptions
- set the service level
- add subscriptions based on pool id
- enable/disable yum repositories based on repo id
- alter the rhsm_baseurl and server-hostname in ``/etc/rhsm/rhs.conf``.
Zcc_rh_subscriptionzRed Hat Subscriptionz.Register Red Hat Enterprise Linux based systemZfedoraZrhelz� rh_subscription:
username: joe@foo.bar
## Quote your password if it has symbols to be safe
password: '1234abcd'
zk rh_subscription:
activation-key: foobar
org: 12345
a� rh_subscription:
activation-key: foobar
org: 12345
auto-attach: true
service-level: self-support
add-pool:
- 1a1a1a1a1a1a1a1a1a1a1a1a1a1a1a1a
- 2b2b2b2b2b2b2b2b2b2b2b2b2b2b2b2b
enable-repo:
- repo-id-to-enable
- other-repo-id-to-enable
disable-repo:
- repo-id-to-disable
- other-repo-id-to-disable
# Alter the baseurl in /etc/rhsm/rhsm.conf
rhsm-baseurl: http://url
# Alter the server hostname in /etc/rhsm/rhsm.conf
server-hostname: foo.bar.com
�rh_subscription)�id�name�titleZ descriptionZdistrosZ frequencyZexamplesZactivate_by_schema_keys�metaN)r �cfg�cloudr �args�returnc
C sf t ||d�}|�� s$|�d| � d S |�� �sXz�|�� \}}|dk rLt|��|�� }|s`td��|jr�|jd 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 )Nr z%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�
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 r ZsmZverifyZ
verify_msgZcontZ pool_failZ return_stat�e� r+ �E/usr/lib/python3/dist-packages/cloudinit/config/cc_rh_subscription.py�handleS sN
�
� r- c @ s e Zd ZdS )r N)�__name__�
__module__� __qualname__r+ r+ r+ r, r � 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
r8 r9 r7 r2 r1 r3 r4 r: r6 r5 r; )�LOGr r �get�rhel_cfg� rhsm_baseurl�server_hostnamer! �activation_keyr1 �useridr4 r � enable_repo� disable_repor )�selfr r r+ r+ r, �__init__� s z SubscriptionManager.__init__c C s | j �|� dS )z>Simple wrapper for logging info messages. Useful for unittestsN)r �info�rE �msgr+ r+ r, r' � s z SubscriptionManager.log_successc C s | j �|� dS )zASimple wrapper for logging warning messages. Useful for unittestsN)r ZwarningrH r+ r+ r, r( � 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)
r> �
valid_rh_keysr% �joinr r Zis_trueZis_falser r) )rE �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)rE �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)rA r1 r% r? �appendr@ rN r rO �stdoutr( �stderrrB r4 �split�rstripr r )rE rP �
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 rS �
z*Setting the service level failed with: {0}FT)
r% r rN r rO rV rY rX r( rW �stripr r �rE rP rZ r* �liner+ r+ r, r 1 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 ) Nr[ r\ r z Auto-attach failed with: {0}Fr] rS T)rN r rO r( r% rX r r r_ r+ r+ r, r K 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 r] z
--consumed)rN rY rX )rE Z available�consumedrP Zresultsr+ r+ r, � _getPoolsW 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 r] zRepo ID:�:rT z--list-disabled)rN rX rU r^ )rE rP rZ � active_reposZrepo�inactive_reposr+ r+ r, � _getReposk 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 attachTr[ z
--pool={0}zPool {0} is not availablez/Attached the following pools to your system: %srJ z--pool=rS z$Unable to attach pool {0} due to {1}FN)
�lenr r rb rU r% r( �extendrN rL �replacer rO )rE r! Zpool_availableZ
pool_consumedZ pool_listrP Zpoolr* r+ r+ r, r$ � 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 enabledrc z Unable to alter repos due to {0}z Enabled the following repos: %srJ z --enable=rS z Disabled the following repos: %sz
--disable=)rC rD r" r# r( rh r r rg rU r% ri rN r rO rL rj )
rE ZereposZdreposre rf Z enable_listZenable_list_failZrepoidZ disable_listZdisable_list_failZfailrP r* r+ r+ r, r& � s|
� �
��z SubscriptionManager.update_reposc C s t | jr| jp| j�S )N)�boolrB r4 rA )rE r+ r+ r, r � s z!SubscriptionManager.is_configured)N)r. r/ r0 rK rF r' r( r r r r r rb rg r$ r& r r+ r+ r+ r, r � s2 �
#H "Vr Fc C s t j dg| |d�S )z�
Uses the prefered 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 )rP rR r+ r+ r, rN � s rN )F) �__doc__Zloggingr �textwrapr Z cloudinitr r r Zcloudinit.cloudr Zcloudinit.configr Zcloudinit.config.schemar
r Zcloudinit.settingsr Z getLoggerr. r< ZMODULE_DESCRIPTIONr �__annotations__r) r# r- � Exceptionr r rN r+ r+ r+ r, �<module> sR
����'�1 � 8 p