File: //usr/lib/python3.6/site-packages/tracer/resources/__pycache__/applications.cpython-36.pyc
3
�|
`" � @ s� d dl mZ d dlmZ d dlmZmZ d dlmZ d dl m
Z
d dl m Z d dl
mZ d dlmZ d d lmZ d d
lZd d
lZG d d � d e�ZG d
d� d�ZG dd� de�Zd
S )� )�absolute_import)�minidom)�DATA_DIR�USER_CONFIG_DIRS)� PathNotFound)�ApplicationsCollection)�_)� Processes)�System)� SystemdDbusNc @ s� e Zd Zedd� ege �Zddddddd �Zed
Zd Z e
d d
� �Z e
dd� �Z e
dd� �Z
edd� �Zedd� �Ze
i fdd��Ze
dd� �Zd S )� Applicationsc C s | d S )Nz/applications.xml� )�xr
r
�"/usr/lib/python3.6/applications.py�<lambda># s zApplications.<lambda>ZdaemonZstaticZsession� applicationZerasedZ undefined)�DAEMON�STATIC�SESSION� APPLICATIONZERASED�UNDEFr Nc C sZ t jst j� x0t jD ]&}|j| krd|kr0|S t j|j�S qW t| t jd d d dd��S )N�renamer F)�name�type�helper�note�ignore)r �_apps�_load_definitionsr �findr � Application�TYPES)Zapp_name�appr
r
r r 0 s
zApplications.findc C s t jst j� t jS )N)r r r r
r
r
r �allB s zApplications.allc C sb t � t_xTtjD ]J} ytj| � W q tk
rX } ztjj| �t krH|�W Y d d }~X qX qW d S )N)
r r r � DEFINITIONS�_loadr �os�path�dirnamer )�fileZexr
r
r r I s z Applications._load_definitionsc
C s� y(t |d��}tj|j� �}W d Q R X W n tk
rD td��Y nX x�|jd�D ]�}| j|� x||jD ]r}|j dkr�t
|j j � �}t
j|� |j dkrh| j|� x6|jD ],}t
|j j � �}t
|j j � �} t
j|| � q�W qhW qRW d S )N�rr � applicationsr" �group)�openr Z parseString�read�IOErrorr ZgetElementsByTagName�_remove_unwanted_children�
childNodesZnodeName�dictZ
attributes�itemsr �_append_application)
�clsr) �fZxmldocr+ �childZattrsr" Z app_attrsZ group_attrsr
r
r r% R s"
zApplications._loadc C s* x$|j D ]}|j|jkr|j|� qW d S )N)r1 ZnodeTypeZ ELEMENT_NODEZ removeChild)r5 Znoder7 r
r
r r0 h s z&Applications._remove_unwanted_childrenc C s� t | �}|j|� |tjkr:tjj|�}tj| j|� nD|jdtj� |jdtj|�� |jdd � |jdd� tjj|� d S )Nr r r r F) r �updater r �index�
setdefault� DEFAULT_TYPE�_helper�append)Z
default_attrsZspecific_attrsr �ir
r
r r4 n s
z Applications._append_applicationc C sj | j tjd kr6tj� dkr(dj| j�S dj| j�S n0| j tjd krNtd�S | j tjd krftd�S d S ) Nr �systemdzsystemctl restart {0}zservice {0} restartr z%You will have to reboot your computerr z'You will have to log out & log in again)r r r! r
� init_system�formatr r )r" r
r
r r< } s zApplications._helper)�__name__�
__module__� __qualname__�mapr r r$ r! r; r � staticmethodr r# r � classmethodr% r0 r4 r< r
r
r
r r ! s" r c @ s� e Zd ZdZeZdZdd� Zdd� Zdd� Z d d
� Z
d d � Z d
d� Z dd� Z
dd� Zdd� Zdd� Zdd� Zedd� �Zedd � �Zed d � �Zed d � �Zed!d"� �Zed#d$� �Zed%d&� �Zed'd(� �Zed)d*� �ZdZ dS )+r a�
Represent the application defined in ``applications.xml``
:param str name: The name of the application
:param str type: See ``Applications.TYPES`` for possible values
:param str helper: Describes how to restart the applications
:param bool note: Provides additional informations to the ``helper``
:param bool ignore: If ``True``, the application won't be printed
:param Processes processes_factory: Class providing list of running processes
Nc C s
|| _ d S )N)� _attributes)�selfZattributes_dictr
r
r �__init__� s zApplication.__init__c C s t |t�o| j|jkS )N)�
isinstancer r )rI �otherr
r
r �__eq__� s zApplication.__eq__c C s | j |� S )N)rM )rI rL r
r
r �__ne__� s zApplication.__ne__c C s
t | j�S )N)�hashr )rI r
r
r �__hash__� s zApplication.__hash__c C s
| j | S )N)rH )rI �itemr
r
r � __getattr__� s zApplication.__getattr__c C s
t | j�S )N)�lenrH )rI r
r
r �__len__� s zApplication.__len__c C s
|| j kS )N)rH )rI rQ r
r
r � __contains__� s zApplication.__contains__c C s d| j j d | jd d S )N�<z: r �>)� __class__rB rH )rI r
r
r �__str__� s zApplication.__str__c C s | j � S )N)rY )rI r
r
r �__repr__� s zApplication.__repr__c C s | j j||� d S )N)rH r: )rI �key�valuer
r
r r: � s zApplication.setdefaultc C s t |t�r|j}| jj|� d S )N)rK r rH r8 )rI �valuesr
r
r r8 � s
zApplication.updatec C s | j o| j d jS )Nr )� instances�is_interpreted)rI r
r
r r_ � s zApplication.is_interpretedc C s( t jdt| j��rdS | jo&| jd jS )Nzssh-.*-sessionTr )�re�search�strr r^ �
is_session)rI r
r
r rc � s zApplication.is_sessionc C sH | j d tjd ks | j d S | jr.tjd S | jr>tjd S tjS d S )Nr r r r )rH r r! rc �
is_servicer; )rI r
r
r r � s
zApplication.typec C s$ t j� dkr t� jdj| j��S d S )Nr? z {0}.service)r
r@ r Zunit_path_from_idrA r )rI r
r
r rd � s zApplication.is_servicec C sV | j d r| j d ntj| �}tj� dkrR| jtjd krR|rR|jd� rRd| }|S )Nr �rootr zsudo )rH r r< r
�userr r! �
startswith)rI r r
r
r r � s
zApplication.helperc C s | j s
d S ttjd| j ��S )Nz\{.*\})r �boolr` ra )rI r
r
r �helper_contains_formating� s z%Application.helper_contains_formatingc C s | j s
d S ttjd| j ��S )Nz\{NAME\})r rh r` ra )rI r
r
r �helper_contains_name� s z Application.helper_contains_namec C sT g }| j rP| js |j| j � n2x0| jD ]&}|j| j j| j|j|j|jd�� q&W |S )ab
Return the list of helpers which describes how to restart the application.
When no ``helper_format`` was described, empty list will be returned.
If ``helper_format`` contains process specific arguments such a ``{PID}``, etc.
list will contain helper for every application instance.
In other cases, there will be just one helper in the list.
)�NAMEZPNAMEZPID�EXE)r ri r= �affected_instancesrA r �pidZexe)rI �helpers�processr
r
r ro � s
zApplication.helpersc s � j j� j� fdd��S )zo
Return collection of processes with same name as application.
I.e. running instances of the application
c s � j d | j� | jgkS )Nr )rH r � real_name)rp )rI r
r r s z'Application.instances.<locals>.<lambda>)�processes_factoryr# Zfiltered)rI r
)rI r r^ s
zApplication.instances) rB rC rD �__doc__r rr rH rJ rM rN rP rR rT rU rY rZ r: r8 �propertyr_ rc r rd r ri rj ro r^ rm r
r
r
r r � s0 r c @ s e Zd Zedd� �ZdS )�AffectedApplicationc C s� t j� dkrpt� }| jrp|j| jd j�rp|j| jd jd�sp|j| jd jd�}|rptj d|�rptj
dd|�S | j r�| jd j S | j
d S )Nr? r ZPAMName�Idz
\.service$� r )r
r@ r r^ Zunit_path_from_pidrn Z has_service_property_from_pidZget_unit_property_from_pidr` ra �subr_ rq rH )rI Zbusrv r
r
r r s zAffectedApplication.nameN)rB rC rD rt r r
r
r
r ru s ru )Z
__future__r Zxml.domr Z tracer.pathsr r Ztracer.resources.exceptionsr Z tracer.resources.collectionsr Ztracer.resources.langr Ztracer.resources.processesr Ztracer.resources.systemr
Z tracer.resources.SystemdDbusr r& r` �objectr r ru r
r
r
r �<module> s m