Edit File: plugin_service.cpython-36.pyc
3 ��c�) � @ s� d dl mZ ddlZddljZd dlT ddlZddlZddlZ ddl m Z e j j � Ze � ZG dd� d�ZG dd � d �ZG d d � d e�ZG d d � d e�ZG dd� de�ZG dd� de�ZG dd� de�ZG dd� dej�ZdS )� )�base� N)�*)�commandsc @ s e Zd Zddd�ZdS )�ServiceNc C s || _ || _|| _|| _d S )N)�enable�start�cfg_file�runlevel)�selfr r r r � r �$/usr/lib/python3.6/plugin_service.py�__init__ s zService.__init__)NNNN)�__name__� __module__� __qualname__r r r r r r s r c @ s4 e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd S ) � InitHandlerc C s( t jdg�\}}|dkr$|j� d S d S )Nr r r ���)�cmd�execute�split)r �retcode�outr r r � runlevel_get s zInitHandler.runlevel_getc C s t jddg� d S )NZtelinit�q)r r )r r r r � daemon_reload s zInitHandler.daemon_reloadc C s d S )Nr )r �namer r r r � cfg_install s zInitHandler.cfg_installc C s d S )Nr )r r r r r r � cfg_uninstall s zInitHandler.cfg_uninstallc C s d S )Nr )r r r r r r � cfg_verify" s zInitHandler.cfg_verifyN)r r r r r r r r r r r r r s r c @ s< e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd S )�SysVBasicHandlerc C s t jd|dg� d S )N�servicer )r r )r r r r r r ' s zSysVBasicHandler.startc C s t jd|dg� d S )Nr! �stop)r r )r r r r r r" * s zSysVBasicHandler.stopc C s t � �d S )N)�NotImplementedError)r r r r r r r - s zSysVBasicHandler.enablec C s t � �d S )N)r# )r r r r r r �disable0 s zSysVBasicHandler.disablec C s" t jd|dgdgd�\}}|dkS )Nr! �statusr )� no_errors)r r )r r r r r r r � is_running3 s zSysVBasicHandler.is_runningc C s t � �d S )N)r# )r r r r r r � is_enabled7 s zSysVBasicHandler.is_enabledN) r r r r r" r r$ r' r( r r r r r & s r c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )� SysVHandlerc C s t jdd||dg� d S )N� chkconfigz--level�on)r r )r r r r r r r ; s zSysVHandler.enablec C s t jdd||dg� d S )Nr* z--level�off)r r )r r r r r r r$ > s zSysVHandler.disablec C sB t jdd|g�\}}|dkr>|jdt|� �d d d� dkS d S )Nr* z--listr z%s:r � r+ )r r r �str)r r r r r r r r r( A s zSysVHandler.is_enabledN)r r r r r$ r( r r r r r) : s r) c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )� SysVRCHandlerc C s t jdd||dg� d S )Nz sysv-rc-confz--levelr+ )r r )r r r r r r r F s zSysVRCHandler.enablec C s t jdd||dg� d S )Nz sysv-rc-confz--levelr, )r r )r r r r r r r$ I s zSysVRCHandler.disablec C sB t jdd|g�\}}|dkr>|jdt|� �d d d� dkS d S )Nz sysv-rc-confz--listr z%s:r r- r+ )r r r r. )r r r r r r r r r( L s zSysVRCHandler.is_enabledN)r r r r r$ r( r r r r r/ E s r/ c @ sD e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dS )� OpenRCHandlerc C s& t jddg�\}}|dkr"|j� S d S )Nz rc-statusz-rr )r r �strip)r r r r r r r Q s zOpenRCHandler.runlevel_getc C s t jd|dg� d S )Nz rc-servicer )r r )r r r r r r U s zOpenRCHandler.startc C s t jd|dg� d S )Nz rc-servicer" )r r )r r r r r r" X s zOpenRCHandler.stopc C s t jdd||g� d S )Nz rc-update�add)r r )r r r r r r r [ s zOpenRCHandler.enablec C s t jdd||g� d S )Nz rc-update�del)r r )r r r r r r r$ ^ s zOpenRCHandler.disablec C s" t jd|dgdgd�\}}|dkS )Nz rc-servicer% r )r& )r r )r r r r r r r r' a s zOpenRCHandler.is_runningc C s2 t jdd|g�\}}ttjdtj|� d |��S )Nz rc-updateZshowz\b)r r �bool�re�search�escape)r r r r r r r r r( e s zOpenRCHandler.is_enabledN) r r r r r r" r r$ r' r( r r r r r0 P s r0 c @ s\ e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dd� Z dd� Z dd� Z dS )�SystemdHandlerc C s dS )N� r )r r r r r k s zSystemdHandler.runlevel_getc C s t jdd|g� d S )N� systemctlZrestart)r r )r r r r r r n s zSystemdHandler.startc C s t jdd|g� d S )Nr: r" )r r )r r r r r r" q s zSystemdHandler.stopc C s t jdd|g� d S )Nr: r )r r )r r r r r r r t s zSystemdHandler.enablec C s t jdd|g� d S )Nr: r$ )r r )r r r r r r r$ w s zSystemdHandler.disablec C s" t jdd|gdgd�\}}|dkS )Nr: z is-activer )r& )r r )r r r r r r r r' z s zSystemdHandler.is_runningc C s> t jdd|gdgd�\}}|j� }|dkr.dS |dkr:dS d S ) Nr: z is-enabledr )r& ZenabledTZdisabledF)r r r1 )r r r r r r% r r r r( ~ s zSystemdHandler.is_enabledc C s� t jd||f � tjj|�s0t jd| � d S tj| }ytj|tj � W n2 t k r~ } zt jd||f � d S d }~X nX t j ||� | j � d S )NzCinstalling service configuration overlay file '%s' for service '%s'z)Unable to find service configuration '%s'z#Unable to create directory '%s': %s)�log�info�os�path�exists�error�consts�SERVICE_SYSTEMD_CFG_PATH�makedirsZDEF_SERVICE_CFG_DIR_MODE�OSErrorr �copyr )r r r �dirpath�er r r r � s zSystemdHandler.cfg_installc C sn t jd||f � tj| }d|tjj|�f }tj|� | j � ytj |� W n t t fk rh Y nX d S )NzEuninstalling service configuration overlay file '%s' for service '%s'z%s/%s) r; r< rA rB r= r> �basenamer �unlinkr �rmdir�FileNotFoundErrorrD )r r r rF r> r r r r � s z SystemdHandler.cfg_uninstallc C s� |d kr d S dt j| tjj|�f }tjj|�sHtjd||f � dS tjj|�sjtjd||f � dS tj |�}tj |�}||kS )Nz%s/%sz.Unable to find service '%s' configuration '%s'Fz0Service '%s' configuration not installed in '%s') rA rB r= r> rH r? r; r@ r Z sha256sum)r r r r> Z sha256sum1Z sha256sum2r r r r � s zSystemdHandler.cfg_verifyN) r r r r r r" r r$ r' r( r r r r r r r r8 i s r8 c sj e Zd ZdZ� fdd�Zdd� Zdd� Zdd � Zd d � Zd d � Z dd� Z dd� Z dd� Z ddd�Z � ZS )� ServicePlugina1 `service`:: Plug-in for handling sysvinit, sysv-rc, openrc and systemd services. + The syntax is as follows: + [subs="+quotes,+macros"] ---- [service] service.__service_name__=__commands__[,file:__file__] ---- + Supported service-handling `_commands_` are `start`, `stop`, `enable` and `disable`. The optional `file:__file__` directive installs an overlay configuration file `__file__`. Multiple commands must be comma (`,`) or semicolon (`;`) separated. If the directives conflict, the last one is used. + The service plugin supports configuration overlays only for systemd. In other init systems, this directive is ignored. The configuration overlay files are copied to `/etc/systemd/system/__service_name__.service.d/` directories. Upon profile unloading, the directory is removed if it is empty. + With systemd, the `start` command is implemented by `restart` in order to allow loading of the service configuration file overlay. + NOTE: With non-systemd init systems, the plug-in operates on the current runlevel only. + .Start and enable the `sendmail` service with an overlay file ==== ---- [service] service.sendmail=start,enable,file:${i:PROFILE_DIR}/tuned-sendmail.conf ---- The internal variable `${i:PROFILE_DIR}` points to the directory from which the profile is loaded. ==== c s& t t| �j||� d| _| j� | _d S )NT)�superrL r Z_has_dynamic_options�_detect_init_system� _init_handler)r �args�kwargs)� __class__r r r � s zServicePlugin.__init__c C s t j|dgd�\}}|dkS )Nr )r& )r r )r Zcommandr r r r r � _check_cmd� s zServicePlugin._check_cmdc C s� | j ddg�r tjd� t� S | j dg�r:tjd� t� S | j ddg�rXtjd� t� S | j d dg�rvtjd � t� S tjd ��d S ) Nr: r% zdetected systemdr* zdetected generic sysvinitz update-rc.dz-hzdetected sysv-rcz rc-updatezdetected openrcz8Unable to detect your init system, disabling the plugin.) rS r; �debugr8 r) r/ r0 � exceptionsZNotSupportedPluginException)r r r r rN � s z!ServicePlugin._detect_init_systemc C s� t jd|�}t� }x~|D ]v}|dkr,d|_q|dkr<d|_q|dkrLd|_q|dkr\d|_q|d d� d kr||dd � |_qtjd ||f � qW |S ) Nz \s*[,;]\s*r Tr$ Fr r" � zfile:z*service '%s': invalid service option: '%s')r5 r r r r r r; r@ )r r �val�lr! �ir r r �_parse_service_options� s z$ServicePlugin._parse_service_optionsc s6 d|_ d|_tj� fdd�|jj� D ��� _i |_d S )NFTc sT g | ]L\}}|d d� dkrt |�dkr|dd � � j|dd � � jj|��f�qS )N� zservice.)�lenrZ Z _variables�expand)�.0Zoption�value)r r r � <listcomp> s z0ServicePlugin._instance_init.<locals>.<listcomp>)Z_has_dynamic_tuningZ_has_static_tuning� collections� OrderedDictZoptions�items� _services�_services_original)r �instancer )r r �_instance_init s z ServicePlugin._instance_initc C s d S )Nr )r rf r r r �_instance_cleanup s z ServicePlugin._instance_cleanupc C sT |r| j j|� n|d k r&| j j|� |r:| j j||� n|d k rP| j j||� d S )N)rO r r" r r$ )r r r r r r r r �_process_service s z ServicePlugin._process_servicec C s� | j j� }|d kr tjd� d S x�| jj� D ]�}| j j|d |�}| j j|d �}t|||d j |�|j |d < |d j r�| j j |d |d j � | j |d |d j |d j|� q,W d S )NzCannot detect runlevelr r )rO r r; r@ rd rc r( r' r r re r ri r r )r rf r r! r( r' r r r �_instance_apply_static s z$ServicePlugin._instance_apply_staticc C sH | j j� }|d kr&tjtjd � dS d}�x| jj� D �]}| j j|d |d j �}|r~tj tj d |d |d j f � nR|d k r�tjtjd |d |d j f � d}n"tj tj d |d |d j f � | j j |d |�}| j j|d �} | jd |d |d j| |�dk�rd}| jd |d |d j||�dkr8d}q8W |S ) Nzcannot detect runlevelFTr r z'service '%s' configuration '%s' matchesz'service '%s' configuration '%s' differsz service '%s' configuration '%s'z %s runningz %s enabled)rO r r; r@ rA ZSTR_VERIFY_PROFILE_FAILrd rc r r r< ZSTR_VERIFY_PROFILE_OKZ STR_VERIFY_PROFILE_VALUE_MISSINGr( r' Z _verify_valuer r ) r rf Zignore_missingZdevicesr �retr! Zret_cfg_verifyr( r' r r r �_instance_verify_static$ s( $""$"z%ServicePlugin._instance_verify_staticFc C sL xFt |jj� �D ]4\}}|jr.| jj||j� | j||j|j|j � qW d S )N) �listre rc r rO r ri r r r )r rf Z full_rollbackr r_ r r r �_instance_unapply_static< s z&ServicePlugin._instance_unapply_static)F)r r r �__doc__r rS rN rZ rg rh ri rj rl rn � __classcell__r r )rR r rL � s ( rL )r9 r ra Z tuned.constsrA Z decoratorsr= r5 Z tuned.logsZtunedZtuned.utils.commandsr Zlogs�getr; r r r r r) r/ r0 r8 ZPluginrL r r r r �<module> s" B