File: //usr/lib/python3.6/site-packages/cloudinit/config/__pycache__/cc_rh_subscription.cpython-36.pyc
3
�xh�C �
@ s� U d Z ddlZddlmZ ddlmZmZ ddlmZ ddl m
Z
ddl m Z m
Z
ddlmZ eje�Zd Zd
d d ed
ddgeed�ed�ed�gdgd�Ze e
e�Z ee
eedd�dd�ZG dd� de�ZG dd� d�Zd d d �ZdS ) zDRed Hat Subscription: Register Red Hat Enterprise Linux based system� N)�dedent)�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 openeulerz� 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: "ABC"
a� rh_subscription:
activation-key: foobar
org: "ABC"
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�title� descriptionZdistrosZ frequencyZexamplesZactivate_by_schema_keys)r �cfg�cloud�args�returnc C sj t |td�}|j� s$tjd| � d S |j� �s\y�|j� \}}|dk rLt|��|j� }|s`td��|jr�|j d k r�|j
� s�td��q�|j jd� n$|jr�|j � s�td��n |j jd� |j
d k � r�t|j
t�s�d }t|��|j|j
�} | � s�td
j|j
���|j� } | �std ��|jd � W n< tk
�rX }
z |jt|
�� |jd
� W Y d d }
~
X nX n
|jd� 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* �(/usr/lib/python3.6/cc_rh_subscription.py�handleR sJ
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 �orgzactivation-key�username�passwordz disable-repoz enable-repozadd-poolz rhsm-baseurlzserver-hostnamez auto-attachz
service-levelNc C s� |d kr t }|| _|| _| jjdi �| _| jjd�| _| jjd�| _| jjd�| _| jjd�| _| jjd�| _ | jjd�| _
| jjd�| _ | jjd �| _ | jjd
�| _
| jjd �| _| jjd �| _d S )
Nr
z rhsm-baseurlzserver-hostnamezadd-poolzactivation-keyr0 r1 r2 z auto-attachz enable-repoz disable-repoz
service-level)r r r �get�rhel_cfg� rhsm_baseurl�server_hostnamer �activation_keyr0 �useridr2 r � enable_repo� disable_repor )�selfr r r* r* r+ �__init__� s z SubscriptionManager.__init__c C s | j j|� dS )z>Simple wrapper for logging info messages. Useful for unittestsN)r �info)r; �msgr* r* r+ r& � s z SubscriptionManager.log_successc C s | j j|� dS )zASimple wrapper for logging warning messages. Useful for unittestsN)r Zwarning)r; r> r* r* r+ r' � s z SubscriptionManager.log_warnc C s� x4| j D ]*}|| jkrdj|dj| j��}d|fS qW | jdk rftj| j�pVtj| j� rfd}d|fS | jdk r�| j s�tjt | 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}z, 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: TrueT)TN)
r4 �
valid_rh_keysr$ �joinr r Zis_trueZis_falser r( )r; �kZbad_keyZnot_boolZno_autor* r* r+ r � s
z SubscriptionManager._verify_keysc C s. dg}y t |� W n tjk
r( 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�ddj| j �dj| j�g}| jdk rJ|jdj| j�� | jdk rf|jdj| j�� yt|dd�d }W n> tjk
r� } z |j d
kr�| j
d j|j �� d S d}~X nX n�| j dk o�| j
dk �r�dd
j| j �dj| j
�g}| jdk �r
|jdj| j�� | jdk �r(|jdj| j�� yt|dd�d }W nB tjk
�r~ } z"|j d
k�rp| j
d j|j �� d S d}~X nX n| j
d� | j
d� d S |jd�d j� }| jjd|� 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)r7 r0 r$ r5 �appendr6 rB r rC �stdoutr' �stderrr8 r2 �split�rstripr r )r; rD �
return_outr) Zreg_idr* r* r+ r � sL
z SubscriptionManager.rhn_registerc C s� dddj | j�g}yt|�d }W np tjk
r� } zR|jj� dkrnx>|jjd�D ]}|dkrR| j|� qRW n| jdj |j j
� �� dS d }~X nX x&|jd�D ]}|dkr�| j j |� q�W d S )
N�attachz--autoz--servicelevel={0}r rG �
z*Setting the service level failed with: {0}FT)
r$ r rB r rC rJ rM rL r' rK �stripr r )r; rD rN r) �liner* r* r+ r . s$ z&SubscriptionManager._set_service_levelc C sx ddg}yt |�d }W n2 tjk
rJ } z| jdj|�� dS d }~X nX x&|jd�D ]}|dkrX| jj|� qXW dS ) NrO z--autor z Auto-attach failed with: {0}FrP rG T)rB r rC r' r$ rL r r )r; rD rN r) rR r* r* r+ r H s z$SubscriptionManager._set_auto_attachc C sX g }g }dddg}t |�d }|j� jd�}dddg}t |�d }|j� jd�}||fS )zh
Gets the list pools for the active subscription and returns them
in list form.
r" z --availablez --pool-onlyr rP z
--consumed)rB rM rL )r; Z available�consumedrD �resultsr* r* r+ � _getPoolsT s
z SubscriptionManager._getPoolsc C s� ddg}t |�d }g }x2|jd�D ]$}d|kr$|j|jd�d j� � q$W ddg}t |�d }g }x2|jd�D ]$}d|krp|j|jd�d j� � qpW ||fS ) zl
Obtains the current list of active yum repositories and returns
them in list form.
�reposz--list-enabledr rP zRepo ID:�:rH z--list-disabled)rB rL rI rQ )r; rD rN � active_reposZrepo�inactive_reposr* r* r+ � _getReposh s z SubscriptionManager._getReposc C s� t |�dkr | jjd� dS | j� \}}g }dg}x>|D ]6}||kr^||kr^|jdj|�� q8| jdj|�� q8W t |�dkr�|j|� y(t|� | jjddj |�j
d d
�� dS t j k
r� } z| jd j||�� d S d
}~X nX d
S )zl
Takes a list of subscription pools and "attaches" them to the
current subscription
r zNo pools to attachTrO z
--pool={0}zPool {0} is not availablez/Attached the following pools to your system: %sz, z--pool=rG z$Unable to attach pool {0} due to {1}FN)
�lenr r rU rI r$ r' �extendrB r@ �replacer rC )r; r Zpool_availableZ
pool_consumedZ pool_listrD Zpoolr) r* r* r+ r# ~ s,
zSubscriptionManager.addPoolc
C s@ | j }| j}|dkrg }|dkr$g }t|t�s<| jd� dS t|t�sT| jd� dS t|�dkr|t|�dkr|| jjd� dS | j� \}}g }g }x0|D ](}||kr�|j dj
|�� q�|j |� q�W g }g } x0|D ](}||kr�|j dj
|�� q�| j |� q�W t|�dk�rDx8|D ]0}
|
|k�r.| jjd |
� n| jd
j
|
�� �qW t| �dk�rpx | D ]}
| jjd |
� �qXW d g} t|�dk�r�| j |� t|�dk�r�| j |� y t | � W n4 t
jk
�r� } z| jd
j
| �� dS d} ~ X nX t|�dk�r| jjddj|�jdd�� t|�dk�r<| jjddj|�jdd�� 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 enabledrV z Unable to alter repos due to {0}z Enabled the following repos: %sz, z --enable=rG z Disabled the following repos: %sz
--disable=)r9 r: r! r" r' r[ r r rZ rI r$ r\ rB r rC r@ r] )
r; ZereposZdreposrX rY Z enable_listZenable_list_failZrepoidZ disable_listZdisable_list_failZfailrD r) r* r* r+ r% � sr
z SubscriptionManager.update_reposc C s t | jr| jp| j�S )N)�boolr8 r2 r7 )r; r* r* r+ r � s z!SubscriptionManager.is_configured)N)r- r. r/ r? r< r& r' r r r r r rU rZ r# r% r r* r* r* r+ r � s0
#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 )rD rF r* r* r+ rB � s rB )F) �__doc__Zlogging�textwrapr Z cloudinitr r Zcloudinit.cloudr Zcloudinit.configr Zcloudinit.config.schemar r Zcloudinit.settingsr Z getLoggerr- r ZMODULE_DESCRIPTION�metar( r" r, � Exceptionr r rB r* r* r* r+ �<module> s:
6 p