3 `��g�4�@s�ddlmZddlZddlZddlmZddlZddljZddlm Z dgZ ejj�Z Gdd�de�ZGdd�dejjj�ZdS) �)�exportsN)�TunedException)�commands� Controllerc@s,eZdZdd�Zdd�Zdd�Zdd�Zd S) � TimerStorecCst�|_tj�|_dS)N)�dict�_timers� threadingZLock�_timers_lock)�self�r� /usr/lib/python3.6/controller.py�__init__szTimerStore.__init__c Cs |j�||j|<WdQRXdS)N)r r)r�token�timerrrr �store_timerszTimerStore.store_timercCsB|j�2y|j|}|j�|j|=WnYnXWdQRXdS)N)r r�cancel)rrrrrr � drop_timers zTimerStore.drop_timerc Cs<|j�,x|jj�D]}|j�qW|jj�WdQRXdS)N)r r�valuesr�clear)rrrrr � cancel_allszTimerStore.cancel_allN)�__name__� __module__�__qualname__rrrrrrrr r s rcs�eZdZdZ�fdd�Zdd�Zdd�Zdd �Zej d �dd��Z d d�Zejdd�dUdd��Z ejdd�dVdd��Zejdd�dWdd��ZdXdd�Zejdd�dYdd��Zejdd�dZdd ��Zd!d"�Zejdd#�d[d$d%��Zejdd#�d\d&d'��Zejdd�d]d(d)��Zejdd*�d^d+d,��Zejdd�d_d-d.��Zejdd�d`d/d0��Zejdd�dad1d2��Zejdd3�dbd4d5��Zejdd6�dcd7d8��Zejdd9�ddd:d;��Zejdd�dedd?��Zejdd�dgd@dA��Z ejddB�dhdCdD��Z!ejdd�didEdF��Z"ejddG�djdHdI��Z#ejdd�dkdJdK��Z$ejdLd#�dldMdN��Z%ejddO�dmdPdQ��Z&ejddR�dndSdT��Z'�Z(S)orz� Controller's purpose is to keep the program running, start/stop the tuning, and export the controller interface (currently only over D-Bus). cs8tt|�j�||_||_tj�|_t�|_ t �|_dS)N)�superrr�_daemon�_global_configr ZEvent� _terminater�_cmdr�_timer_store)r�daemonZ global_config)� __class__rr r+s zController.__init__cCsxtjd�|j�}|jjtjtj�}|r6|r6tj�|rb|j j �x|jj|j d�s`tj �qFWtjd�|j�dS)z1 Controller main loop. The call is blocking. zstarting controller�zterminating controllerN)�log�info�startr�get_bool�consts� CFG_DAEMON�CFG_DEF_DAEMONrrrr�waitZperiod_check�stop)r�resr rrr �run3s zController.runcCs|jj�dS)N)r�set)rrrr � terminateFszController.terminatecCs0|jjj�s,|jjj�|j�s,|jjj�dS)N)rZ_sighup_processingZis_setr.�reloadr)rrrr �sighupIszController.sighupZsbscCsdS)Nr)r�profile_name�resultZerrstrrrr �profile_changedOszController.profile_changedcCstjj|�|jj|�dS)N)�tuned�logs�log_capture_finishrr)rrrrr �_log_capture_abortXszController._log_capture_abortZii�sNcCsf|dkrdStjj|�}|dkr$dS|dkrVtj||j|gd�}|jj||�|j�|dkrbdS|S)N�r)�args) r5r6�log_capture_startr ZTimerr8rrr%)rZ log_levelZtimeout�callerrrrrr r<\szController.log_capture_startcCs4|dkrdStjj|�}|jj|�|dkr0dS|S)Nr:)r5r6r7rr)rrr=r,rrr r7js zController.log_capture_finishr:�bcCsD|dkrdS|jjtjtj�r:|jj�r,dS|jj�s:dS|jj�S)Nr:FT) rr&r'r(r)r� is_running� is_enabledr%)rr=rrr r%rs zController.startFcCs,|jj�sd}n|jj|d�}|jj�|S)NT)�profile_switch)rr?r+rr)rrAr,rrr �_stop}s zController._stopcCs|dkrdS|jdd�S)Nr:F)rA)rB)rr=rrr r+�szController.stopcCsp|dkrdS|jj�r*|jdd�}|s*dSy|jj�Wn.tk rf}ztjd|�dSd}~XnX|j�S)Nr:FT)rAzFailed to reload TuneD: %s)rr?rBZreload_profile_configrr#�errorr%)rr=Zstop_ok�errr r0�s zController.reloadcCs�|jj�}d}d}d}z�y$|r,|jjdd�|jj||�Wnftjjk r�}zFd}t|�}|r�|jjj |kr�t jd|�d}nt jd|�WYdd}~XnXWd|r�|r�t jd|�n|s�t j d�|jj�X||fS) N�OKTF)rAz@Failed to reapply profile '%s'. Did it change on disk and break?zFailed to apply profile '%s'z>Applying previously applied (possibly out-dated) profile '%s'.z$Applying previously applied profile.)rr?r+Zset_profiler5� exceptionsr�str�profile�namer#rC�warnr$r%)rr2�manualZwas_running�msg�successZreapplyrDrrr �_switch_profile�s, $ zController._switch_profilez(bs)cCs(|dkrdS|jj|�sdS|j|d�S)Nr:F�Unauthorized�Invalid profile_nameT)FrO)FrP)r� is_valid_namerN)rr2r=rrr �switch_profile�s zController.switch_profilecCs |dkrdS|j�}|j|d�S)Nr:FrO)FrO)�recommend_profilerN)rr=r2rrr �auto_profile�szController.auto_profilecCs*|dkrdS|jjdk r"|jjjSdSdS)Nr:)rrHrI)rr=rrr �active_profile�s zController.active_profilez(ss)cCs�|dkrdS|jj}|dkrpy"|jj�\}}|dkr<|dk }Wn0tk rn}zd}t|�}||fSd}~XnX|rztjntj}|dfS)Nr:�unknownrO)rVrO) rrKrZget_active_profilerrGr'ZACTIVE_PROFILE_MANUALZACTIVE_PROFILE_AUTO)rr=rKrHrD�moderCrrr �profile_mode�szController.profile_modecCs|dkrdS|jjpdS)Nr:)r�post_loaded_profile)rr=rrr rY�szController.post_loaded_profilecCsB|dkrdS|jj�r |jj�|jj�r>|jjddddd�dS)Nr:FT)Zsave_instantly)rr?r+r@Zset_all_profiles)rr=rrr �disable�s zController.disablecCs|dkrdS|jj�S)Nr:F)rr?)rr=rrr r?�szController.is_running�ascCs|dkrgS|jjjj�S)Nr:)r�profile_loader�profile_locatorZget_known_names)rr=rrr �profiles�szController.profilesza(ss)cCs|dkrgS|jjjj�S)Nr:)rr\r]Zget_known_names_summary)rr=rrr � profiles2�szController.profiles2z(bsss)cCsT|dks|jj|�rdS|dks*|dkr2|j�}t|jjjj|tj tj gdg��S)Nr:F)Fr:r:r:)rrQrU�tuplerr\r]Zget_profile_attrsr'ZPROFILE_ATTR_SUMMARYZPROFILE_ATTR_DESCRIPTION)rr2r=rrr �profile_infos zController.profile_infocCs|dkrdS|jjj�S)Nr:)rZprofile_recommenderZ recommend)rr=rrr rS szController.recommend_profilecCs|dkrdS|jjdd�S)Nr:F)�ignore_missing)r�verify_profile)rr=rrr rcszController.verify_profilecCs|dkrdS|jjdd�S)Nr:FT)rb)rrc)rr=rrr �verify_profile_ignore_missingsz(Controller.verify_profile_ignore_missingz a{sa{ss}}cCsz|dkriSi}xd|jj�D]V}|jjd�djdd�d}|j�}i||<x$|j�D]\}}t|�|||<qVWqW|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)rr=ZpluginsZplugin_class�plugin_nameZconf_options�key�valrrr rhszController.get_all_pluginscCs*|dks|jj|�rdS|jjt|��S)z"Return docstring of plugin's classr:)rrQr�get_plugin_documentationrG)rrkr=rrr rn.sz#Controller.get_plugin_documentationza{ss}cCs*|dks|jj|�riS|jjt|��S)z�Return dictionary with plugin's parameters and their hints Parameters: plugin_name -- name of plugin Return: dictionary -- {parameter_name: hint} r:)rrQr�get_plugin_hintsrG)rrkr=rrr ro5s zController.get_plugin_hintscCsD|dks|jj|�rdS|jjr@|jjjr@|jjjj|�dSdS)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)rrQrZ_applicationZ_unix_socket_exporterZregister_signal_path)r�pathr=rrr �register_socket_signal_pathCs z&Controller.register_socket_signal_pathZssc Cs�|dkrdS|jj|�sdS|jj|�s,dSd}x2|jjjD]$}|j|kr�sz*Controller.get_instances..cs|jj�kS)N)ryrI)r})rkrr r�sTrEcSs|j|jjfS)N)rIry)r}rrr r�s)rrQrh�keysr#rC�filterrrtru�list�map)rrkr=r|rur)rkr � get_instancess zController.get_instancesz(bsas)cCsv|dkrddgfS|jj|�s(ddgfSx0|jjjD]"}|j|kr4ddtt|j��fSq4Wd|}t j |�d|gfS)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:FrOzInvalid instance_nameTrEzInstance '%s' not found)rrQrrtrurI�sortedr�rwr#rC)rrzr=r}r|rrr �instance_get_devices�s zController.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))rrr�__doc__rr-r/r1r�signalr4r8Zexportr<r7r%rBr+r0rNrRrTrUrXrYrZr?r^r_rarSrcrdrhrnrorqr~r�r�� __classcell__rr)r!r r%sv ' )r5rZ tuned.logsZtuned.exceptionsrr Ztuned.constsr'Ztuned.utils.commandsr�__all__r6�getr#�objectrZ interfacesZExportableInterfacerrrrr �s