File: //usr/lib/python3.9/site-packages/tuned/admin/__pycache__/admin.cpython-39.pyc
a
,�g�? � @ s� d dl mZ d dlZd dlmZ d dlmZ ddl m
Z
d dl m Z d dl
mZ d dlmZ d dlZd dlZd dlZd dlZd dlZd dlZG d d
� d
e�ZdS ) � )�print_functionN)�commands)�Locator� )�TunedAdminDBusException)�TunedException)�ProfileRecommenderc @ s� e Zd Zdddejejfdd�Zdd� Zdd� Z d d
� Z
d d � Z d
d� Z dd� Z
dd� Zdd� Zdd� Zdd� Zdd� Zdd � Zd d � Zd d � Zdhd"d#�Zdid$d%�Zd&d'� Zd(d)� Zd*d+� Zd,d-� Zd.d/� Z d0d1� Z d2d3� Z d4d5� Z d6d7� Z d8d9� Z!d:d;� Z"d<d=� Z#d>d?� Z$d@dA� Z%dBdC� Z&dDdE� Z'dFdG� Z(dHdI� Z)dJdK� Z*dLdM� Z+dNdO� Z,dPdQ� Z-djdSdT�Z.dkdUdV�Z/dldWdX�Z0dmdYdZ�Z1d[d\� Z2d]d^� Z3d_d`� Z4dadb� Z5dcdd� Z6dedf� Z7dgS )n�AdminTFc C sz || _ || _|| _|| _t|�| _t|�| _t� � | _
d| _ d| _ d| _
d | _d | _|| _t� | _| j rp| �� nd| _d S )N� TF)�_dbus�_debug�_async�_timeoutr �_cmd�profiles_locator�_profiles_locator� threading�Event�_daemon_action_finished�_daemon_action_profile�_daemon_action_result�_daemon_action_errstr� _controller�
_log_token�
_log_levelr �_profile_recommender�
_init_dbus�
_dbus_working)�selfZ profile_dirsZdbus�debugZasynco�timeoutZ log_level� r! �5/usr/lib/python3.9/site-packages/tuned/admin/admin.py�__init__ s
zAdmin.__init__c
C sj t j�tjtjtj| j�| _z| j� tj
| j � W dS t yd } z| �
|� W Y d }~dS d }~0 0 d S �NTF)�tunedZadminZDBusController�constsZDBUS_BUSZDBUS_INTERFACEZ DBUS_OBJECTr r Zset_signal_handlerZSIGNAL_PROFILE_CHANGED�_signal_profile_changed_cbr �_error�r �er! r! r" r % s
zAdmin._init_dbusc C s t |tjd� d S )N)�file)�print�sys�stderr)r �messager! r! r" r( . s z Admin._errorc C s* | j �� s&|| _|| _|| _| j �� d S �N)r �is_setr r r �set)r � profile_name�resultZerrstrr! r! r" r' 1 s
z Admin._signal_profile_changed_cbc
C s| z t �t| j�tj��d� W nX tyN } z|jtj kW Y d }~S d }~0 t
t fyv } zW Y d }~dS d }~0 0 dS )Nr FT) �os�kill�intr Z read_filer& ZPID_FILE�OSError�errno�EPERM�
ValueError�IOErrorr) r! r! r" �_tuned_is_running9 s zAdmin._tuned_is_runningc
O sP |d u s|dkrdS d }d }d}zt | d| �}W n* ty\ } zd| _W Y d }~n
d }~0 0 zt | d| �}W nN ty� } z6| js�| �t|�d| � W Y d }~dS W Y d }~n
d }~0 0 | j�r6z6| j�| j� | jj|g|�R i |�� | j� � }W n6 t
�y4 } z | �|� d| _W Y d }~n
d }~0 0 | j�sL||i |��}|S )Nr
FZ
_action_dbus_Z_action_z , action '%s' is not implemented) �getattr�AttributeErrorr r( �strr Zset_on_exit_action�_log_capture_finish�
set_action�runr )r � action_name�args�kwargs�actionZ action_dbus�resr* r! r! r" rG C s8 $�
z Admin.actionc C sd t d� |D ]R}|d d urN|d dkrNt | j�d|d dd|d �� q t d|d � q d S )NzAvailable profiles:r r
z- %sr � )r, r Z align_str)r �
profile_names�profiler! r! r" �_print_profilesa s
&zAdmin._print_profilesc
C sf z| j �� }W n8 tyF } z dd� | j �� D �}W Y d }~n
d }~0 0 | �|� | �� | j �d�S )Nc S s g | ] }|d f�qS )r
r! )�.0rK r! r! r" �
<listcomp>n � z4Admin._action_dbus_list_profiles.<locals>.<listcomp>T)r Z profiles2r �profilesrL �_action_dbus_active�exit)r rJ r* r! r! r" �_action_dbus_list_profilesi s *
z Admin._action_dbus_list_profilesc C s | � | j�� � | �� dS �NT)rL r Zget_known_names_summary�_action_active�r r! r! r" �_action_list_profiless s zAdmin._action_list_profilesc C s& | j �� }|dkrd }| j �d� |S )Nr
T)r �active_profilerR �r r3 r! r! r" �_dbus_get_active_profilex s
z Admin._dbus_get_active_profilec C s | j �� \}}|S r0 )r �get_active_profile)r r3 �manualr! r! r" �_get_active_profile s zAdmin._get_active_profilec C s. | j �� \}}|d u r |d u}|r(tjS tjS r0 )r r[ r& ZACTIVE_PROFILE_MANUALZACTIVE_PROFILE_AUTO)r rK r\ r! r! r" �_get_profile_mode� s zAdmin._get_profile_modec C s | j �� }|dkrd }|S )Nr
)r �post_loaded_profilerY r! r! r" � _dbus_get_post_loaded_profile� s
z#Admin._dbus_get_post_loaded_profilec C s | j �� }|S r0 )r Zget_post_loaded_profilerY r! r! r" �_get_post_loaded_profile� s
z Admin._get_post_loaded_profilec C sl |d dkrXt d� t |d � t � t d� t |d � t � t d� t |d � dS t d | � d
S d S ) Nr Tz
Profile name:r zProfile summary:� zProfile description:� z,Unable to get information about profile '%s'F�r, )r rK � profile_infor! r! r" �_print_profile_info� s zAdmin._print_profile_infor
c C sB |dkr| � � }|r*| �|| j�|��}n td� d}| j�|�S �Nr
�No current active profile.F)rZ rf r re r, rR )r rK rH r! r! r" �_action_dbus_profile_info� s z Admin._action_dbus_profile_infoc
C s� |dkr`z"| � � }|d u r(td� W dS W n4 ty^ } z | �t|�� W Y d }~dS d }~0 0 | �|| j�|tj tj
gddg��S rg ) r] r, r r( r@ rf r Zget_profile_attrsr& ZPROFILE_ATTR_SUMMARYZPROFILE_ATTR_DESCRIPTION)r rK r* r! r! r" �_action_profile_info� s
zAdmin._action_profile_infoc C s$ |d u rt d� dS t d| � dS )Nrh FzCurrent active profile: %sTrd rY r! r! r" �_print_profile_name� s
zAdmin._print_profile_namec C s |rt d| � d S )Nz Current post-loaded profile: %srd rY r! r! r" �_print_post_loaded_profile� s z Admin._print_post_loaded_profilec C s4 | � � }| �|�}|r(| �� }| �|� | j�|�S r0 )rZ rk r` rl r rR )r rX rH r_ r! r! r" rQ � s
zAdmin._action_dbus_activec
C s� z2| � � }| �� }|r0|r$|d7 }nd}||7 }W n4 tyf } z | �t|�� W Y d }~dS d }~0 0 |d ur�| �� s�td� td| � |r�td| � dS | �|�}| �|� |S )N� r
FzKIt seems that tuned daemon is not running, preset profile is not activated.zPreset profile: %sz Preset post-loaded profile: %sT) r] ra r r( r@ r= r, rk rl )r r3 r_ r* rH r! r! r" rU � s(
zAdmin._action_activec C s t d| � d S )NzProfile selection mode: rd )r �moder! r! r" �_print_profile_mode� s zAdmin._print_profile_modec C sB | j �� \}}| �|� |dkr6| �|� | j �d�S | j �d�S )Nr
FT)r Z profile_modero r( rR )r rn �errorr! r! r" �_action_dbus_profile_mode� s
z Admin._action_dbus_profile_modec
C sR z| � � }| �|� W dS tyL } z | �t|�� W Y d }~dS d }~0 0 d S r$ )r^ ro r r( r@ )r rn r* r! r! r" �_action_profile_mode� s
zAdmin._action_profile_modec C s: |r(| j �� s6| j �� s6| �d� d}n| �d| � |S )NzCannot enable the tuning.Fz Unable to switch profile: %s)r Z
is_running�startr( �r �ret�msgr! r! r" �_profile_print_status� s
zAdmin._profile_print_statusc C sr t � � | j| j kr.td| j � | j�d�S | j�� rn| j|krn| j sbtd| j
� | j�d�S | j�d�S dS )Nz�Operation timed out after waiting %d seconds(s), you may try to increase timeout by using --timeout command line option or using --async.FzError changing profile: %sT) �time�
_timestampr r, r rR r �isSetr r r rY r! r! r" �_action_dbus_wait_profile s
z Admin._action_dbus_wait_profilec
C s~ | j d u s| j dkrd S z2| j�| j �}d | _ t|dtjd� tj�� W n. tyx } z| �d� W Y d }~n
d }~0 0 d S )Nr
)�endr+ zUError: Failed to stop log capture. Restart the TuneD daemon to prevent a memory leak.) r r Zlog_capture_finishr, r- r. �flushr r( )r Zlog_msgsr* r! r! r" rA s �zAdmin._log_capture_finishc C s� t |�dkr| �� S d�|�}|dkr2| j�d�S | j�� | jsl| jd url| j d d }| j�
| j|�| _ | j� |�\}}| js�|s�| j�| �
||��S t�� | _| j�| j|� | �
||�S )Nr rm r
F� � )�len�_action_dbus_list�joinr rR r �clearr
r r �log_capture_startr Zswitch_profilerw rx ry rB r{ )r rP r3 r ru rv r! r! r" �_action_dbus_profile s"
�
zAdmin._action_dbus_profilec C s8 t d� | j�g d��\}}|dkr,td��t d� d S )Nz Trying to (re)start tuned...)Zservicer% Zrestartr z<TuneD (re)start failed, you need to (re)start TuneD by hand.zTuneD (re)started.)r, r Zexecuter rt r! r! r" �_restart_tuned. s
zAdmin._restart_tunedc
C s� || j �� v r�zH| jr4| �� | �� r4| �|g�W S | j�||� | �� td� W dS t y� } z&| �
t |�� | �
d� W Y d }~dS d }~0 0 n| �
d| � dS d S )Nz[TuneD is not active on the DBus, not checking whether the profile was successfully applied.TzUnable to switch profile.Fz%Requested profile '%s' doesn't exist.) r Zget_known_namesr r� r r� r Zsave_active_profiler, r r( r@ )r r3 r\ r* r! r! r" � _set_profile5 s
zAdmin._set_profilec C s6 t |�dkr| �� S d�|�}|dkr*dS | �|d�S )Nr rm r
FT)r� rW r� r� )r rP r3 r! r! r" �_action_profileH s
zAdmin._action_profilec C s� | j �� }| j�� | jsD| jd urD| jd d }| j �| j|�| _| j � � \}}| js\|sp| j �
| � ||��S t � � | _
| j �| j|� | � ||�S )Nr~ r )r �recommend_profiler r� r
r r r� r Z auto_profilerR rw rx ry rB r{ )r r3 r ru rv r! r! r" �_action_dbus_auto_profileP s
�
z Admin._action_dbus_auto_profilec C s | j �� }| �|d�S )NF)r � recommendr� rY r! r! r" �_action_auto_profile` s
zAdmin._action_auto_profilec C s t | j�� � | j�d�S rT )r, r r� rR rV r! r! r" � _action_dbus_recommend_profiled s z$Admin._action_dbus_recommend_profilec C s t | j�� � dS rT )r, r r� rV r! r! r" �_action_recommend_profileh s z Admin._action_recommend_profilec C sr |r| j �� }n
| j �� }|r(td� n0td� td� td� td� td� td� tdtj � | j �|�S ) NzIVerification succeeded, current system settings match the preset profile.zLVerification failed, current system settings differ from the preset profile.z=You can mostly fix this by restarting the TuneD daemon, e.g.:z systemctl restart tuned�orz service tuned restartzNSometimes (if some plugins like bootloader are used) a reboot may be required.z&See TuneD log file ('%s') for details.)r Z verify_profile_ignore_missingZverify_profiler, r& ZLOG_FILErR )r �ignore_missingru r! r! r" �_action_dbus_verify_profilel s
z!Admin._action_dbus_verify_profilec C s t d� dS �Nz Not supported in no_daemon mode.Frd )r r� r! r! r" �_action_verify_profile} s z Admin._action_verify_profilec C s: d}| j �| j|�| _| j �� }|s.| �d� | j �|�S )NrI z Cannot disable active profile.)r r� r r Zoffr( rR )r r ru r! r! r" �_action_dbus_off� s �
zAdmin._action_dbus_offc C s t d� dS r� rd rV r! r! r" � _action_off� s zAdmin._action_offrP c C s( |dkr| � � S |dkr$| j|d�S dS )aD Print accessible profiles or plugins got from TuneD dbus api
Keyword arguments:
list_choice -- argument from command line deciding what will be listed
verbose -- if True then list plugin's config options and their hints
if possible. Functional only with plugin listing, with profiles
this argument is omitted
rP �plugins��verboseN)rS �_action_dbus_list_plugins�r Z list_choicer� r! r! r" r� � s zAdmin._action_dbus_listc C s( |dkr| � � S |dkr$| j|d�S dS )aa Print accessible profiles or plugins with no daemon mode
Keyword arguments:
list_choice -- argument from command line deciding what will be listed
verbose -- Plugins cannot be listed in this mode, so verbose argument
is here only because argparse module always supplies verbose
option and if verbose was not here it would result in error
rP r� r� N)rW �_action_list_pluginsr� r! r! r" � _action_list� s zAdmin._action_listc C s� | j �� }|�� D ]d}t|� |rt|| �dkr4q| j �|�}|| D ],}td| � |�|d�}|rHtd| � qHq| j �d�S )zvPrint accessible plugins
Keyword arguments:
verbose -- if is set to True then parameters and hints are printed
r z %sNz %sT)r Z get_plugins�keysr, r� Zget_plugin_hints�getrR )r r� r� �plugin�hintsZ parameter�hintr! r! r" r� � s
z Admin._action_dbus_list_pluginsc C s t d� dS r� rd )r r� r! r! r" r� � s zAdmin._action_list_pluginsc C s0 | j �||�\}}|s$| �d| � | j �|�S )Nz Unable to acquire devices: %s)r Zinstance_acquire_devicesr( rR )r �devices�instanceru rv r! r! r" �%_action_dbus_instance_acquire_devices� s z+Admin._action_dbus_instance_acquire_devicesc C s t d� dS r� rd )r r� r� r! r! r" � _action_instance_acquire_devices� s z&Admin._action_instance_acquire_devicesc C sZ | j �|�\}}}|s0| �d| � | j �d�S |D ]\}}td||f � q4| j �d�S )Nz Unable to list instances: %sFz%s (%s)T)r Z
get_instancesr( rR r, )r � plugin_nameru rv �pairsr� r� r! r! r" �_action_dbus_get_instances� s z Admin._action_dbus_get_instancesc C s t d� dS r� rd )r r� r! r! r" �_action_get_instances� s zAdmin._action_get_instancesc C sN | j �|�\}}}|s0| �d| � | j �d�S |D ] }t|� q4| j �d�S )NzUnable to list devices: %sFT)r Zinstance_get_devicesr( rR r, )r r� ru rv r� Zdevicer! r! r" �!_action_dbus_instance_get_devices� s
z'Admin._action_dbus_instance_get_devicesc C s t d� dS r� rd )r r� r! r! r" � _action_instance_get_devices� s z"Admin._action_instance_get_devicesN)r
)r
)rP F)rP F)F)F)8�__name__�
__module__� __qualname__r& Z
ADMIN_TIMEOUT�loggingZERRORr# r r( r' r= rG rL rS rW rZ r] r^ r` ra rf ri rj rk rl rQ rU ro rq rr rw r{ rA r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r! r! r! r" r sh �
r )Z
__future__r Z tuned.adminr% Ztuned.utils.commandsr Ztuned.profilesr r �
exceptionsr Ztuned.exceptionsr Z tuned.constsr& Z tuned.utils.profile_recommenderr r5 r- r9 rx r r� �objectr r! r! r! r" �<module> s