File manager

3

�<�e�)�@s�dd
lm
Z

ddlZddljZddlTddlZddlZddlZ	ddl
mZ
e	jj
�Ze�ZGdd�d�ZGdd	�d	�ZGd
d�de�ZGdd
�d
e�ZGdd�de�ZGdd�de�ZGdd�de�ZGdd�de
j�ZdS)�
)
�base�N)
�
*)
�commandsc@seZ
dZddd�
Zd
S)�ServiceNcCs||_|
|_
||_||_dS)
N)�enable�start�cfg_file�runlevel)�selfrrr	r
�r�$/usr/lib/python3.6/plugin_service.py�__init__s



zService.__init__)NNNN)�__name__�
__module__�__qualname__rrrrr
r
s
rc@s4eZ
dZd
d�Zdd�Zdd�Zdd�Zd	d
�ZdS)�InitHandlerc
Cs(tj
d
g
�
\}
}|
dkr$|j�dSdS)Nr
rr
���)�cmd�execute�split)r�retcode�outrrr
�runlevel_gets

zInitHandler.runlevel_getc

Cstj
d
dg�

dS)NZtelinit�
q)rr)
rrrr
�
daemon_reloads
zInitHandler.daemon_reloadc
CsdS)
Nr)r�namer	rrr
�cfg_installs
zInitHandler.cfg_installc
CsdS)
Nr)rrr	rrr
�
cfg_uninstalls
zInitHandler.cfg_uninstallc
CsdS)
Nr)rrr	rrr
�
cfg_verify"s
zInitHandler.cfg_verifyN)rrrrrrrrrrrr
rs

rc@s<eZ
dZd
d�Zdd�Zdd�Zdd�Zd	d
�Zdd�Zd
S)�SysVBasicHandlercCstj
d
|
dg�

dS)N�servicer)rr)rrrrr
r's
zSysVBasicHandler.startcCstj
d
|
dg�

dS)Nr!�stop)rr)rrrrr
r"*s
zSysVBasicHandler.stopc
Cs
t��
dS)
N)
�NotImplementedError)rrr
rrr
r-s
zSysVBasicHandler.enablec
Cs
t��
dS)
N)
r#)rrr
rrr
�disable0s
zSysVBasicHandler.disablecCs"tj
d
|
dgdg
d�\}}|dkS)Nr!�statusr)
�	no_errors)rr)rrrrrrr
�
is_running3s

zSysVBasicHandler.is_runningc
Cs
t��
dS)
N)
r#)rrr
rrr
�
is_enabled7s
zSysVBasicHandler.is_enabledN)	rrrrr"rr$r'r(rrrr
r &s
r c@s$eZ
dZd
d�Zdd�Zdd�ZdS)�SysVHandlercCstj
d
d||
dg�

dS)N�	chkconfigz--level�on)rr)rrr
rrr
r;s
zSysVHandler.enablecCstj
d
d||
dg�

dS)Nr*z--level�off)rr)rrr
rrr
r$>s
zSysVHandler.disablecCsBtj
d
d|
g�
\}}|dkr>|jdt|�
�
ddd�dkSdS)Nr*z--listrz%s:r
�r+)rrr�str)rrr
rrrrr
r(As

zSysVHandler.is_enabledN)rrrrr$r(rrrr
r):s
r)c@s$eZ
dZd
d�Zdd�Zdd�ZdS)�
SysVRCHandlercCstj
d
d||
dg�

dS)Nzsysv-rc-confz--levelr+)rr)rrr
rrr
rFs
zSysVRCHandler.enablecCstj
d
d||
dg�

dS)Nzsysv-rc-confz--levelr,)rr)rrr
rrr
r$Is
zSysVRCHandler.disablecCsBtj
d
d|
g�
\}}|dkr>|jdt|�
�
ddd�dkSdS)Nzsysv-rc-confz--listrz%s:r
r-r+)rrrr.)rrr
rrrrr
r(Ls

zSysVRCHandler.is_enabledN)rrrrr$r(rrrr
r/Es
r/c@sDeZ
dZd
d�Zdd�Zdd�Zdd�Zd	d
�Zdd�Zd
d�Z	dS)�
OpenRCHandlerc
Cs&tj
d
dg�
\}
}|
dkr"|j�SdS)Nz	rc-statusz-rr)rr�strip)rrrrrr
rQs

zOpenRCHandler.runlevel_getcCstj
d
|
dg�

dS)Nz
rc-servicer)rr)rrrrr
rUs
zOpenRCHandler.startcCstj
d
|
dg�

dS)Nz
rc-servicer")rr)rrrrr
r"Xs
zOpenRCHandler.stopcCstj
d
d|
|g�

dS)Nz	rc-update�add)rr)rrr
rrr
r[s
zOpenRCHandler.enablecCstj
d
d|
|g�

dS)Nz	rc-update�del)rr)rrr
rrr
r$^s
zOpenRCHandler.disablecCs"tj
d
|
dgdg
d�\}}|dkS)Nz
rc-servicer%r)
r&)rr)rrrrrrr
r'as

zOpenRCHandler.is_runningcCs2tj
d
d|g�
\}}ttjdtj|
�
d|��
S)Nz	rc-updateZshowz\b)rr�bool�re�search�escape)rrr
rrrrr
r(es

zOpenRCHandler.is_enabledN)
rrrrrr"rr$r'r(rrrr
r0Ps
r0c@s\eZ
dZd
d�Zdd�Zdd�Zdd�Zd	d
�Zdd�Zd
d�Z	dd�Z
dd�Zdd�ZdS)�SystemdHandlerc


Csd
S)N�r)
rrrr
rks
zSystemdHandler.runlevel_getcCstj
d
d|
g�

dS)N�	systemctlZrestart)rr)rrrrr
rns
zSystemdHandler.startcCstj
d
d|
g�

dS)Nr:r")rr)rrrrr
r"qs
zSystemdHandler.stopcCstj
d
d|
g�

dS)Nr:r)rr)rrr
rrr
rts
zSystemdHandler.enablecCstj
d
d|
g�

dS)Nr:r$)rr)rrr
rrr
r$ws
zSystemdHandler.disablecCs"tj
d
d|
gdg
d�\}}|dkS)Nr:z	is-activer)
r&)rr)rrrrrrr
r'zs

zSystemdHandler.is_runningcCs>tj
d
d|
gdg
d�\}}|j�}|dkr.dS|dkr:dSdS)	Nr:z
is-enabledr)
r&ZenabledTZdisabledF)rrr1)rrr
rrr%rrr
r(~s


zSystemdHandler.is_enabledcCs�tj
d
||
f�

tjj|�
s0tjd|�

dStj|
}ytj|tj	�
Wn2t
k
r~
}
ztjd||f�

dSd}~XnXtj||�
|j
�
dS)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)rrr	�dirpath�
errr
r�s












zSystemdHandler.cfg_installcCsntj
d
||
f�

tj|
}d|tjj|�
f}tj|�

|j	�
ytj
|�

Wnttfk
rh


YnXdS)NzEuninstalling service configuration overlay file '%s' for service '%s'z%s/%s)
r;r<rArBr=r>�basenamer�unlinkr�rmdir�FileNotFoundErrorrD)rrr	rFr>rrr
r�s









zSystemdHandler.cfg_uninstallcCs�|dkrdSd
tj
|
tjj|�
f}tjj|�
sHtjd|
|f�

dStjj|�
sjtjd|
|f�

dStj	|�
}tj	|�
}||kS)Nz%s/%sz.Unable to find service '%s' configuration '%s'Fz0Service '%s' configuration not installed in '%s')
rArBr=r>rHr?r;r@rZ	sha256sum)rrr	r>Z
sha256sum1Z
sha256sum2rrr
r�s













zSystemdHandler.cfg_verifyN)
rrrrrr"rr$r'r(rrrrrrr
r8isr8csneZ
dZd
Z�f
dd�Zdd�Zdd�Zdd	�Zd
d�Zdd
�Z	dd�Z
dd�Zdd�Ze
jf
dd�
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&tt
|�j|
|�

d
|_|j�|_dS)NT)�superrLrZ_has_dynamic_options�_detect_init_system�
_init_handler)r�args�kwargs)
�	__class__rr
r�s


zServicePlugin.__init__cCstj
|
d
g
d�\}}|d
kS)Nr)
r&)rr)rZcommandrrrrr
�
_check_cmd�s

zServicePlugin._check_cmdc

Cs�|jd
dg�
rt
jd�

t�S|jdg
�
r:t
jd�

t�S|jddg�
rXt
jd�

t�S|jd	dg�
rvt
jd
�

t�Stjd�
�
dS)Nr:r%zdetected systemdr*zdetected generic sysvinitzupdate-rc.dz-hzdetected sysv-rcz	rc-updatezdetected openrcz8Unable to detect your init system, disabling the plugin.)	rSr;�debugr8r)r/r0�
exceptionsZNotSupportedPluginException)
rrrr
rN�s















z!ServicePlugin._detect_init_systemcCs�tj
d
|�}t�}x~|D]v}|dkr,d|_q|dkr<d|_q|dkrLd|_q|dkr\d|_q|dd�d	kr||dd�|_qtjd
|
|f�

qW|S)Nz
\s*[,;]\s*rTr$Frr"�zfile:z*service '%s': invalid service option: '%s')r5rrrrr	r;r@)rr�val�
lr!�
irrr
�_parse_service_options�s














z$ServicePlugin._parse_service_optionscs6d
|
_d|
_
tj�f
dd�|
jj�D�
�
�_i|
_dS)NFTc
sTg|]L\}
}|
dd
�dkrt|
�
d
kr|
d
d��j
|
d
d��jj|�
�f�qS)N�zservice.)�lenrZZ
_variables�expand)�.0Zoption�value)
rrr
�
<listcomp>
s

z0ServicePlugin._instance_init.<locals>.<listcomp>)Z_has_dynamic_tuningZ_has_static_tuning�collections�OrderedDictZoptions�items�	_services�_services_original)r�instancer)
rr
�_instance_init
s



zServicePlugin._instance_initc
CsdS)
Nr)rrfrrr
�_instance_cleanup	
s
zServicePlugin._instance_cleanupcCsT|r|jj
|
�

n|dk	r&|jj|
�

|r:|jj|
|�
n|dk	rP|jj|
|�
dS)
N)rOrr"rr$)rrrrr
rrr
�_process_service
s







zServicePlugin._process_servicecCs�|jj
�}|dkr tjd
�

dSx�|jj�D]�}|jj|d|�}|jj|d�
}t|||dj	|�|
j
|d<|dj	r�|jj|d|dj	�
|j|d|dj
|dj|�
q,WdS)NzCannot detect runlevelrr
)rOrr;r@rdrcr(r'rr	rerrirr)rrfr
r!r(r'rrr
�_instance_apply_static
s












z$ServicePlugin._instance_apply_staticc
CsH
|jj
�}|dkr&tjtjd
�

dSd}�
x|jj�D�
]}|jj|d|dj	�}|r~tj
tjd|d|dj	f�

nR|dk	r�tjtjd|d|dj	f�

d}n"tj
tjd|d|dj	f�

|jj
|d|�}|jj|d�
}	|jd	|d|dj|	|�dk�
rd}|jd
|d|dj||�dkr8d}q8W|S)Nzcannot detect runlevelFTrr
z'service '%s' configuration '%s' matchesz'service '%s' configuration '%s' differszservice '%s' configuration '%s'z
%s runningz
%s enabled)rOrr;r@rAZSTR_VERIFY_PROFILE_FAILrdrcrr	r<ZSTR_VERIFY_PROFILE_OKZ STR_VERIFY_PROFILE_VALUE_MISSINGr(r'Z
_verify_valuerr)
rrfZignore_missingZdevicesr
�retr!Zret_cfg_verifyr(r'rrr
�_instance_verify_static$
s(








$

"
"


$

"

z%ServicePlugin._instance_verify_staticcCsLxFt|
j
j��
D]4\}}|jr.|jj||j�
|j||j|j|j	�
qWdS)
N)
�listrercr	rOrrirrr
)rrfZrollbackrr_rrr
�_instance_unapply_static<
s



z&ServicePlugin._instance_unapply_static)rrr�__doc__rrSrNrZrgrhrirjrlrAZ
ROLLBACK_SOFTrn�
__classcell__rr)
rRr
rL�s(	
rL)r9rraZtuned.constsrAZ
decoratorsr=r5Z
tuned.logsZtunedZtuned.utils.commandsrZlogs�getr;rrrr r)r/r0r8ZPluginrLrrrr
�<module>
s"









B