File: //lib/python3/dist-packages/cloudinit/config/__pycache__/cc_rh_subscription.cpython-310.pyc
o
�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 dede de
dededd f d d �ZG d d � d e�Z G d!d"� d"�Z d&d$d%�Z d S )'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``.
�cc_rh_subscriptionzRed Hat Subscriptionz.Register Red Hat Enterprise Linux based system�fedora�rhelz� 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�title� description�distros� frequency�examples�activate_by_schema_keys�metar �cfg�cloudr �args�returnNc
C s` t ||d�}|�� s|�d| � d S |�� s�zs|�� \}}|dur%t|��|�� }|s/td��|jrF|jd urF|� � s?td��|j
�d� n|jrW|� � sQtd��|j
�d� |j d urxt
|j t�shd } t| ��|�|j �}
|
sxtd
�|j ���|�� }
|
s�td ��|�d � W d S ty� } z|�t| �� |�d
� W Y d } ~ d S d } ~ ww |�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 �sm�verify�
verify_msg�cont� pool_fail� return_stat�e� r9 �E/usr/lib/python3/dist-packages/cloudinit/config/cc_rh_subscription.py�handleS sR �
���r; c @ s e Zd ZdS )r# N)�__name__�
__module__� __qualname__r9 r9 r9 r: r# � s r# c @ s~ e Zd Zg d�Zd dd�Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zdd � ZdS ) r ) �org�activation-key�username�password� disable-repo� enable-repo�add-pool� rhsm-baseurl�server-hostname� auto-attach�
service-levelNc C s� |d u rt }|| _|| _| 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 rF rG rE r@ r? rA rB rH rD rC rI )�LOGr r �get�rhel_cfg� rhsm_baseurl�server_hostnamer) �activation_keyr? �useridrB r% � enable_repo� disable_repor& )�selfr r r9 r9 r: �__init__� s z SubscriptionManager.__init__c C � | j �|� dS )z>Simple wrapper for logging info messages. Useful for unittestsN)r �info�rS �msgr9 r9 r: r/ � � z SubscriptionManager.log_successc C rU )zASimple wrapper for logging warning messages. Useful for unittestsN)r �warningrW r9 r9 r: r0 � rY z SubscriptionManager.log_warnc C s� | j D ]}|| jvrd�|d�| j��}d|f S q| jdur2t�| j�s2t�| j�s2d}d|fS | jdurH| jrBt�t | j��rHd}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)
rL �
valid_rh_keysr- �joinr% r �is_true�is_falser& r1 )rS �k�bad_key�not_bool�no_autor9 r9 r: r" � s0
� �
�
��
���z SubscriptionManager._verify_keysc C s, dg}zt |� W dS tjy Y dS w )zg
Checks if the system is already registered and returns
True if so, else False
�identityFT)� _sub_man_clir �ProcessExecutionError)rS �cmdr9 r9 r: r! � s
��z!SubscriptionManager.is_registeredc
C s� | j dur`| jdur`dd�| j �d�| j�g}| jdur%|�d�| j�� | jdur3|�d�| j�� z
t|dd�d }W n� tjy_ } z|j d
krT| �
d �|j �� W Y d}~d S d}~ww | j dur�| j
dur�dd
�| j �d�| j
�g}| jdur�|�d�| j�� | jdur�|�d�| j�� z
t|dd�d }W n. tjy� } z|j d
kr�| �
d �|j �� W Y d}~d S d}~ww | �
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)rO r? r- rM �appendrN re r rf �stdoutr0 �stderrrP rB �split�rstripr r )rS rg �
return_outr8 �reg_idr9 r9 r: r$ � s`
�
���
�
�����z SubscriptionManager.rhn_registerc
C s� ddd� | j�g}zt|�d }W n9 tjyJ } z,|j�� dkr4|j�d�D ] }|dkr2| �|� q'n | �d� |j �
� �� W Y d }~dS d }~ww |�d�D ] }|dkr\| j � |� qPd S )
N�attach�--autoz--servicelevel={0}r rj �
z*Setting the service level failed with: {0}FT)
r- r& re r rf rm rp ro r0 rn �stripr r �rS rg rq r8 �liner9 r9 r: r' 1 s4
�
������ �z&SubscriptionManager._set_service_levelc
C sz ddg}zt |�d }W n tjy( } z| �d�|�� W Y d }~dS d }~ww |�d�D ] }|dkr:| j�|� q.dS ) Nrs rt r z Auto-attach failed with: {0}Fru rj T)re r rf r0 r- ro r r rw r9 r9 r: r( K s �� �z$SubscriptionManager._set_auto_attachc C sT g }g }g d�}t |�d }|�� �d�}g d�}t |�d }|�� �d�}||fS )zh
Gets the list pools for the active subscription and returns them
in list form.
)r+ z --available� --pool-onlyr ru )r+ z
--consumedry )re rp ro )rS � available�consumedrg �resultsr9 r9 r: � _getPoolsW s z SubscriptionManager._getPoolsc C s� ddg}t |�d }g }|�d�D ]}d|v r#|�|�d�d �� � qddg}t |�d }g }|�d�D ]}d|v rG|�|�d�d �� � q5||fS ) zl
Obtains the current list of active yum repositories and returns
them in list form.
�reposz--list-enabledr ru zRepo ID:�:rk z--list-disabled)re ro rl rv )rS rg rq � active_repos�repo�inactive_reposr9 r9 r: � _getReposk s � �z SubscriptionManager._getReposc
C s� t |�dkr| j�d� dS | �� \}}g }dg}|D ]}||vr.||v r.|�d�|�� q| �d�|�� qt |�dkru|�|� zt|� | j�dd� |��
d d
�� W dS t j yt } z| �d �||�� W Y d }~d
S d }~ww d S )zl
Takes a list of subscription pools and "attaches" them to the
current subscription
r zNo pools to attachTrs z
--pool={0}zPool {0} is not availablez/Attached the following pools to your system: %sr[ z--pool=rj z$Unable to attach pool {0} due to {1}NF)
�lenr r r} rl r- r0 �extendre r] �replacer rf )rS r) �pool_available�
pool_consumed� pool_listrg �poolr8 r9 r9 r: r, � s6
�
����zSubscriptionManager.addPoolc
C s$ | j }| j}|du r g }|du rg }t|t�s | �d� dS t|t�s*| �d� dS t|�dkr>t|�dkr>| j�d� dS | �� \}}g }g }|D ]}||v rY|� d�
|�� qJ|� |� qJg }g } |D ]}||v rt|� d�
|�� qe| � |� qet|�dkr�|D ]}
|
|v r�| j�d |
� q�| �d
�
|
�� q�t| �dkr�| D ] }
| j�d |
� q�d g} t|�dkr�| � |� t|�dkr�| � |� zt | � W n t
jy� } z| �d
�
| �� W Y d} ~ dS d} ~ ww t|�dkr�| j�dd�|��dd�� t|�dk�r| 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: %sr[ z --enable=rj z Disabled the following repos: %sz
--disable=)rQ rR r* r+ r0 r� r r r� rl r- r� re r rf r] r� )
rS �erepos�dreposr� r� � enable_list�enable_list_fail�repoid� disable_list�disable_list_fail�failrg r8 r9 r9 r: r. � s~
� �
�� ��z SubscriptionManager.update_reposc C s t | jr| jp | j�S �N)�boolrP rB rO )rS r9 r9 r: r � s z!SubscriptionManager.is_configuredr� )r< r= r> r\ rT r/ r0 r" r! r$ r' r( r} r� r, r. r r9 r9 r9 r: r � s
#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)� logstring)r )rg ri r9 r9 r: re � s re )F) �__doc__�loggingr �textwrapr � cloudinitr r r �cloudinit.cloudr �cloudinit.configr �cloudinit.config.schemar
r �cloudinit.settingsr � getLoggerr< rJ �MODULE_DESCRIPTIONr �__annotations__r1 r+ r; � Exceptionr# r re r9 r9 r9 r: �<module> s`
����'�1�����
�8 p