File: //usr/lib64/python3.9/site-packages/rhsm/__pycache__/profile.cpython-39.pyc
a
X�Zh:? � @ s� U d dl Z d dlZd dlZd dlZd dlZd dlmZm Z d dl
m Z d dl
mZ d dlmZmZ d dlmZ z d dlZW n ey� dZY n0 z d dlZW n ey� dZY n0 z d dlZW n ey� dZY n0 ej�d�duZee d< er�d Z nd
Z e� e �Z G d d � d e!�Z"G d
d� d�Z#G dd� d�Z$G dd� d�Z%G dd� d�Z&G dd� d�Z'e(e e'e$e#f d�dd�Z)e'e%e#d�Z*e+e d< dS ) � N)�List�Union)�ourjson)�suppress_output)�SafeConfigParser� ConfigParser)�providerZ zypp_plugin�
use_zypperz$/etc/rhsm/zypper.repos.d/redhat.repoz /etc/yum.repos.d/redhat.repoc @ s e Zd ZdZdS )�InvalidProfileTypezI
Thrown when attempting to get a profile of an unsupported type.
N)�__name__�
__module__� __qualname__�__doc__� r r �2/usr/lib64/python3.9/site-packages/rhsm/profile.pyr
2 s r
c @ s� e Zd Zdd�dd�Zed�dd�Zd ed�dd �Zee e d
�d d ��Z
ed
dd�dd��Z e e
e d�dd��Ze
e d�dd�ZdS )�ModulesProfileN��returnc C s | � � | _d S �N)�_ModulesProfile__generate�content��selfr r r �__init__; s zModulesProfile.__init__c C s
t | j�S r ��strr r r r r �__str__> s zModulesProfile.__str__��otherr c C s | j |j kS r �r �r r r r r �__eq__A s zModulesProfile.__eq__)� module_listr c C sz i }| D ]d}|d |d |d |d |d f}||vr@|||< q|| � dd�du r|� dd�du r|||< qt|�� �S ) z�
Try to uniquify list of modules, when there are duplicated repositories
:param module_list: list of modules
:return: List of modules without duplicity
�name�stream�version�context�arch�activeFT)�get�list�values)r" �ret�module�keyr r r � _uniquifyD s "
$
zModulesProfile._uniquifyzdnf.Base)�baser c C s t �� }|du s|jdkr(t�d� dS t�d� |�� }|du rPt�d� dS zt�|�}W n tjy~ t� d� Y dS 0 d|vr�t�d� dS |d }t�d |� �� |�
| �D ]`}t�d
|j � �� z|� ||� W q� t
�y } z t�d j|d �� W Y d}~q�d}~0 0 q�dS )
z�
Try to fix RHUI repos on AWS systems. When the system is running on AWS, then we have
to fix repository URL. See: https://bugzilla.redhat.com/show_bug.cgi?id=1924126
:param base: DNF base
:return: None
NZawszAThis system is not running on AWS. Skipping fixing of RHUI repos.z=This system is running on AWS. Trying to collect AWS metadatazCUnable to gather metadata from IMDS. Skipping fixing of RHUI repos.zMAWS provided corrupted json metadata document. Skipping fixing of RHUI repos.�regionzGRegion is not specified in AWS metadata. Skipping fixing of RHUI repos.z/Trying to fix URLs of RHUI repos using region: zTrying to fix repository: z Unable to fix RHUI URL: {error})�error)r Zget_cloud_providerZCLOUD_PROVIDER_ID�log�debugZ get_metadata�json�loadsZJSONDecodeError�warningZ
rhui_repos�idZfix_rhui_url_template�
ValueError�format)r0 Zcloud_providerZ metadata_str�metadatar1 Zrepor2 r r r �fix_aws_rhui_reposX s2
z!ModulesProfile.fix_aws_rhui_reposc
C s� g }t d u�r�td u�r�t �� }|jj�|jj� |�� | �|� zt �
d� |� � W n> t j j
y� } z"t �dt|� � g W Y d }~S d }~0 0 z
|j}W n ty� t �d� g Y S 0 |�� }|D ]�}d}d}|�|�� |�� �� r�d}n|�|�� ��rd}|�|�� ��r"d}g } |dk�rBt|�|�� ��} |� |�� |�� |� � |� � |� � d d
� |� � D �| ||d � � q�t!�"|�S ) Nz!Trying to fill dnf sack object...z Unable to create sack object: %sz+DNF does not provide modulemd functionality�defaultF�enabledZdisabledTc S s g | ] }|� � �qS r )�getName)�.0�profiler r r �
<listcomp>� � z-ModulesProfile.__generate.<locals>.<listcomp>) r# r$ r% r& r' Zprofiles�installed_profiles�statusr( )#�dnf�libdnf�Base�conf�
substitutionsZupdate_from_etcZ installrootZread_all_reposr< r3 r4 Z fill_sack�
exceptionsZ RepoErrorr2 r Z_moduleContainer�AttributeErrorr7 ZgetModulePackagesZ isEnabledr? Z getStreamZ
isDisabledZisModuleActiveZgetIdr* ZgetInstalledProfiles�appendZ
getVersionZ
getContextZgetArchZ getProfilesr r/ )
r r" r0 �err�modulesZall_module_listZ
module_pkgrE r( rD r r r �
__generate� sV
��zModulesProfile.__generatec C s | j S )zf
Gather list of modules reported to candlepin server
:return: List of modules
r r r r r �collect� s zModulesProfile.collect)r r r
r r r �boolr! � staticmethodr* r/ r< r r �dictr rQ r r r r r : s '9r c @ s� e Zd Zee d�dd�Zedd�dd�Zed�dd �Zeed
�d d �Z ed�d
d�Z
e ed�dd��Z ed�dd�Z
ed�dd�ZdS )� EnabledReposr c
s� t j�| j�sg S tr t� � nt� � � �| j� � fdd�� �� D �}g }|D ]B}z$|� || �
� � |d��gd�� W qN t y� Y q�Y qN0 qN|S )Nc s g | ]}� � |d �r|�qS )r> )Z
getboolean)r@ �section�Zconfigr r rB � rC z+EnabledRepos.__generate.<locals>.<listcomp>�baseurl)Z repositoryidrX )
�os�path�exists�repofiler r r �readZsectionsrM �_format_baseurlr) � ImportError)r Zenabled_sections�
enabled_reposrV r rW r rP � s$ �� zEnabledRepos.__generateN�� repo_filer c C s< t durt j �� | _ntdur(t�� | _|| _| �� | _dS )zo
Initialize EnabledRepos
:param repo_file: A repo file path used to filter the report.
N) rF rH �db�yumZYumBase�ybr\ �_EnabledRepos__generater �r rb r r r r � s
zEnabledRepos.__init__c C s
t | j�S r r r r r r r � s zEnabledRepos.__str__)�repo_urlr c C s6 t r| �|�S | �� }|�d|d ��d|d �S dS )zu
Returns a well formatted baseurl string
:param repo_url: a repo URL that you want to format
z $releasever�
releaseverz $basearch�basearchN)r �_cut_question_mark�_obtain_mappings�replace)r rh Zmappingsr r r r^ � s
�z EnabledRepos._format_baseurlc C s |d|� d�� S )z�
Returns a string where everything after the first occurrence of '?' is truncated
:param repo_url: a repo URL that you want to modify
N�?)�find)r rh r r r rk � s z EnabledRepos._cut_question_markc C s2 t dur| �� S tdur | �� S t�d� t�dS )z�
returns a hash with "basearch" and "releasever" set. This will try dnf first, and them yum if dnf is
not installed.
NzCUnable to load module for any supported package manager (dnf, yum).)rF �_obtain_mappings_dnfrd �_obtain_mappings_yumr3 r2 r_ r r r r rl s
z EnabledRepos._obtain_mappingsc C s | j jjd | j jjd d�S �Nri rj )ri rj )rc rI rJ r r r r rp s �z!EnabledRepos._obtain_mappings_dnfc C s | j jjd | j jjd d�S rr )re rI Zyumvarr r r r rq s z!EnabledRepos._obtain_mappings_yum)r r r
r rT rf r r r r^ rk r rl rp rq r r r r rU � s
rU c @ sF e Zd ZdZefedd�dd�Zd ed�dd�Ze e
d �d
d �Z dS ) �EnabledReposProfilez8
Collect information about enabled repositories
Nra c C s t |�| _d S r )rU �_enabled_reposrg r r r r s z EnabledReposProfile.__init__r c C s | j j|j jkS r �rt r r r r r r! " s zEnabledReposProfile.__eq__r c C s | j jS )zc
Gather list of enabled repositories
:return: List of enabled repositories
ru r r r r rQ % s zEnabledReposProfile.collect) r r r
r �REPOSITORY_PATHr r rR r! r rT rQ r r r r rs s rs c @ st e Zd ZdZdeeeeeedd�dd�Zed�dd �Zd e d
�d d �Z
ed�d
d�Z e e
eef ed�dd��ZdS )�Packagez7
Represents a package installed on the system.
r N)r# r% �releaser' �epoch�vendorr c C s( || _ || _|| _|| _|| _|| _d S r �r# r% rx r' ry rz )r r# r% rx r' ry rz r r r r 2 s zPackage.__init__r c C sB | � | j�| � | j�| � | j�| � | j�| � | j�| � | j�d�S )z3Returns a dict representation of this package info.r{ )�_normalize_stringr# r% rx r' ry rz r r r r �to_dict< s
�zPackage.to_dictr c C sn t | t|��sdS | j|jkrj| j|jkrj| j|jkrj| j|jkrj| j|jkrj| �| j�| �|j�krjdS dS �zV
Compare one profile to another to determine if anything has changed.
FT) �
isinstance�typer# r% rx r' ry r| rz r r r r r! G s
�
�
�
�
��zPackage.__eq__c C s d| j | j| jf S )Nz<Package: %s %s %s>)r# r% rx r r r r r Z s zPackage.__str__)�valuer c C s t | �tu r| �dd�S | S )Nzutf-8rm )r� �bytes�decode)r� r r r r| ^ s zPackage._normalize_string)r N)r r r
r r �intr rT r} rR r! r rS r r� r| r r r r rw - s � �
rw c @ s^ e Zd Zdejdd�dd�Zeee ee d�dd��Z
ee d�d d
�Z d e d �d d
�Z
dS )�
RPMProfileN)� from_filer c C s� g | _ |rlt�d|j � |�� }t�|�}|D ]8}| j �t|d |d |d |d |d |d d�� q0n0t�d � t �
� }|� d
� |� � }| �
|�| _ d S ) z�
Load the RPM package profile from a given file, or from rpm itself.
NOTE: from_file is a file descriptor, not a file name.
z!Loading RPM profile from file: %sr# r% rx r' ry rz r{ z Loading current RPM profile.���N)�packagesr3 r4 r# r] r5 r6 rM rw �rpmZTransactionSetZ
setVSFlagsZdbMatch�_accumulate_profile)r r� Z json_buffer� pkg_dictsZpkg_dict�tsZ installedr r r r f s*
��
zRPMProfile.__init__)�rpm_header_listr c C sZ g }| D ]L}|d dkrq|� t|d |d |d |d |d pBd|d pLd d
�� q|S ) z�
Accumulates list of installed rpm info
@param rpm_header_list: list of rpm headers
@return: list of package info dicts
r# z
gpg-pubkeyr% rx r' ry r rz Nr{ )rM rw )r� Zpkg_list�hr r r r� � s
��
z RPMProfile._accumulate_profiler c C s" g }| j D ]}|�|�� � q
|S )z�
Returns a list of dicts containing the package info.
See 'packages' member on this object for a list of actual objects.
@return : list of package info dicts
@rtype: list
)r� rM r} )r r� �pkgr r r rQ � s
zRPMProfile.collectr c C sJ t | t|��sdS t| j�t|j�kr*dS | jD ]}||jvr0 dS q0dS r~ )r r� �lenr� )r r r� r r r r! � s
zRPMProfile.__eq__)N)r r r
�_io�
TextIOWrapperr rS r rT rw r� rQ rR r! r r r r r� e s
r� )� profile_typer c C s | t vrtd| ��t | � }|S )z�
Returns an instance of a Profile object
@param profile_type: profile type
Returns an instance of a Profile object
z$Could not find profile for type [%s])� PROFILE_MAPr
)r� rA r r r � get_profile� s
r� )r� r` Zmodulemdr� ),r� Zlogging�importlib.util� importlibr� Zos.pathrY �typingr r Zrhsmr r5 Z
rhsm.utilsr Ziniparser r Z
cloud_whatr rF r_ rG rd �util� find_specr rR �__annotations__rv Z getLoggerr r3 � Exceptionr
r rU rs rw r� r r� r� rT r r r r �<module> sL
X8Z�