D7net
Home
Console
Upload
information
Create File
Create Folder
About
Tools
:
/
lib
/
python3.6
/
site-packages
/
cloudinit
/
config
/
__pycache__
/
Filename :
cc_rh_subscription.cpython-36.pyc
back
Copy
3 �7si�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