Sindbad~EG File Manager
3
" eq# � @ s� d dl mZ d dlmZ ddlZddlZddlZddlZddlmZm Z ddl
m Z ddl m
Z
G dd � d e�ZG d
d � d e�ZG d d
� d
e�ZG dd� de
ee��ZG dd� de�ZdS )� )�ProcessesCollection)�FilenameCleaner� N)�PIPE�Popen)�Timer)�with_metaclassc @ s$ e Zd Zedd� �Zedd� �ZdS )� Processesc C s& yt j� S tk
r t j� S X d S )N)�psutil�pids�AttributeErrorZ get_pid_list� r
r
� /usr/lib/python3.6/processes.pyr ! s zProcesses.pidsc C sZ t � } xNtj� D ]B}y| jt|�� W q tjk
r< Y q tjk
rP Y qX qW | S )N)r r r �append�Processr
Z
NoSuchProcess� AccessDenied)Z processes�pidr
r
r �all( s
z
Processes.allN)�__name__�
__module__� __qualname__� staticmethodr r r
r
r
r r s r c @ s� e Zd ZdZd!dd�Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dd� Z dd� Z
d"dd�Zdd� Zd d � Zd#d d �ZdS )$�ProcessWrappera�
Wrapper for ``psutil.Process class``
Library ``psutil`` is not backward compatible from version 2.x.x to 1.x.x.
Purpose of this class is cover incompatibility in ``psutil.Process`` class and
provide interface of new version. It allows using new interface even with
old version of ``psutil``.
Note that, for performance reasons, process information is cached at
object creation. To force a refresh, invoke the ``rebuild_cache()``
method.
Nc C s t j|�| _| j� d S )N)r
r �_process�
rebuild_cache)�selfr r
r
r �__init__A s zProcessWrapper.__init__c C s
t | j�S )N)�boolr )r r
r
r � __nonzero__E s zProcessWrapper.__nonzero__c C s | j jddddddgd�| _d S )N�name�exe�cmdline�ppid�username� create_time)Zattrs)r Zas_dict� _procdict)r r
r
r r H s z ProcessWrapper.rebuild_cachec C s| yZ| j d�dkrX| j d�| j d�krXt| j d��dkrX| j d�d jd�d }dj|�S W n tjk
rp Y nX | j d�S ) Nr Zsshdr r! r �@r zssh-{0}-session)�_attr�len�split�formatr
r )r r# r
r
r r K s &zProcessWrapper.namec C s
| j d�S )Nr )r' )r r
r
r r W s zProcessWrapper.exec C s
| j d�S )Nr! )r' )r r
r
r r! Z s zProcessWrapper.cmdlinec C s
| j d�S )Nr" )r' )r r
r
r r" ] s zProcessWrapper.ppidc C s
| j d�S )N�parent)r' )r r
r
r r+ ` s zProcessWrapper.parentc C s
| j d�S )Nr# )r' )r r
r
r r# c s zProcessWrapper.usernamec C s
| j d�S )Nr$ )r' )r r
r
r r$ f s zProcessWrapper.create_timeFc C s\ dj |�}|| jkrRy| jj|�| j|< W n& tk
rP | jj|�| j|< Y nX | j| S )Nz children-{0})r* r% r �childrenr Z get_children)r � recursive�keyr
r
r r, i s
zProcessWrapper.childrenc C sP || j krFt| j|�}y|� | j |< W n tk
rD || j |< Y nX | j | S )N)r% �getattrr � TypeError)r r �attrr
r
r r' r s
zProcessWrapper._attrc C s t | j|�S )N)r/ r )r �itemr
r
r � __getattr__{ s zProcessWrapper.__getattr__Tc C s` dj |�}|| jkrVy| jj|d�| j|< W n( tk
rT | jj|d�| j|< Y nX | j| S )Nzmemory_maps-{0})�grouped)r* r% r � memory_mapsr Zget_memory_maps)r r4 r. r
r
r r5 s
zProcessWrapper.memory_maps)N)F)T)r r r �__doc__r r r r r r! r" r+ r# r$ r, r' r3 r5 r
r
r
r r 3 s
r c s( e Zd ZdZ� fdd�Zdd� Z� ZS )� ProcessMetaz�
Caching metaclass that ensures that only one ``Process`` object is ever
instantiated for any given PID. The cache can be cleared by calling
``Process.reset_cache()``.
Based on https://stackoverflow.com/a/33458129
c s6 t t� �j|||� � fdd�}|� t� d|� d S )Nc s
i � _ d S )N)�_cacher
)�clsr
r � reset_cache� s z)ProcessMeta.__init__.<locals>.reset_cacher: )�superr7 r �setattr)r9 r �basesZ
attributesr: )� __class__)r9 r r � s zProcessMeta.__init__c O sJ |d }|| j kr@| j| f|�|�}| j|f|�|� || j |< | j | S )Nr )r8 �__new__r )r9 �args�kwargsr r r
r
r �__call__� s
zProcessMeta.__call__)r r r r6 r rB �
__classcell__r
r
)r> r r7 � s r7 c s� e Zd ZdZdd� Zdd� Zdd� Zed d d
��Ze d d � �Z
d
d� Z � fdd�Z d � fdd� Z
e � fdd��Ze dd� �Ze dd� �Ze dd� �Ze d d � �Z� ZS ) r aS
Represent the process instance uniquely identifiable through PID
For all class properties and methods, please see
http://pythonhosted.org/psutil/#process-class
Below listed are only reimplemented ones.
For performance reasons, instances are cached based on PID, and
multiple instantiations of a ``Process`` object with the same PID will
return the same object. To clear the cache, invoke
``Process.reset_cache()``. Additionally, as with ``ProcessWrapper``,
process information is cached at object creation. To force a refresh,
invoke the ``rebuild_cache()`` method on the object.
c C s | j |j kS )zBFor our purposes, two processes are equal when they have same name)r )r �processr
r
r �__eq__� s zProcess.__eq__c C s | j |� S )N)rE )r rD r
r
r �__ne__� s zProcess.__ne__c C s
t | j�S )N)�hashr )r r
r
r �__hash__� s zProcess.__hash__� �?c
C sJ t dd| gttd�}t||j�}z|j� |j� |jdkS |j� X dS )z�
Process arguments could be referring to files on remote filesystems and
os.path.isfile will hang forever if the shared FS is offline.
Instead, use a subprocess that we can time out if we can't reach some file.
Ztestz-f)�stdout�stderrr N)r r r �kill�startZ communicate�
returncodeZcancel)Z file_pathZtimeoutrD Ztimerr
r
r � safe_isfile� s
zProcess.safe_isfilec C sl g }x"| j � D ]}|jtj|j�� qW x:| j� dd � D ]&}tjj|�sLq:tj |�r:|j|� q:W t
|�S )Nr ) r5 r r �strip�pathr! �os�isabsr rO �sorted)r �filesZmmap�argr
r
r rU � s
z
Process.filesc C s | j � rt| j � �S dS )zGThe parent process casted from ``psutil.Process`` to tracer ``Process``N)r" r )r r
r
r r+ � s zProcess.parentc s( yt t| �j� S tk
r" dS X dS )zcThe user who owns the process. If user was deleted in the meantime,
``None`` is returned instead.N)r; r r# �KeyError)r )r> r
r r# � s zProcess.usernameFc s" t t| �j|�}tdd� |D ��S )zjThe collection of process's children. Each of them casted from ``psutil.Process``
to tracer ``Process``.c S s g | ]}t |j��qS r
)r r )�.0Zchildr
r
r �
<listcomp>� s z$Process.children.<locals>.<listcomp>)r; r r, r )r r- r, )r> r
r r, � s zProcess.childrenc sB t t| �j� }|jd�r$|dd� }d|kr>|d|jd�� }|S )zcThe absolute path to process executable. Cleaned from arbitrary strings
which appears on the end.z#newr � �;���)r; r r �endswith�index)r r )r> r
r r � s
z Process.exec C s | j � dkS )N�python)r_ )r )r r
r
r �is_interpreted s zProcess.is_interpretedc C s8 | j � }|d krd S | j� }|d ks0||j � kr4dS d S )NT)�terminalr+ )r ra r+ r
r
r �
is_session s zProcess.is_sessionc C s@ | j r8x0| j� dd � D ] }tjj|�rtjj|�S qW | j� S )Nr )r` r! rR rQ �isfile�basenamer )r rV r
r
r � real_name s
zProcess.real_namec C s� t j jtj� �}t j j| j� �}|| }d}|jdkrFt|j�d }n`|jdkrhtt|jd ��d }n>|jdkr�tt|jd ��d }n |jdkr�tt|j��d }|S )
z�
The time of how long process is running. Returned as string
in format ``XX unit`` where unit is one of
``days`` | ``hours`` | ``minutes`` | ``seconds``
� r z days�< z hoursz minutesz secondsi i )�datetimeZ
fromtimestamp�timer$ Zdays�strZseconds�int)r ZnowZstartedZ started_strr
r
r �str_started_ago s
zProcess.str_started_ago)rI )F)r r r r6 rE rF rH r rO �propertyrU r+ r# r, r r` rb re rl rC r
r
)r> r r � s
r c @ s& e Zd ZdZdZddd�Zdd� ZdS )�AffectedProcessNc C s t j| |� t� | _t� | _d S )N)r r �set�packagesrU )r r r
r
r r 9 s zAffectedProcess.__init__c C s$ | j j|j �| _ | jj|j�| _d S )N)rU �unionrp )r rD r
r
r �update> s zAffectedProcess.update)N)r r r rp rU r rr r
r
r
r rn 5 s
rn )� collectionsr r r
rh ri rR �
subprocessr r Z threadingr Zsixr �objectr r �typer7 r rn r
r
r
r �<module> s V
Sindbad File Manager Version 1.0, Coded By Sindbad EG ~ The Terrorists