[EDIT] FILE: profile_recommender.cpython-36.pyc
3 �<�e3 � @ s� d dl Z d dlZd dlZd dlZd dlZd dlmZmZ yd dlZ dZ W n dZ Y nX d dl j Z d dl Zd dlmZ ejj� ZG dd� d�ZdS )� N)� ConfigParser�ErrorTF)�commandsc @ s0 e Zd Zd dd�Zdd� Zd dd�Zd d � Zd S )�ProfileRecommenderFc C s || _ t� | _d | _d S )N)� _is_hardcodedr � _commands� _chassis_type)�selfZ is_hardcoded� r �)/usr/lib/python3.6/profile_recommender.py�__init__ s zProfileRecommender.__init__c C s t j}| jr|S tj� dk}|s*tjd� | jt j|d�}|d k rF|S i }x�t j D ]|}g }ytj |�}W n@ t k r� } z$|j t j kr�tjd||f � W Y d d }~X nX x"|D ]}tjj||�} | ||< q�W qRW x6t|j� �D ]&}|| } | j| |d�}|d k r�|S q�W |S )Nr ztProfile recommender is running without root privileges. Profiles with virt recommendation condition will be omitted.)�has_rootzerror accessing %s: %s)�constsZDEFAULT_PROFILEr �os�geteuid�logZwarning�process_configZRECOMMEND_CONF_FILEZRECOMMEND_DIRECTORIES�listdir�OSError�errno�ENOENT�error�path�join�sorted�keys) r Zprofiler Zmatching�filesZ directory�contents�e�namer r r r � recommend s6 $ z ProfileRecommender.recommendTc ! C s� d }d}�y�t jj|�s d S tdddd�}t|_t|��}|j||� W d Q R X �x:|j� D �],}d}�x|j |�D �]�} |j || dd�} | dkr�d} | d kr�|s�d}P t j | | j jd g�d t j�s�d}qt| d k� r�t j | | j jtjdd �t j��sjd}qt| d dk�r<t jj| � �s6t j | | j j| �t j� �rjd}qt| dd� dk�r~tj� } | j� t| jt j| ���dk�rjd}qt| dk�r�| j� } t j | | t j��sjd}qt| dkrtd} t�r:t j j t j j dd�}y|j� |j d } W nR t!t"t#fk �r6 } z.t$|d��r&|j%t%j&k�r&t'j(d| � W Y d d }~X nX n|�sRt'j(d| � d}t j | | t j�d krtd}qtW |r\t jd�}|j)d|�}P q\W W n< t!t"t*fk �r� } zt'j(d||f � W Y d d }~X nX |S )NF�=�#)Z delimitersZinline_comment_prefixes�strictT)�raw� z^$Zvirtz virt-what� �system)Zno_errorr �/� Zprocess� chassis_typeZsyspurpose_role)Zraise_on_error�roler z/Failed to load the syspurpose file: %sz\Failed to process 'syspurpose_role' in '%s' , the syspurpose module is not availablez,[^,]*$zerror processing '%s', %s)+r r �isfiler �strZ optionxform�openZ read_fileZsectionsZoptions�get�re�matchr Zexecute�Sr ZSYSTEM_RELEASE_FILE�exists�procfsZpidstatsZreload_threads�lenZ find_by_regex�compile�_get_chassis_type� IGNORECASE�have_syspurpose� syspurposer ZSyspurposeStoreZUSER_SYSPURPOSEr �IOErrorr �KeyError�hasattrr r r r �subr )r Zfnamer Zmatching_profileZsyspurpose_error_logged�config�fZsectionr1 Zoption�valueZpsr* r+ �sr �rr r r r 9 s� $z!ProfileRecommender.process_configc % C s� | j d k r tjd| j � | j S dddddddd d d d d ddddddddddddddd d d d d d!d"d#d$d%d&g%}y0td'd(��}t|j� �}W d Q R X || | _ W nT tk r� tjd)| � Y n4 tt fk r� } ztj d*| � W Y d d }~X nX | j �rtjd+| j � | j S yXt j d,d-d.gt j t j d/d0�}|j� \}}|j�rZtjd1|j|f � n|j� j� | _ W n6 tt fk �r� } ztj d2| � W Y d d }~X nX | j �s�tjd3� d| _ ntjd+| j � | j S )4Nz"returning cached chassis type '%s'r% ZOtherZUnknownZDesktopzLow Profile Desktopz Pizza Boxz Mini TowerZTowerZPortableZLaptopZNotebookz Hand HeldzDocking Stationz All In Onez Sub Notebookz Space-savingz Lunch BoxzMain Server ChassiszExpansion Chassisz Sub ChassiszBus Expansion ChassiszPeripheral Chassisz RAID ChassiszRack Mount ChassiszSealed-case PCz Multi-systemZ CompactPCIZ AdvancedTCAZBladezBlade EnclosingZTabletZ ConvertibleZ Detachablez IoT Gatewayz Embedded PCzMini PCzStick PCz(/sys/devices/virtual/dmi/id/chassis_typerC z/Unknown chassis type id read from dmi sysfs: %dz"error accessing dmi sysfs file: %szchassis type - %sZ dmidecodez-sz chassis-typeT)�stdout�stderrZ close_fdsz,dmidecode finished with error (ret %d): '%s'z#error executing dmidecode tool : %sz!could not determine chassis type.)r r �debugr. �int�read� IndexErrorr r r; �warn� subprocess�Popen�PIPEZ communicate� returncode�strip�decode)r ZDMI_CHASSIS_TYPESZsysfs_chassis_typeZchassis_type_idr Zp_dmiZ dmi_outputZ dmi_errorr r r r7 � sF z$ProfileRecommender._get_chassis_typeN)F)T)�__name__� __module__� __qualname__r r r r7 r r r r r s Ir )r r0 r r4 rK Ztuned.utils.config_parserr r Zsyspurpose.filesr: r9 Z tuned.constsr Z tuned.logsZtunedZtuned.utils.commandsr Zlogsr/ r r r r r r �<module> s
SAVE
CANCEL