Current File : //lib/python3.9/site-packages/tracer/controllers/__pycache__/default.cpython-39.opt-1.pyc
a
" e� � @ s� d dl Z d dlZd dlmZ d dlmZ d dlmZ d dlm Z d dl
m Z d dl m
Z
d dlmZ d d lmZ d d
lmZ d d lmZ d d lmZ zeZW n ey� Y n0 G d
d� de�Z dS )� N)�
HooksObserver)� DefaultView)�InteractiveView)�NoteForHiddenView)�_)�Tracer)�System)� dump_memory)� Applications)�Rules)�HelperControllerc @ sX e Zd ZdZdZdZdd� Zdd� Zdd� Zdd � Z d
d � Z
d d
� Z dd� Z dd� Z
dS )�DefaultControllerNc C s� || _ ttj|jd�tttt� |jd�| _ |j
| j _
|j d | j _ |rN|| j _ | j �
| �|j��| _| j jr�| j�tjd g�| _d S )N)�erased)ZmemoryZhooks_observerr r �DAEMON)�argsr r Zpackage_managerr r r
r r �tracerZnowZ timestampZspecified_packagesZtrace_affected�_user�user� applicationsZ daemons_onlyZ filter_types�TYPES)�selfr Zpackages� r �>/usr/lib/python3.9/site-packages/tracer/controllers/default.py�__init__/ s �
zDefaultController.__init__c C sJ | j js:t� }|�d| j� |�d| j � |�� | �� t| �� � d S )Nr r ) r Z
hooks_onlyr �assignr �render�create_reboot_required_file�exit� status_code)r �viewr r r r B s zDefaultController.renderc C s� t | j�}| �| j| j�D ]}|�|| j� td� qt� }|�d| j� |�dt| j�� |�d| j� t
j d �� |�d| j� t
j d �� |� � d S )N� r � total_count�
session_count�SESSION� static_count�STATIC)
r r �_restartable_applicationsr � print_helper�printr r �len�
count_typer
r r )r �helper_controllerZ applicationr r r r �render_helpersK s
z DefaultController.render_helpersc
C s` t | j�}| �| j| j��d�}t� }|�d|� |�d| j� |�dt| j�� |�d| j�t j
d �� |�d| j�t j
d �� |� � |� d�|� d�|� d� kr��q\t
d td
� � td �}zL|d kr�W d S t|�d
ks�t|�t|�kr�t�|�|t|�d | j� W n& tttf�y: t
td�� Y n0 tj�dtd� d � t� q d S )N�namer r r! r" r# r$ r% �
z0Press application number for help or 'q' to quitz--> �qr � zWrong application numberz
-- z)Press <enter> to get list of applicationsz --)r r r& r �sortedr r r) r* r
r r �getr( r �input�int�
IndexErrorr' � SyntaxError�
ValueError�sys�stdout�write)r r+ Zfilteredr Zanswerr r r �render_interactiveX s*
z$DefaultController.render_interactivec C s\ d}t | j�dkrd}| j�tjd �r,d}| j�tjd �rBd}| j�tjd �rXd}|S ) z�
0 - No affected applications
101 - Found some affected applications
102 - Found some affected daemons
103 - Session restart needed
104 - Reboot needed
r �e r �f r# �g r% �h )r) r r* r
r )r �coder r r r v s z DefaultController.status_codec C sJ | j �tjd �rFtdd��}|�d� W d � n1 s<0 Y dS )zm
If a reboot is needed, create a /run/reboot-required file.
This is how Debian/Ubuntu distros does it.
r% z/run/reboot-required�wz Tracer says reboot is required
N)r r* r
r �openr: )r �fpr r r r � s z-DefaultController.create_reboot_required_filec C s$ |j s |�tjd tjd g�S |S )Nr% r# )�allZ
exclude_typesr
r )r r r r r r r& � s ���z+DefaultController._restartable_applicationsc C s: |dgks|dkrd S |dkr"|S |s.t �� S |d S d S )N�*�rootr )r r )r r r r r r � s zDefaultController._user)�__name__�
__module__� __qualname__r r Z processesr r r, r; r r r& r r r r r r
) s
r
) �osr8 Z tracer.hooksr Ztracer.views.defaultr Ztracer.views.interactiver Z tracer.views.note_for_hiddenr Ztracer.resources.langr Ztracer.resources.tracerr Ztracer.resources.systemr Ztracer.resources.memoryr Z tracer.resources.applicationsr
Ztracer.resources.rulesr Ztracer.controllers.helperr Z raw_inputr3 � NameError�objectr
r r r r �<module> s"
Mini Shell