EVOLUTION-NINJA
Edit File: controller.cpython-36.pyc
3 `��g�4 � @ s� d dl mZ d dlZ d dlZ d dlmZ d dlZd dljZd dlm Z dgZ e j j � Z G dd� de�ZG dd� de jjj�ZdS ) � )�exportsN)�TunedException)�commands� Controllerc @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S ) � TimerStorec C s t � | _tj� | _d S )N)�dict�_timers� threadingZLock� _timers_lock)�self� r � /usr/lib/python3.6/controller.py�__init__ s zTimerStore.__init__c C s | j � || j|< W d Q R X d S )N)r r )r �token�timerr r r � store_timer s zTimerStore.store_timerc C sB | j �2 y | j| }|j� | j|= W n Y nX W d Q R X d S )N)r r �cancel)r r r r r r � drop_timer s zTimerStore.drop_timerc C s< | j �, x| jj� D ] }|j� qW | jj� W d Q R X d S )N)r r �valuesr �clear)r r r r r � cancel_all s zTimerStore.cancel_allN)�__name__� __module__� __qualname__r r r r r r r r r s r c s� e Zd ZdZ� fdd�Zdd� Zdd� Zdd � Zej d �d d � �Z d d� Z ej dd�dUdd��Z ej dd�dVdd��Zej dd�dWdd��ZdXdd �Zej dd�dYd d ��Zej dd�dZd d ��Zd!d"� Zej dd#�d[d$d%��Zej dd#�d\d&d'��Zej dd�d]d(d)��Zej dd*�d^d+d,��Zej dd�d_d-d.��Zej dd�d`d/d0��Zej dd�dad1d2��Zej dd3�dbd4d5��Zej dd6�dcd7d8��Z ej dd9�ddd:d;��Z ej dd�ded<d=��Z ej dd�dfd>d?��Z ej dd�dgd@dA��Z ej ddB�dhdCdD��Z!ej dd�didEdF��Z"ej ddG�djdHdI��Z#ej dd�dkdJdK��Z$ej dLd#�dldMdN��Z%ej ddO�dmdPdQ��Z&ej ddR�dndSdT��Z'� Z(S )or z� Controller's purpose is to keep the program running, start/stop the tuning, and export the controller interface (currently only over D-Bus). c s8 t t| �j� || _|| _tj� | _t� | _ t � | _ d S )N) �superr r �_daemon�_global_configr ZEvent� _terminater �_cmdr � _timer_store)r �daemonZ global_config)� __class__r r r + s zController.__init__c C sx t jd� | j� }| jjtjtj�}| r6|r6tj� |rb| j j � x | j j | j d�s`tj � qFW t jd� | j� dS )z1 Controller main loop. The call is blocking. zstarting controller� zterminating controllerN)�log�info�startr �get_bool�consts� CFG_DAEMON�CFG_DEF_DAEMONr r r r �waitZ period_check�stop)r �resr r r r �run3 s zController.runc C s | j j� d S )N)r �set)r r r r � terminateF s zController.terminatec C s0 | j jj� s,| j jj� | j� s,| j jj� d S )N)r Z_sighup_processingZis_setr. �reloadr )r r r r �sighupI s zController.sighupZsbsc C s d S )Nr )r � profile_name�resultZerrstrr r r �profile_changedO s zController.profile_changedc C s t jj|� | jj|� d S )N)�tuned�logs�log_capture_finishr r )r r r r r �_log_capture_abortX s z Controller._log_capture_abortZii�sNc C sf |dkr dS t jj|�}|d kr$dS |dkrVtj|| j|gd�}| jj||� |j� |d krbdS |S )N� r )�args) r5 r6 �log_capture_startr ZTimerr8 r r r% )r Z log_levelZtimeout�callerr r r r r r< \ s z Controller.log_capture_startc C s4 |dkr dS t jj|�}| jj|� |d kr0dS |S )Nr: )r5 r6 r7 r r )r r r= r, r r r r7 j s z Controller.log_capture_finishr: �bc C sD |dkr dS | j jtjtj�r:| jj� r,dS | jj� s:dS | jj� S )Nr: FT) r r& r' r( r) r � is_running� is_enabledr% )r r= r r r r% r s zController.startFc C s, | j j� sd}n| j j|d�}| jj� |S )NT)�profile_switch)r r? r+ r r )r rA r, r r r �_stop} s zController._stopc C s |dkr dS | j dd�S )Nr: F)rA )rB )r r= r r r r+ � s zController.stopc C sp |dkr dS | j j� r*| jdd�}|s*dS y| j j� W n. tk rf } ztjd| � dS d }~X nX | j� S )Nr: FT)rA zFailed to reload TuneD: %s)r r? rB Zreload_profile_configr r# �errorr% )r r= Zstop_ok�er r r r0 � s zController.reloadc C s� | j j� }d}d}d}z�y$|r,| j jdd� | j j||� W nf tjjk r� } zFd}t|�}|r�| j jj |kr�t j d| � d}nt j d| � W Y d d }~X nX W d |r�|r�t j d| � n|s�t j d� | j j� X ||fS ) N�OKTF)rA z@Failed to reapply profile '%s'. Did it change on disk and break?z Failed to apply profile '%s'z>Applying previously applied (possibly out-dated) profile '%s'.z$Applying previously applied profile.)r r? r+ Z set_profiler5 � exceptionsr �str�profile�namer# rC �warnr$ r% )r r2 �manualZ was_running�msg�successZreapplyrD r r r �_switch_profile� s, $ zController._switch_profilez(bs)c C s( |dkr dS | j j|�s dS | j|d�S )Nr: F� Unauthorized�Invalid profile_nameT)FrO )FrP )r � is_valid_namerN )r r2 r= r r r �switch_profile� s zController.switch_profilec C s |dkr dS | j � }| j|d�S )Nr: FrO )FrO )�recommend_profilerN )r r= r2 r r r � auto_profile� s zController.auto_profilec C s* |dkr dS | j jd k r"| j jjS dS d S )Nr: )r rH rI )r r= r r r �active_profile� s zController.active_profilez(ss)c C s� |dkr dS | j j}|d krpy"| jj� \}}|d kr<|d k }W n0 tk rn } zd}t|�}||fS d }~X nX |rztjntj}|dfS )Nr: �unknownrO )rV rO ) r rK r Zget_active_profiler rG r' ZACTIVE_PROFILE_MANUALZACTIVE_PROFILE_AUTO)r r= rK rH rD �moderC r r r � profile_mode� s zController.profile_modec C s |dkr dS | j jpdS )Nr: )r �post_loaded_profile)r r= r r r rY � s z Controller.post_loaded_profilec C sB |dkr dS | j j� r | j j� | j j� r>| j jd dd dd� dS )Nr: FT)Zsave_instantly)r r? r+ r@ Zset_all_profiles)r r= r r r �disable� s zController.disablec C s |dkr dS | j j� S )Nr: F)r r? )r r= r r r r? � s zController.is_running�asc C s |dkr g S | j jjj� S )Nr: )r �profile_loader�profile_locatorZget_known_names)r r= r r r �profiles� s zController.profilesza(ss)c C s |dkr g S | j jjj� S )Nr: )r r\ r] Zget_known_names_summary)r r= r r r � profiles2� s zController.profiles2z(bsss)c C sT |dks| j j|� rdS |d ks*|dkr2| j� }t| jjjj|tj tj gdg��S )Nr: F)Fr: r: r: ) r rQ rU �tupler r\ r] Zget_profile_attrsr' ZPROFILE_ATTR_SUMMARYZPROFILE_ATTR_DESCRIPTION)r r2 r= r r r � profile_info s zController.profile_infoc C s |dkr dS | j jj� S )Nr: )r Zprofile_recommenderZ recommend)r r= r r r rS s z Controller.recommend_profilec C s |dkr dS | j jdd�S )Nr: F)�ignore_missing)r �verify_profile)r r= r r r rc s zController.verify_profilec C s |dkr dS | j jdd�S )Nr: FT)rb )r rc )r r= r r r � verify_profile_ignore_missing s z(Controller.verify_profile_ignore_missingz a{sa{ss}}c C sz |dkr i S i }xd| j j� D ]V}|jjd�d jdd�d }|j� }i ||< x$|j� D ]\}}t|�|| |< qVW q W |S )zuReturn dictionary with accesible plugins Return: dictionary -- {plugin_name: {parameter_name: default_value}} r: �.r" �_���)r �get_all_pluginsr �splitZ_get_config_options�itemsrG )r r= ZpluginsZ plugin_class� plugin_nameZ conf_options�key�valr r r rh s zController.get_all_pluginsc C s* |dks| j j|� rdS | jjt|��S )z"Return docstring of plugin's classr: )r rQ r �get_plugin_documentationrG )r rk r= r r r rn . s z#Controller.get_plugin_documentationza{ss}c C s* |dks| j j|� ri S | jjt|��S )z�Return dictionary with plugin's parameters and their hints Parameters: plugin_name -- name of plugin Return: dictionary -- {parameter_name: hint} r: )r rQ r �get_plugin_hintsrG )r rk r= r r r ro 5 s zController.get_plugin_hintsc C sD |dks| j j|� rdS | jjr@| jjjr@| jjjj|� dS dS )z�Allows to dynamically add sockets to send signals to Parameters: path -- path to socket to register for sending signals Return: bool -- True on success r: FT)r rQ r Z _applicationZ_unix_socket_exporterZregister_signal_path)r �pathr= r r r �register_socket_signal_pathC s z&Controller.register_socket_signal_pathZssc C s� |dkr dS | j j|�s dS | j j|�s,dS d}x2| jjjD ]$}|j|kr<tjd|j � d}P q<W |s�d| }tj|� d|fS t | j j |��}tjd |jt |�f � x�| jjjD ]�}|j |@ } t | �r�|| 8 }tjd t | �|j|jf � |jj|jjk�r6d |j|jjt | �|j|jjf }tj|� d|fS |jj|| � |jj|| � q�W t |��r~d t |� }tj|� d|fS dS )Nr: FrO �Invalid devices�Invalid instance_namezFound instance '%s'.TzInstance '%s' not foundz-Instance '%s' trying to acquire devices '%s'.z8Moving devices '%s' from instance '%s' to instance '%s'.ztTarget instance '%s' is of type '%s', but devices '%s' are currently handled by instance '%s' which is of type '%s'.z2Ignoring devices not handled by any instance '%s'.rE )FrO )Frr )Frs )TrE )r rQ r � _unit_manager� instancesrI r# �debugrC r. Z devstr2devsrG �processed_devices�lenr$ �pluginZ_remove_devices_nocheckZ_add_devices_nocheck) r Zdevices� instance_namer= �foundZinstance_target�retsZdevs�instanceZ devs_movingr r r �instance_acquire_devicesW sJ z#Controller.instance_acquire_devicesz (bsa(ss))c s� |dkrddg fS | j j� �s(ddg fS � dkr\� | j� j� kr\d� }tj|� d|g fS tdd� | jjj �}� dkr�t� fdd�|�}d d t t d d� |��fS ) a Return a list of active instances of a plugin or all active instances Parameters: plugin_name -- name of the plugin or an empty string Return: bool -- True on success string -- error message or "OK" list of string pairs -- [(instance_name, plugin_name)] r: FrO zInvalid plugin_namezPlugin '%s' does not existc S s | j S )N)Zactive)r} r r r �<lambda>� s z*Controller.get_instances.<locals>.<lambda>c s | j j� kS )N)ry rI )r} )rk r r r � s TrE c S s | j | jj fS )N)rI ry )r} r r r r � s ) r rQ rh �keysr# rC �filterr rt ru �list�map)r rk r= r| ru r )rk r � get_instances s zController.get_instancesz(bsas)c C sv |dkrddg fS | j j|�s(ddg fS x0| jjjD ]"}|j|kr4ddtt|j��fS q4W d| }t j |� d|g fS )z�Return a list of devices assigned to an instance Parameters: instance_name -- name of the instance Return: bool -- True on success string -- error message or "OK" list of strings -- device names r: FrO zInvalid instance_nameTrE zInstance '%s' not found) r rQ r rt ru rI �sortedr� rw r# rC )r rz r= r} r| r r r �instance_get_devices� s z Controller.instance_get_devices)N)N)N)F)N)N)N)N)N)N)N)N)N)N)N)N)N)N)N)N)N)N)N)N)N)N))r r r �__doc__r r- r/ r1 r �signalr4 r8 Zexportr< r7 r% rB r+ r0 rN rR rT rU rX rY rZ r? r^ r_ ra rS rc rd rh rn ro rq r~ r� r� � __classcell__r r )r! r r % sv ' )r5 r Z tuned.logsZtuned.exceptionsr r Z tuned.constsr' Ztuned.utils.commandsr �__all__r6 �getr# �objectr Z interfacesZExportableInterfacer r r r r �<module> s