x[h< UdZddlZddlmZmZddlmZddlmZddl m Z ddl m Z ejeZdgd e d gd Ze ed <d ededededdf dZGddeZGddZddZy)zDRed Hat Subscription: Register Red Hat Enterprise Linux based systemN)subputil)Cloud)Config) MetaSchema) PER_INSTANCEcc_rh_subscription)fedorarhel openeulerrh_subscription)iddistros frequencyactivate_by_schema_keysmetanamecfgcloudargsreturnc"t|t}|jstjd|y|j st |j \}}|dur t ||j}|s t d|jrC|j7|js t d|jjdnB|jr6|js t d|jjd|jht|jts d }t ||j!|j} | s$t d j#|j|j%} | s t d |j'd y|j'dy#t $r5} |j)t+| |j)d Yd} ~ yd} ~ wwxYw)N)logz%s: module not configured.Tz-Registration failed or did not run completelyzSetting 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}zUnable to add or remove reposz-rh_subscription plugin completed successfullyz4rh_subscription plugin did not complete successfullyzSystem is already registered)SubscriptionManagerLOG is_configureddebug is_registered _verify_keysSubscriptionError rhn_register auto_attach servicelevel_set_service_levelr_set_auto_attachpools isinstancelistaddPoolformat update_repos log_successlog_warnstr) rrrrsmverify verify_msgcont pool_fail return_states E/usr/lib/python3/dist-packages/cloudinit/config/cc_rh_subscription.pyhandler7s Sc *B    .5   * P!#!2 FJT!' 33??$D'C~~"//"=,,.+,MNNFFLL!KL**,+,HIIFFLL!89xx#!"((D1 DI+I66 jj2 "+4;;BHHE//+K'(GHH NNJ K 56 ! P KKA  KKN O O Ps E2G H+H  Hc eZdZy)r N)__name__ __module__ __qualname__r6r r Psr=r cdeZdZgdZddZdZdZdZdZdZ d Z d Z d Z d Z d ZdZdZy)r) orgactivation-keyusernamepassword disable-repo enable-repoadd-pool rhsm-baseurlserver-hostname auto-attach service-levelNc2|t}||_||_|jjdi|_|jjd|_|jjd|_|jjd|_|jjd|_|jjd|_ |jjd|_ |jjd|_ |jjd |_ |jjd |_ |jjd |_|jjd |_y) Nr rFrGrEr@r?rArBrHrDrCrI)rrrgetrhel_cfg rhsm_baseurlserver_hostnamer&activation_keyr?useridrBr" enable_repo disable_repor#)selfrrs r6__init__zSubscriptionManager.__init__cs, ;C %6;  MM--n=#}}001BC]]&&z2 "mm//0@A==$$U+mm'' 3  ))*5 ==,,];==,,]; MM--n= MM--o>r=c:|jj|y)z>Simple wrapper for logging info messages. Useful for unittestsN)rinforSmsgs r6r,zSubscriptionManager.log_successus  cr=c:|jj|y)zASimple wrapper for logging warning messages. Useful for unittestsN)rwarningrWs r6r-zSubscriptionManager.log_warnys r=c|jD]A}||jvsdj|dj|j}d|fcS|jDt j |js%t j|jsd}d|fS|j:|jr(t jt|jrd}d|fSy)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}, Fz8The 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"ris_trueis_falser#r.)rSkbad_keynot_boolno_autos r6rz SubscriptionManager._verify_keys}s  &A*** &DIId.@.@$AB g~% &    ( LL)) *dmmD !r=cTdg} t|y#tj$rYywxYw)zg Checks if the system is already registered and returns True if so, else False identityFT) _sub_man_clirProcessExecutionError)rScmds r6rz!SubscriptionManager.is_registereds7 l   ))  s ''c^|j|jddj|jdj|jg}|j*|j dj|j|j *|j dj|j  t |dd }n|j|jdd j|jdj|jg}|j*|j dj|j|j *|j dj|j  t |dd }n#|jd|jdy |jddj}|j j#d|y#tj$rC}|jd k(r*|jd j|jYd}~y d}~wwxYw#tj$rC}|jd k(r*|jd j|jYd}~y d}~wwxYw)z Registers the system by userid and password or activation key and org. Returns True when successful False when not. Nregisterz--activationkey={0}z --org={0}z --baseurl={0}z--serverurl={0}T) logstring_valrzRegistration 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)rOr?r*rMappendrNrgrrhstdoutr-stderrrPrBsplitrstriprr)rSri return_outr5reg_ids r6r!z SubscriptionManager.rhn_registersM    +$((2F%,,T-@-@A""488,C  , ?11$2C2CDE##/ ,33D4H4HIJ )#TB1E kk%DMM,E '' 4 '' 6C  , ?11$2C2CDE##/ ,33D4H4HIJ )#TB1E  MMJ  MMK !!&)!,335 ;VD[-- 88r>MM9@@J  6-- 88r>MM9@@J  s0=G= I=I9IIJ,)9J''J,c&dddj|jg} t|d}|jdD]#}|dk7s |jj|%y #tj$r}|j j dk7r8|j jdD]}|dk7s |j|n8|jdj|jjYd}~yd}~wwxYw) Nattach--autoz--servicelevel={0}rrm z*Setting the service level failed with: {0}FT) r*r#rgrrhrprsrrr-rqstriprrrSrirtr5lines r6r$z&SubscriptionManager._set_service_levels   ' '(9(9 :  %c*1-J$$T* %Drzt$ %)) xx B&HHNN40,Drz d+, @GG(  sA&&D9AD :A D  Dcddg} t|d}|j dD]#}|dk7s |j j|%y#tj$r*}|jdj |Yd}~yd}~wwxYw) NrwrxrzAuto-attach failed with: {0}FryrmT)rgrrhr-r*rrrrr{s r6r%z$SubscriptionManager._set_auto_attachs" %c*1-J$$T* %Drzt$ % ))  MM8??B C sA B  BB cg}g}gd}t|d}|jjd}gd}t|d}|jjd}||fS)zh Gets the list pools for the active subscription and returns them in list form. )r(z --available --pool-onlyrry)r(z --consumedr)rgrsrr)rS availableconsumedriresultss r6 _getPoolszSubscriptionManager._getPoolsss  5s#A&^^%,,T2 4s#A&NN$++D1(""r=cddg}t|d}g}|jdD]8}d|vs|j|jddj:ddg}t|d}g}|jdD]8}d|vs|j|jddj:||fS) zl Obtains the current list of active yum repositories and returns them in list form. reposz--list-enabledrryzRepo ID::rnz--list-disabled)rgrrrorz)rSrirt active_reposrepoinactive_reposs r6 _getReposzSubscriptionManager._getRepos0s ()!#&q)  $$T* BDT!##TZZ_Q%7$>$>$@A B)*!#&q) $$T* DDT!%%tzz#q'9&@&@&BC D^++r=c`|s|jjdy|j\}}g}dg}|D]K}||vr%||vr!|jdj |,|j dj |Mt |dkDrY|j| t||jjddj|jd d yy #tj$r+}|j d j |Yd }~y d }~wwxYw)zl Takes a list of subscription pools and "attaches" them to the current subscription zNo pools to attachTrwz --pool={0}zPool {0} is not availablerz/Attached the following pools to your system: %sr\z--pool=rmz$Unable to attach pool {0} due to {1}NF) rrrror*r-lenextendrgr^replacerrh)rSr&pool_available pool_consumed pool_listripoolr5s r6r)zSubscriptionManager.addPoolFs HHNN/ 0(,(8%  j HDM)0F  !4!4T!:; 9@@FG  H y>A  JJy ! S!EYYy)229bA --  :AA$J  s'AC//D-!D((D-c|j}|j}|g}|g}t|ts|j dyt|ts|j dy|s|s|j j dy|j\}}g}g}|D]8}||vr!|jdj|(|j|:g}g} |D]8}||vr!|jdj|(| j|:t|dkDrH|D]C} | |vr|j j d | $|j d j| Et| dkDr#| D]} |j j d |  d g} t|dkDr| j|t|dkDr| j| t| t|dkDr;|j j ddj|j!ddt|dkDr;|j j ddj|j!ddy#tj$r*} |j d j| Yd} ~ yd} ~ wwxYw)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.Fz No repo IDs to enable or disableTz --enable={0}z --disable={0}rzRepo %s is already enabledz!Repo {0} does not appear to existz.Repo %s not disabled because it is not enabledrz Unable to alter repos due to {0}zEnabled the following repos: %sr\z --enable=rmz Disabled the following repos: %sz --disable=)rQrRr'r(r-rrrror*rrrgrrhr^r) rSereposdreposrr enable_listenable_list_failrepoid disable_listdisable_list_failfailrir5s r6r+z SubscriptionManager.update_reposhs!!"" >F >F&$' MMB C&$' MMB C HHNN= >'+~~'7$ n  0F'"">#8#8#@A ''/  0  1F%##O$:$:6$BC!((0  1  1 $( <'HHNN#?FMM;BB4H    !A %) Dd  i | q JJ| $ { a  JJ{ #    { a  HHNN1;'00bA  | q HHNN2<(11,C ))  MM<CCAF G s# JJ> J99J>cdt|jxr |jxs |jSN)boolrPrBrO)rSs r6rz!SubscriptionManager.is_configureds%T[[2T]]Jt7J7JKKr=r)r9r:r;r]rTr,r-rrr!r$r%rrr)r+rr<r=r6rrTsQ M?$!F FP4 #(,, DTlLr=rc8tjdg|z|S)z Uses the preferred 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)rirls r6rgrgs 99,-3} MMr=)F)__doc__logging cloudinitrrcloudinit.cloudrcloudinit.configrcloudinit.config.schemarcloudinit.settingsr getLoggerr9rr__annotations__r.r(r7 Exceptionr rrgr<r=r6rs K !#.+g! . 12 j3737637%37t3737l  kLkL\ Nr=