File: //usr/lib64/python3.9/site-packages/cloud_what/__pycache__/provider.cpython-39.pyc
a
X�ZhY) �
@ s� d Z ddlmZmZmZ ddlZddlZz ddlmZ ddl m
Z
W n* e yn ddl m
Z ddl mZ
Y n0 ddlmZ dd lmZ dd
lmZ dd lmZ eeegZe�e�ZG d d
� d
ej�Z e d�dd�Z dde j fe e e ee!e"f d�dd�Z#dde j fe e e eeeeedf d�dd�Z$dde j fe e e ee% d�dd�Z&edk�r�ddl'Z'e�� Z(e(�)ej*� e�+e'j,�Z-e-�)ej*� e�.d�Z/e-�0e/� e(�1e-� e&� Z2e3de2� �� ddl m
Z
e
� Z4e4�5� Z6e&e6�Z2e3d e2� �� dS ) zC
This module contains several utils used for VMs running on clouds
� )�Union�Tuple�ListN)�
HostCollector)�CustomFactsCollector)�MiniHostCollector)�MiniCustomFactsCollector)�BaseCloudProvider)�AWSCloudProvider)�AzureCloudProvider)�GCPCloudProviderc @ s( e Zd ZdZe�� Ze�� ZeeB ZdS )�DetectionMethodzN
Enumeration of allowed methods used for detection of cloud providers
N) �__name__�
__module__� __qualname__�__doc__�enum�auto�STRONG� HEURISTIC�ALL� r r �9/usr/lib64/python3.9/site-packages/cloud_what/provider.pyr
3 s r
)�returnc C s( i } | � t� �� � | � t� �� � | S )zx
Try to gather system facts necessary for detection of cloud provider
:return: Dictionary with system facts
)�updater �get_allr ��factsr r r �gather_system_factsB s r g �?)r � threshold�methodsr c s� � du rt � � � fdd�tD �}t�d� g }tj|v r�|D ] }|�� }|du r<|�|� q<t|�dkr�t�dj |d j
d �� |dfS t|�dkr�t�d
� q�t|�dkr�t� d j d � d
d� |D ��d�� n
t�d� tj
|v �r�g }|D ]:}|�� }t�d|j
� d|� �� ||kr�|�||f� q�|jdd� dd� dd� |D �}t|�dk�rdt�d� n"t�dj d � dd� |D ��d�� n
t�d� |dfS )aI
This method tries to detect cloud providers and return list of possible cloud providers
:param facts: Dictionary with system facts
:param threshold: Threshold using for detection of cloud provider
:param methods: The flag of methods used for detecting of cloud providers
:return: List of cloud providers
Nc s g | ] }|� ��qS r r )�.0�clsr r r �
<listcomp>d � z(_get_cloud_providers.<locals>.<listcomp>z Trying to detect cloud providerT� z:Detected one cloud provider using strong signs: {provider}r �Zproviderz-No cloud provider detected using strong signszFMore than one cloud provider detected using strong signs ({providers})z, c S s g | ]
}|j �qS r ��CLOUD_PROVIDER_ID�r! �cloud_providerr r r r# ~ r$ )Z providersz7Skipping detection of cloud provider using strong signszCloud provider z has probability: c S s | d S )Nr r )�xr r r �<lambda>� r$ z&_get_cloud_providers.<locals>.<lambda>)�key�reversec S s g | ] }|d �qS )r% r )r! �itemr r r r# � r$ z+No cloud provider detected using heuristicsz@Following cloud providers detected using heuristics: {providers}c S s g | ]
}|j �qS r r' r) r r r r# � r$ z5Skipping detection of cloud providers using heuristicF)r �CLOUD_PROVIDERS�log�debugr
r Zis_running_on_cloud�append�len�formatr( �error�joinr Zis_likely_running_on_cloud�sort)r r r Zcloud_providers�
cloud_listr* Zcloud_detectedZ probabilityr r r �_get_cloud_providersS sX
�� ��
��
r: c C s| t | ||�\}}t|�dkr,|du r,|d S t|�dkrx|D ]0}|�� }|dur<t�dj|jd�� | S q<t�d� dS )ai
This method tries to detect cloud provider and return corresponding instance of
cloud provider.
:param facts: Dictionary with system facts
:param threshold: Threshold used for heuristic detection of cloud provider
:param methods: The flag of methods used for detecting of cloud providers
:return: Instance of cloud provider or None
r% Tr NzKMetadata gathered from cloud provider detected using heuristics: {provider}r& zHUnable to get metadata from any cloud provider detected using heuristics)r: r4 Z get_metadatar1 �infor5 r( r2 )r r r r9 � strong_signr* �metadatar r r �get_cloud_provider� s
��
r> c C s" t | ||�\}}dd� |D �}|S )a�
This method tries to detect cloud provider using hardware information provided by dmidecode.
When there is strong sign that the VM is running on one of the cloud provider, then return
list containing only one provider. When there is no strong sign of one cloud provider, then
try to detect cloud provider using heuristics methods. In this case this method will return
list of all cloud providers sorted according detected probability
:param facts: dictionary of facts. When no facts are provided, then hardware, virtualization
and custom facts are gathered.
:param threshold: Threshold used for heuristic detection of cloud provider
:param methods: The flag of methods used for detection of cloud providers (possible enumerates
are following in DetectionMethod). When only STRONG is listed, then detection is
performed only using strong signs. When only HEURISTIC is listed, then only heuristics
detections is used. When both methods are listed, then this method tries to use detection
using strong signs first and if no cloud provider is detected, then it falls back to
heuristics detection.
:return: List of string representing detected cloud providers. E.g. ['aws'] or ['aws', 'gcp']
c S s g | ]
}|j �qS r r' r) r r r r# � r$ z)detect_cloud_provider.<locals>.<listcomp>)r: )r r r r9 r< r r r �detect_cloud_provider� s r? �__main__z4%(asctime)s - %(name)s - %(levelname)s - %(message)sz >>> debug <<< detector result: z.>>> debug <<< detector result (minimalistic): )7r �typingr r r r ZloggingZ rhsmlib.facts.host_collectorr Zrhsmlib.facts.customr � ImportErrorZcloud_what.fact_collectorr r Zcloud_what._base_providerr Zcloud_what.providers.awsr
Zcloud_what.providers.azurer Zcloud_what.providers.gcpr r0 Z getLoggerr r1 �Flagr
�dictr r �float�list�boolr: r> �strr? �sys�rootZsetLevel�DEBUGZ
StreamHandler�stdoutZhandlerZ Formatter� formatterZ setFormatterZ
addHandlerZ_detector_result�printZ collectorr Z_factsr r r r �<module> s~ �
�
� R�� (�� !