OwlCyberSecurity - MANAGER
Edit File: backend_webagg.cpython-38.pyc
U ��]�+ � @ s> d Z ddlmZ ddlZddlmZ ddlZddlmZ ddl Z ddl Z ddl Z ddl Z ddl Z z ddlZW n ek r� ed��Y nX ddlZddlZddlZddlmZ ddlmZ dd lmZ d d lmZ d d lm Z G d d� de j �Z e � Z G dd� de j!�Z"G dd� dej#j$�Z%dd� Z&ej'G dd� de��Z(dS )z8 Displays Agg images in the browser, with interactivity � )�contextmanagerN)�BytesIO)�Pathz$The WebAgg backend requires Tornado.)�rcParams)�_Backend)�Gcf� )�backend_webagg_core�� TimerTornadoc @ s e Zd Zdd� ZdS )� ServerThreadc C s t jj�� �� d S �N)�tornado�ioloop�IOLoop�instance�start��self� r �D/usr/lib/python3/dist-packages/matplotlib/backends/backend_webagg.py�run* s zServerThread.runN)�__name__� __module__� __qualname__r r r r r r ) s r c @ s e Zd Zdd� Zdd� ZdS )�FigureCanvasWebAggc C s t � d S r )�showr r r r r 2 s zFigureCanvasWebAgg.showc O s t ||�S r r )r �args�kwargsr r r � new_timer7 s z FigureCanvasWebAgg.new_timerN)r r r r r r r r r r 1 s r c s� e Zd ZdZdZG dd� dejj�ZG dd� dejj�Z G dd� dejj�Z G dd � d ejj�Z G d d � d ejj�Z G d d � d ej j�Zd� fdd� Zeddd��Zedd� �Z� ZS )�WebAggApplicationFc @ s e Zd Zdd� ZdS )zWebAggApplication.FavIconc C s. | � dd� ttd dd�}| �|�� � d S )N� Content-Type� image/png�datapathZimageszmatplotlib.png)� set_headerr r �writeZ read_bytes)r Z image_pathr r r �getA s z WebAggApplication.FavIcon.getN�r r r r& r r r r �FavIcon@ s r( c s* e Zd Zdd�� fdd� Zdd� Z� ZS )z"WebAggApplication.SingleFigurePage� �� url_prefixc s || _ t� j||f|� d S r �r+ �super�__init__�r Z application�requestr+ r �� __class__r r r. G s z+WebAggApplication.SingleFigurePage.__init__c C sF t |�}t�|�}dj| j| jd�}| jd| j||tjj |j d� d S )N�ws://{req.host}{prefix}/�Zreq�prefixzsingle_figure.html)r5 �ws_uri�fig_id� toolitems�canvas) �intr �get_fig_manager�formatr0 r+ �render�core�NavigationToolbar2WebAggr8 r9 )r �fignum�managerr6 r r r r& K s ��z&WebAggApplication.SingleFigurePage.get�r r r r. r& � __classcell__r r r1 r �SingleFigurePageF s rD c s* e Zd Zdd�� fdd� Zdd� Z� ZS )z WebAggApplication.AllFiguresPager) r* c s || _ t� j||f|� d S r r, r/ r1 r r r. Z s z)WebAggApplication.AllFiguresPage.__init__c C s: dj | j| jd�}| jd| j|ttj�� �tj j d� d S )Nr3 r4 zall_figures.html)r5 r6 Zfiguresr8 ) r< r0 r+ r= �sortedr Zfigs�itemsr> r? r8 )r r6 r r r r& ^ s � �z$WebAggApplication.AllFiguresPage.getrB r r r1 r �AllFiguresPageY s rG c @ s e Zd Zdd� ZdS )zWebAggApplication.MplJsc C s$ | � dd� tj�� }| �|� d S )Nr! zapplication/javascript)r$ r> �FigureManagerWebAggZget_javascriptr% )r Z js_contentr r r r& i s zWebAggApplication.MplJs.getNr' r r r r �MplJsh s rI c @ s e Zd Zdd� ZdS )zWebAggApplication.Downloadc C sf t |�}t�|�}ddddddddd�}| �d |�|d �� t� }|jjj||d � | � |� � � d S ) Nzapplication/postscriptzapplication/pdfz image/svg+xmlr" z image/jpegz image/tiffzapplication/emf)ZpsZepsZpdfZsvgZpngZjpegZtifZemfr! �binary)r< ) r: r r; r$ r& r r9 �figureZsavefigr% �getvalue)r r@ ZfmtrA Z mimetypesZbuffr r r r& q s � z WebAggApplication.Download.getNr' r r r r �Downloadp s rM c @ s8 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd d � Zd S ) zWebAggApplication.WebSocketTc C s< t |�| _t�| j�| _| j�| � t| d�r8| �d� d S )N� set_nodelayT)r: r@ r r; rA Zadd_web_socket�hasattrrN )r r@ r r r �open� s z WebAggApplication.WebSocket.openc C s | j �| � d S r )rA Zremove_web_socketr r r r �on_close� s z$WebAggApplication.WebSocket.on_closec C sD t �|�}|d dkr"|d | _n t�| j�}|d k r@|�|� d S )N�type�supports_binary�value)�json�loadsrS r r; r@ Z handle_json)r �messagerA r r r � on_message� s z&WebAggApplication.WebSocket.on_messagec C s | � t�|�� d S r )� write_messagerU �dumps)r Zcontentr r r � send_json� s z%WebAggApplication.WebSocket.send_jsonc C s< | j r| j|dd� n"d�|�d��dd��}| �|� d S )NT)rJ zdata:image/png;base64,{0}�base64� r) )rS rY r< �encode�replace)r ZblobZdata_urir r r � send_binary� s �z'WebAggApplication.WebSocket.send_binaryN) r r r rS rP rQ rX r[ r` r r r r � WebSocket� s ra r) c s� |r$|d dkr |d dks$t d��t� j|d tjjdtj�� if|d | j f|d | j d |if|d | j d |if|d | j f|d | j f|d | jfgtj�� d� d S )Nr �/���z6url_prefix must start with a "/" and not end with one.z /_static/(.*)�pathz /favicon.icoz /([0-9]+)r+ z/?z /js/mpl.jsz /([0-9]+)/wsz /([0-9]+)/download.([a-z0-9.]+))Z template_path)�AssertionErrorr- r. r �webZStaticFileHandlerr> rH �get_static_file_pathr( rD rG rI ra rM )r r+ r1 r r r. � s0 � � � � ���zWebAggApplication.__init__Nc C s� | j r d S | |d�}|| _dd� }|d kr6td | _n|| _td | _|| jtd �D ]T}z|�|| j� W n2 tjk r� } z|jtj kr�� W 5 d }~X Y qVX || _ q�qVt d��d| _ d S ) Nr* c s sN t td|��D ]}| | V qt |d �D ] }| t�d| d| � V q*dS )z� Generate a list of n random ports near the given port. The first 5 ports will be sequential, and the remaining n-5 will be randomly selected in the range [port-2*n, port+2*n]. � ���� N)�range�min�randomZrandint)�port�n�ir r r � random_ports� s z2WebAggApplication.initialize.<locals>.random_portszwebagg.addressz webagg.portzwebagg.port_retrieszSThe webagg server could not be started because an available port could not be foundT) � initializedr+ r �addressrn Zlisten�socket�error�errnoZ EADDRINUSE� SystemExit)�clsr+ rn rs Zapprq �er r r � initialize� s* �z WebAggApplication.initializec sl � j r d S tjj�� �� �fdd��t��fdd��}d� _ td� tj� � |� � �� � W 5 Q R X d S )Nc s$ �� � td� tj�� d� _d S )NzServer is stoppedF)�stop�print�sys�stdout�flush�startedr )rx r r r �shutdown s z)WebAggApplication.start.<locals>.shutdownc 3 s8 t � t j� �fdd��} z d V W 5 t � t j| � X d S )Nc s � � ��S r )Zadd_callback_from_signal)Zsig�frame�r r� r r �<lambda> � z?WebAggApplication.start.<locals>.catch_sigint.<locals>.<lambda>)�signal�SIGINT)Z old_handlerr� r r � catch_sigint s � z-WebAggApplication.start.<locals>.catch_sigintTz"Press Ctrl+C to stop WebAgg server) r� r r r r r r| r} r~ r r )rx r� r )rx r r� r r � s zWebAggApplication.start)r) )r) NN)r r r rr r� r rf ZRequestHandlerr( rD rG rI rM Z websocketZWebSocketHandlerra r. � classmethodrz r rC r r r1 r r < s '# ,r c C sj dd l }t�� t�� s t�� | j}ttj � � d�� � }|j � |�}|jtj|tjj| jtjd��d�S )Nr zipython_inline_figure.html)r5 r7 r8 r9 rn zutf-8)Ztornado.templater rz �webagg_server_threadZis_aliver Znumberr r> rH rg Z read_text�templateZTemplateZgenerater+ r? r8 r9 rn �decode)rK r r@ Ztpl�tr r r �ipython_inline_display% s$ � ��r� c @ s. e Zd ZeZejZedd� �Z edd� �Z dS )�_BackendWebAggc C s | j �� d S r )r9 Z draw_idle)rA r r r �trigger_manager_draw= s z#_BackendWebAgg.trigger_manager_drawc C sT t �� djt jt jt jd�} td r:dd l}|�| � nt d�| �� t � � d S )Nz http://{address}:{port}{prefix})rs rn r5 zwebagg.open_in_browserr zTo view figure, visit {0}) r rz r< rs rn r+ r � webbrowserrP r| r )Zurlr� r r r r A s � z_BackendWebAgg.showN) r r r r Z FigureCanvasr> rH Z FigureManager� staticmethodr� r r r r r r� 8 s r� ))�__doc__� contextlibr rv �ior rU �pathlibr rm r} r� rt Z threadingr � ImportError� RuntimeErrorZ tornado.webZtornado.ioloopZtornado.websocketZ matplotlibr Zmatplotlib.backend_basesr Zmatplotlib._pylab_helpersr r) r r> r ZThreadr r� ZFigureCanvasWebAggCorer rf Z Applicationr r� Zexportr� r r r r �<module> s< j