Edit File: Monitor.cpython-37.pyc
B �]: � @ s0 d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dlZd dl Z y d dl Z W n e k rp Y nX d dl Z d dl Z d dlT d dlmZ d dlmZ d dlmZ d dlmZ G dd� dej�Zd d � Zdd d �Zd ZG dd� dej �Z G dd� dej �Z G dd� d�Z dd� Z!e"dk�r,e!� dS )� N)�*)�Console)� mutnodeset)�ptable)�cmdc @ s e Zd ZdS )�ServerN)�__name__� __module__� __qualname__� r r �F/opt/alt/python37/lib64/python3.7/site-packages/guppy/heapy/Monitor.pyr s r c C s t � | gg g |�\}}}t|�S )N)�select�len)�fd�wait�r�w�xr r r �ioready s r c C s4 x.y | j dd�S tjk r* |r&P Y qX qW d S )Ng�������?)�timeout)�get�queue�Empty)�q�noblockr r r �queue_get_interruptible$ s r )ZCLOSEDc @ sJ e Zd ZdZdd� Zdd� Zdd� Zdd d �Zd d � Zd d� Z dd� Z dS )�Handler� c C s@ | j �d�s<| jj�| � | j�t� | j� d� | j� � d S )Nr � ) �isclosedZtas�server�monitor�remove_connection�dataq�put� CONN_CLOSED�request�shutdown�close)�selfr r r r( 5 s z Handler.closec C s( |� d�s|d7 }| j�|�d�� d S )N� zutf-8)�endswithr& �send�encode)r) r r r r �send_cmd< s zHandler.send_cmdc C s> | j dkr| jddd� | j dkr0| jddd� | j|dd�S )Nz>>> r r )�retdataz<Annex> Zcont)�prompt�exec_cmd)r) r r r r � browser_cmdA s zHandler.browser_cmdr c C s� |d k r| � |� d| _g }x�t| j|�}|d krH| jr<P n t�d� q |tkrTt�|d dkr�d| _|rz|�|d � q�t j � |d � q |d dkr�|d | _ | j� � r�d| _P q�d| _q ds t�q W |r�d�|�S d S )NFr r �DATA�PROMPTT� )r. � promptstater r# �time�sleepr% �EOFError�append�sys�stdout�writer0 �empty�AssertionError�join)r) r r/ r Zdatas�pr r r r1 H s6 zHandler.exec_cmdc C sN xH| j �d�D ]8}|�|�rd|krB|�d�}||d d � �� }|S qW dS )Nr* �=r r5 )� firstdata�split� startswith�index�strip)r) �name�lineZixr r r �get_psi s zHandler.get_psc C s | � d| �}t�|�S )Nzdump %s)r2 �pickle�loads)r) �expr�datar r r �get_valr s zHandler.get_valc C s� d | _ d| _t� | _t�� | _| jj� | � x�y2| j � � � d�}|sLt d��|�t�r^t d��W n t tjfk r| P Y nX |�t�r�|d tt� � }| j�d|f� | j d kr�| jd dd�| _q.| j�d|f� q.W | �� d S ) NFzutf-8z End of file�DONEr4 r )r r/ r3 )r0 r6 r r r �Queuer# r r! �add_connectionZrfile�readline�decoder9 r+ rP �socket�errorZREADLINEr r$ r1 rC r( )r) rN r0 r r r �handlev s* zHandler.handleN)r r ) r r r Zallow_reuse_addressr( r. r2 r1 rJ rO rW r r r r r 2 s ! r c @ s� e Zd ZdZdd� Zdd� Zdd� Zdd � Zd d � Zd d � Z dd� Z dd� Z dd� Z e jjZdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d!� Zd"d#� Zd$d%� Zd&d'� Zd(d)� Zd*d+� Z d,S )-�MonitorConnectionr c C sl i | _ tj�| � d| _d| _d| _d| _|| _t t t ft � | _ }|| j _tj| jdd�| _| j�� d S )Nr z <Monitor> r )�target�args)�aliasesr �Cmd�__init__Zhnor � forceexitr0 r! r Z LOCALHOSTZ HEAPYPORTr r � threading�Thread� run_server�st�start)r) r! �sr r r r] � s zMonitorConnection.__init__c C s. d| _ | jj�d� | j�� dd� | j_d S )Nr r c S s dS )Nr r )r �yr r r �<lambda>� � z)MonitorConnection.close.<locals>.<lambda>)r r rU r' � server_closeZverify_request)r) r r r r( � s zMonitorConnection.closec C s t j�| |� t j�| d� d S )Nr5 )r r\ �default�do_help)r) rI r r r ri � s zMonitorConnection.defaultc C s& | j }x| js|�� qW |�� d S )N)r r Zhandle_requestrh )r) rd r r r ra � s z MonitorConnection.run_serverc C sT |sd S |}y$| � |�}| �|�}| �||�}|S tk rN | �|� Y nX d S )N)�precmdZonecmdZpostcmd� BaseException� handle_error)r) r rI �stopr r r r1 � s zMonitorConnection.exec_cmdc C sR t dtjd� t ddtjd� t t|�tjd� ddl}|�� t dtjd� dS )ztHandle an error gracefully. May be overridden. The default is to print a traceback and continue. z(----------------------------------------)�filez0Exception happened during processing the command� )�endro r N)�printr; �stderr�repr� traceback� print_exc)r) Zcmdlineru r r r rm � s z MonitorConnection.handle_errorc C s, || _ |s|S |�� }x�|d | jkr�| j|d }d|kr�d}xR|dd� D ]B}|�dt|� |�}|�dt|� d�||d� ��}|d }qNW |�dd�|dd� ��}n|d d�|dd� � }|�� }qW |d dk�r(|�d �}|dk�r(||d d� �� }| j� |� |d|� � � }|S ) z*Handle alias expansion and ';;' separator.r �%r Nz%>=rp z%*�aliasz;;r ) ZcurlinerD r[ �replace�strr@ �find�lstripZcmdqueuer: �rstrip)r) rI rZ �iiZtmpArg�marker�nextr r r rk � s0 zMonitorConnection.precmdc C s d| _ dS )Nr )r^ )r) �argr r r �do_exit� s zMonitorConnection.do_exitc C s t d� d S )Nz�exit ----- Exit from the monitor and from the Python process that started it. This makes sure to exit without waiting for the server thread to terminate. See also the q command.)rr )r) r r r � help_exit� s zMonitorConnection.help_exitc C s t d� d S )Nz�h(elp) ----- Without argument, print the list of available commands. With a command name as argument, print help about that command.)rr )r) r r r �help_h� s zMonitorConnection.help_hc C s | � � d S )N)r� )r) r r r � help_help s zMonitorConnection.help_helpc C s0 z(t | j| j| jd�}|�d|j � W d X d S )N)�stdinr< �localsz9Local interactive console. To return to Monitor, type %r.)r r� r< �__dict__ZinteractZEOF_key_sequence)r) r� �conr r r �do_int s zMonitorConnection.do_intc C s t d� d S )Nz�int ----- Local interactive console. This will bring up a Python console locally in the same interpreter process that the Monitor itself.)rr )r) r r r �help_int s zMonitorConnection.help_intc C sL |s | j }t|�}| jj| }|�d�r2td� ntd| � |�t� d S )NZnoninterruptiblezCError: Can not interrupt this remote connection (uses Python < 2.4)z+Sending KeyboardInterrupt to connection %s.)�conid�intr! � connectionsrJ rr r. ZKEYBOARDINTERRUPT)r) r� �cr r r �do_ki s zMonitorConnection.do_kic C s t d� d S )Na\ ki <connection ID> ----- Keyboard Interrupt Send a command to interrupt the remote thread on the specified connection (default is the last one connected to). Notes: It currently only works with Python >= 2.4. The remote thread will not always be awakened, for example if it is waiting in time.sleep(). Sometimes using several ki commands helps.)rr )r) r r r �help_ki% s zMonitorConnection.help_kic C sV dddgg}x8t | jj�� �D ]$\}}|�||�d�|�d�g� q W t|| j� d S )NZCIDZPIDZARGVz target.pidztarget.sys.argv)�listr! r� �itemsr: rJ r r< )r) r� �tableZcidr� r r r �do_lc3 s zMonitorConnection.do_lcc C s t d� d S )Na� lc ----- List Connections. List the currently open connections. The table printed has one line per connection in this form: CID PID ARGV 1 17999 ['/home/nilsson/bin/solitaire.py'] CID is the connection ID, which may be used as argument to the sc command. PID is the process ID of the target interpreter process. In Linux, this is the parent of the remote control interpreter thread that runs the Annex that the connection is talking to. ARGV is the argument vector in the target Python interpereter.)rr )r) r r r �help_lc< s zMonitorConnection.help_lcc C s6 |rt |�| _td| j � | j�| jj| j � d S )NzFRemote connection %d. To return to Monitor, type <Ctrl-C> or .<RETURN>)r� r� rr r! �set_connectionr� )r) r� r r r �do_scO s zMonitorConnection.do_scc C s t d� d S )Nap sc <connection ID> ----- Set connection to communicate with a remote thread. With an argument, set current connection to the number specified. Without argument, use the same connection as last time. You will then be talking to a remote process via that connection. You can return to Monitor at any time by <Ctrl-C>. You may also use the '.' command (followed by <Return>), if the remote process is waiting for input. The '.' character may be followed by a monitor command, to execute it directly instead of returning to the monitor. For example, when talking to a connection, '.sc 1' will directly change to connection 1.)rr )r) r r r �help_scU s zMonitorConnection.help_scc C s dS )Nr r )r) r� r r r �do_qc s zMonitorConnection.do_qc C s t d� d S )Nz�q ----- Quit from the monitor. This will not exit from Python itself if called from an interactive interpreter. To make sure to exit from Python, use the exit command.)rr )r) r r r �help_qf s zMonitorConnection.help_qN) r r r � use_raw_inputr] r( ri ra r1 rm rk r� r� r r\ rj Zdo_hr� r� r� r� r� r� r� r� r� r� r� r� r r r r rX � s. rX c @ sP e Zd ZdZdd� Zdd� Zdd� Zdd � Zd d � Zd d � Z dd� Z dd� Z dS )�Monitorr c C s& t | � | _| _i | _d| _d | _d S )Nr )rX � connection�monitor_connectionr� �idsr0 )r) r r r r] q s zMonitor.__init__c C sF | j sd| _| j| j_n(tdgdd� t| j �� �D � �d | _| jS )Nr c S s g | ]}|�qS r r )�.0r� r r r � <listcomp>| s z!Monitor.newid.<locals>.<listcomp>)r� r� r� r� �maxr� �keys)r) r r r �newidw s (z Monitor.newidc C s* | � � }|| j|< ||_| �d| � d S )Nz *** Connection %s opened ***)r� r� � monitor_id� print_async)r) r� Zhidr r r rR s zMonitor.add_connectionc C s2 | j dk r.td| � tj�| j � tj�� dS )zS Print text only if we are waiting for input, and then restore the prompt. Nr* )r0 rr r; r< r= �flush)r) �textr r r r� � s zMonitor.print_asyncc C s4 | j |j= || jkr | �| j� | �d|j � d S )Nz *** Connection %s closed ***)r� r� r� r� r� r� )r) r� r r r r" � s zMonitor.remove_connectionc C sB �z*d}�x |�s(y�x�|s�| j }|j| _|| jk r@|jd dd� t|j�}d | _d }|�d�r�|dkrp| j| _ q�|dd � }| j}n| j }|ry|�|�}W n tk r� d}Y nX || jkr|rd}qW W q tk r� td� | j| _ d}Y q tk �r$ td� td� | j| _ w Y q X q W W d d | _| � � X d S )Nr r )r r �.z%*** End Of File - Exiting Monitor ***�KeyboardInterruptz1Use the ki command to interrupt a remote process.) r� r0 r� r1 �inputrE r9 rr r� r( )r) rn Zconnr r r r r �run� sH z Monitor.runc C sx x:t | j�� �D ](}y |�� W q tjk r6 Y qX qW y| j�� W n tjk r` Y nX | jjrtt� d� d S )Nr ) r� r� �valuesr( rU rV r� r^ �os�_exit)r) r� r r r r( � s z Monitor.closec C s || _ |j| _d S )N)r� r0 )r) r� r r r r� � s zMonitor.set_connectionN) r r r r� r] r� rR r� r" r� r( r� r r r r r� n s , r� c C s& ddl m} | �� t� }|�� dS )z�monitor() [0] Start an interactive remote monitor. This can be used to get information about the state, in particular the memory usage, of separately running Python processes. References [0] heapy_Use.html#heapykinds.Use.monitorr )�RemoteN)Z guppy.heapyr� �offr� r� )r� �mr r r r! � s r! �__main__)r )#r� �pprint�signalrU Z socketserverr; r_ r7 ru rK rS � ImportErrorr r Zguppy.heapy.RemoteConstantsZguppy.heapy.Consoler Z guppy.setsr Z guppy.etc.etcr Z guppy.etcr ZThreadingTCPServerr r r r% ZStreamRequestHandlerr r\ rX r� r! r r r r r �<module> s> _ ^d