HOME

�
oB�]c@s�dZd
dl
mZ
d
dl
mZ
eje�
ZddgZd�Z	de
f
d��YZd	ef
d
��YZ
ed�
ZdS(
s"
RedHat Subscription
-------------------
**Summary:** register red hat enterprise linux based system

Register a RedHat system either by username and password *or* activation and
org. Following a sucessful registration, you can auto-attach subscriptions, set
the service level, add subscriptions based on pool id, enable/disable yum
repositories based on repo id, and alter the rhsm_baseurl and server-hostname
in ``/etc/rhsm/rhs.conf``. For more details, see the ``Register RedHat
Subscription`` example config.

**Internal name:** ``cc_rh_subscription``

**Module frequency:** per instance

**Supported distros:** rhel, fedora

**Config keys**::

    rh_subscription:
        username: <username>
        password: <password>
        activation-key: <activation key>
        org: <org number>
        auto-attach: <true/false>
        service-level: <service level>
        add-pool: <list of pool ids>
        enable-repo: <list of yum repo ids>
        disable-repo: <list of yum repo ids>
        rhsm-baseurl: <url>
        server-hostname: <hostname>
i����(
tlog(
tutiltfedoratrhelcCs�
t|
d
|�

}|j
�s2|jd|�
dS|j�s�
yi
|j�\}}|tk	rnt|�
�
n|j�}|s�td�
�
n|j	r�|j
dk	r�|j�s�td�
�
q
|jjd�

n7|j	r
|j
�s�td�
�
q
|jjd�

n|jdk	rx
t|jt�sB
d}	t|	�
�
n|j|j�
}
|
sx
td	j|j�
�
�
qx
n|j�}
|
s�
td
�
�
n|jd�

Wq�
tk
r�

}
|jt|�
�

|jd�

q�
Xn
|jd
�

dS(NRs%s: module not configured.s-Registration failed or did not run completelysSetting of service-level faileds(Completed auto-attach with service levelsSetting auto-attach failedsCompleted auto-attachs"Pools must in the format of a listsUnable to attach pools {0}sUnable to add or remove reposs-rh_subscription plugin completed successfullys4rh_subscription plugin did not complete successfullysSystem is already registered(tSubscriptionManagert
is_configuredtdebugtNonet
is_registeredt_verify_keystTruetSubscriptionErrortrhn_registertauto_attachtservicelevelt_set_service_levelRt_set_auto_attachtpoolst
isinstancetlisttaddPooltformattupdate_repostlog_successtlog_warntstr(tnametcfgt_cloudRt_argstsmtverifyt
verify_msgtcontt	pool_failtreturn_statt
e((sG/usr/lib/python2.7/site-packages/cloudinit/config/cc_rh_subscription.pythandle1sH













	






	







Rc
BseZ
RS((t__name__t
__module__(((sG/usr/lib/python2.7/site-packages/cloudinit/config/cc_rh_subscription.pyRfs
RcBs�eZ
dd
dddddddd	d
gZdd�
Zd�Zd
�Zd�Zd�Zd�Z	d�Z
d�Zd�Zd�Z
d�Zd�Zd�ZRS(torgsactivation-keytusernametpasswordsdisable-reposenable-reposadd-poolsrhsm-baseurlsserver-hostnamesauto-attachs
service-levelcCs*
|dkrt
}n||_|
|_|jjd
i�|_|jjd�
|_|jjd�
|_|jjd�
|_|jjd�
|_	|jjd�
|_
|jjd�
|_|jjd�
|_|jjd	�
|_
|jjd
�
|_|jjd�
|_|jjd�
|_dS(
Ntrh_subscriptionsrhsm-baseurlsserver-hostnamesadd-poolsactivation-keyR(R)R*sauto-attachsenable-reposdisable-repos
service-level(RtLOGRRtgettrhel_cfgtrhsm_baseurltserver_hostnameRtactivation_keyR(tuseridR*R
tenable_repotdisable_repoR(tselfRR((sG/usr/lib/python2.7/site-packages/cloudinit/config/cc_rh_subscription.pyt__init__ps 

	
	
	











cCs|jj
|
�

d
S(s>Simple wrapper for logging info messages. Useful for unittestsN(Rtinfo(R5tmsg((sG/usr/lib/python2.7/site-packages/cloudinit/config/cc_rh_subscription.pyR�scCs|jj
|
�

d
S(sASimple wrapper for logging warning messages. Useful for unittestsN(Rtwarning(R5R8((sG/usr/lib/python2.7/site-packages/cloudinit/config/cc_rh_subscription.pyR�sc
Cs�xH|jD]=}
|
|j
kr
d
j|
dj|j
�
�}t|fSq
W|jdk	r�tj|j�
p{tj	|j�
r�d}t|fS|j
dk	r�|js�tj	t|j�
�
r�d}t|fStdfS(sq
        Checks that the keys in the rh_subscription dict from the user-data
        are what we expect.
        s?{0} is not a valid key for rh_subscription. Valid keys are: {1}s, s8The key auto-attach must be a boolean value (True/False sqThe service-level key must be used in conjunction with the auto-attach key.  Please re-run with auto-attach: TrueN(
R.t
valid_rh_keysRtjointFalseR
RR
tis_truetis_falseRRR
(R5t
ktbad_keytnot_booltno_auto((sG/usr/lib/python2.7/site-packages/cloudinit/config/cc_rh_subscription.pyR	�s










c
Cs3d
g
}
yt|
�

Wnt
jk
r.


tSXtS(sg
        Checks if the system is already registered and returns
        True if so, else False
        tidentity(t_sub_man_cliR
tProcessExecutionErrorR<R
(R5tcmd((sG/usr/lib/python2.7/site-packages/cloudinit/config/cc_rh_subscription.pyR�s	


c
CsH|jdk	r�|jdk	r�d
dj|j�
dj|j�
g}
|jdk	rp|
jdj|j�
�

n|jdk	r�|
jdj|j�
�

nyt|
dt�

d}Wqt	j
k
r�
}
|jdkr�|jd	j|j
�
�

ntSXn
|jdk	r�
|jdk	r�
d
d
j|j�
dj|j�
g}
|jdk	rm
|
jdj|j�
�

n|jdk	r�
|
jdj|j�
�

nyt|
dt�

d}Wqt	j
k
r�

}
|jdkr�
|jd	j|j
�
�

ntSXn|jd�

|jd
�

tS|jd�
dj�}|jjd|�
tS(s�
        Registers the system by userid and password or activation key
        and org.  Returns True when successful False when not.
        tregisters--activationkey={0}s	--org={0}s
--baseurl={0}s--serverurl={0}t
logstring_valitsRegistration failed due to: {0}s--username={0}s--password={0}s8Unable to register system due to incomplete information.s9Use either activationkey and org *or* userid and passwordsID: i
s"Registered successfully with ID %sN(R1RR(RR/tappendR0RDR
R
REtstdoutRtstderrR<R2R*tsplittrstripRR(R5RFt
return_outR$treg_id((sG/usr/lib/python2.7/site-packages/cloudinit/config/cc_rh_subscription.pyR�sD	





















c
Cs�d
ddj|j
�
g}
yt|
�
d}Wn�tjk
r�
}
|jj�dkr�xX|jjd�
D]"}|dkrl|j|�

qlqlWn|jdj|j	j
��
�

tSXx6|jd�
D]%}|dkr�|jj
|�

q�q�WtS(Ntattachs--autos--servicelevel={0}iRIs

s*Setting the service level failed with: {0}(RRRDR
RERKRNRMRRLtstripR<RRR
(R5RFROR$tline((sG/usr/lib/python2.7/site-packages/cloudinit/config/cc_rh_subscription.pyR�s 













c
Cs�d
dg}
yt|
�
d}Wn-t
jk
rO
}
|jdj|�
�

tSXx6|jd�
D]%}|dkr`|jj|�

q`q`Wt	S(NRQs--autoisAuto-attach failed with: {0}s

RI(
RDR
RERRR<RMRRR
(R5RFROR$RS((sG/usr/lib/python2.7/site-packages/cloudinit/config/cc_rh_subscription.pyR
s









c
Cs~g}
g}d
ddg}t|�
d}|j
�jd�
}
d
ddg}t|�
d}|j
�jd�
}|
|fS(sh
        Gets the list pools for the active subscription and returns them
        in list form.
        Rs--availables--pool-onlyis

s
--consumed(RDRNRM(R5t	availabletconsumedRFtresults((sG/usr/lib/python2.7/site-packages/cloudinit/config/cc_rh_subscription.pyt	_getPools
s




c
Cs�d
dg}
t|
�
d}g}xF|j
d�
D]5}d|kr2|j|j
d�
dj��

q2q2Wd
dg}
t|
�
d}g}xF|j
d�
D]5}d|kr�|j|j
d�
dj��

q�q�W||fS(	sl
        Obtains the current list of active yum repositories and returns
        them in list form.
        treposs--list-enabledis

sRepo ID:t
:i
s--list-disabled(RDRMRJRR(R5RFROtactive_repostrepotinactive_repos((sG/usr/lib/python2.7/site-packages/cloudinit/config/cc_rh_subscription.pyt	_getRepos'
s




'



'
cCs2
t|
�
d
kr&|j
jd�

tS|j�\}}g}dg
}xU|
D]M}||kr�||kr�|jdj|�
�

qN|jdj|�
�

qNWt|�
d
kr.
|j|�

y:t	|�

|j
jddj
|�
jdd	��
tSWq.
tj
k
r*

}
|jd
j||��

tSXndS(sl
        Takes a list of subscription pools and "attaches" them to the
        current subscription
        isNo pools to attachRQs
--pool={0}sPool {0} is not availables/Attached the following pools to your system: %ss, s--pool=RIs$Unable to attach pool {0} due to {1}N(tlenRRR
RWRJRRtextendRDR;treplaceR
RER<(R5Rtpool_availablet
pool_consumedt	pool_listRFtpoolR$((sG/usr/lib/python2.7/site-packages/cloudinit/config/cc_rh_subscription.pyR=
s,



	



















c

Cs|j}
|j
}|
dkr'g}
n|dkr<g}nt|
t�s\|jd
�

tSt|t�s||jd
�

tSt|
�
dkr�t|�
dkr�|jj	d�

t
S|j�\}}g}g}x@|
D]8}||kr
|jdj
|�
�

q�|j|�

q�Wg}g}	x@|D]8}||krS
|jdj
|�
�

q(
|	j|�

q(
Wt|�
dkr�
xI|D]>}
|
|kr�
|jj	d|
�
q}
|jdj
|
�
�

q}
Wnt|	�
dkr�
x$|	D]}
|jj	d|
�
q�
Wnd	g
}t|�
dkr&|j|�

nt|�
dkrH|j|�

nyt|�

Wn-tjk
r�
}
|jd
j
|�
�

tSXt|�
dkr�|jj	ddj|�
jd
d��
nt|�
dkr|jj	ddj|�
jdd��
nt
S(s�
        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
        s&Repo IDs must in the format of a list.is No repo IDs to enable or disables--enable={0}s
--disable={0}sRepo %s is already enableds!Repo {0} does not appear to exists.Repo %s not disabled because it is not enabledRXs Unable to alter repos due to {0}sEnabled the following repos: %ss, s	--enable=RIs Disabled the following repos: %ss
--disable=N(R3R4RRRRR<R^RRR
R]RJRR_RDR
RER;R`(
R5terepostdreposRZR\tenable_listtenable_list_failtrepoidtdisable_listtdisable_list_failtfailRFR$((sG/usr/lib/python2.7/site-packages/cloudinit/config/cc_rh_subscription.pyR]
sn	
	

	

	






$




















	












c

Cst|j
r|jp|j�
S(
N(tboolR2R*R1(
R5((sG/usr/lib/python2.7/site-packages/cloudinit/config/cc_rh_subscription.pyR�
s
N(R&R'R:RR6RRR	RRRRRWR]RRR(((sG/usr/lib/python2.7/site-packages/cloudinit/config/cc_rh_subscription.pyRjs"

	

					>					 	PcCstj
d
g
|d|
�

S(s�
    Uses the prefered cloud-init subprocess def of util.subp
    and runs subscription-manager.  Breaking this to a
    separate function for later use in mocking and unittests
    ssubscription-managert	logstring(R
tsubp(RFRH((sG/usr/lib/python2.7/site-packages/cloudinit/config/cc_rh_subscription.pyRD�
s
N(t__doc__t	cloudinitRtloggingR
t	getLoggerR&R,tdistrosR%t	ExceptionRtobjectRR<RD(((sG/usr/lib/python2.7/site-packages/cloudinit/config/cc_rh_subscription.pyt<module>'s
	5�H