Current File : //opt/alt/python37/lib/python3.7/site-packages/paste/__pycache__/progress.cpython-37.pyc
B
a�N� � @ s� d Z ddlZddlmZ dZdZdZdZdZG d d
� d
e �Z
G d d � d e �Z G d
d� de �Z d dgZ
dekr�ddlZejejd� dS )a~
Upload Progress Monitor
This is a WSGI middleware component which monitors the status of files
being uploaded. It includes a small query application which will return
a list of all files being uploaded by particular session/user.
>>> from paste.httpserver import serve
>>> from paste.urlmap import URLMap
>>> from paste.auth.basic import AuthBasicHandler
>>> from paste.debug.debugapp import SlowConsumer, SimpleApplication
>>> # from paste.progress import *
>>> realm = 'Test Realm'
>>> def authfunc(username, password):
... return username == password
>>> map = URLMap({})
>>> ups = UploadProgressMonitor(map, threshold=1024)
>>> map['/upload'] = SlowConsumer()
>>> map['/simple'] = SimpleApplication()
>>> map['/report'] = UploadProgressReporter(ups)
>>> serve(AuthBasicHandler(ups, realm, authfunc))
serving on...
.. note::
This is experimental, and will change in the future.
� N)� catch_errorsi i, zpaste.bytes_receivedzpaste.request_startedzpaste.request_finishedc @ s< e Zd ZdZdd� Zdd� Zddd�Zd d
� Zdd d
�Zd S )�
_ProgressFilezy
This is the input-file wrapper used to record the number of
``paste.bytes_received`` for the given request.
c C s( || _ || _|j| _|j| _|j| _d S )N)�_ProgressFile_environ�_ProgressFile_rfile�flush�write�
writelines)�self�environZrfile� r �?/opt/alt/python37/lib/python3.7/site-packages/paste/progress.py�__init__1 s
z_ProgressFile.__init__c s& | j � t| j��� �fdd�}t|�S )Nc 3 s, x&�D ] } � t t| �7 < | V qW d S )N)�ENVIRON_RECEIVED�len)�chunk)r
�riterr r �iterwrap; s
z(_ProgressFile.__iter__.<locals>.iterwrap)r �iterr )r r r )r
r r �__iter__8 s
z_ProgressFile.__iter__���c C s& | j �|�}| jt t|�7 < |S )N)r �readr r r )r �sizer r r r r A s z_ProgressFile.readc C s$ | j �� }| jt t|�7 < |S )N)r �readliner r r )r r r r r r F s
z_ProgressFile.readlineNc C s& | j �|�}| jt t|�7 < |S )N)r � readlinesr r r )r Zhintr r r r r K s z_ProgressFile.readlines)r )N) �__name__�
__module__� __qualname__�__doc__r
r r r r r r r r r + s
r c @ s* e Zd ZdZd dd�Zdd� Zdd� ZdS )
�UploadProgressMonitora<
monitors and reports on the status of uploads in progress
Parameters:
``application``
This is the next application in the WSGI stack.
``threshold``
This is the size in bytes that is needed for the
upload to be included in the monitor.
``timeout``
This is the amount of time (in seconds) that a upload
remains in the monitor after it has finished.
Methods:
``uploads()``
This returns a list of ``environ`` dict objects for each
upload being currently monitored, or finished but whose time
has not yet expired.
For each request ``environ`` that is monitored, there are several
variables that are stored:
``paste.bytes_received``
This is the total number of bytes received for the given
request; it can be compared with ``CONTENT_LENGTH`` to
build a percentage complete. This is an integer value.
``paste.request_started``
This is the time (in seconds) when the request was started
as obtained from ``time.time()``. One would want to format
this for presentation to the user, if necessary.
``paste.request_finished``
This is the time (in seconds) when the request was finished,
canceled, or otherwise disconnected. This is None while
the given upload is still in-progress.
TODO: turn monitor into a queue and purge queue of finished
requests that have passed the timeout period.
Nc C s$ || _ |p t| _|pt| _g | _d S )N)� application�DEFAULT_THRESHOLD� threshold�DEFAULT_TIMEOUT�timeout�monitor)r r r! r# r r r r
� s
z UploadProgressMonitor.__init__c s� � � dd�}|rxt|�| jkrx| j�� � d� t< t�� � t< d � t< t � � d �� d< d� fdd� }t
| j � |||�S | � � |�S )N�CONTENT_LENGTHr z
wsgi.inputc s t � � � t< d S )N)�time�REQUEST_FINISHED)�exc_info)r
r r � finalizer� s z1UploadProgressMonitor.__call__.<locals>.finalizer)N) �get�intr! r$ �appendr r& �REQUEST_STARTEDr' r r r )r r
�start_response�lengthr) r )r
r �__call__� s
z UploadProgressMonitor.__call__c C s | j S )N)r$ )r r r r �uploads� s z UploadProgressMonitor.uploads)NN)r r r r r
r0 r1 r r r r r P s 3
r c @ s0 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
S ) �UploadProgressReporterag
reports on the progress of uploads for a given user
This reporter returns a JSON file (for use in AJAX) listing the
uploads in progress for the given user. By default, this reporter
uses the ``REMOTE_USER`` environment to compare between the current
request and uploads in-progress. If they match, then a response
record is formed.
``match()``
This member function can be overriden to provide alternative
matching criteria. It takes two environments, the first
is the current request, the second is a current upload.
``report()``
This member function takes an environment and builds a
``dict`` that will be used to create a JSON mapping for
the given upload. By default, this just includes the
percent complete and the request url.
c C s
|| _ d S )N)r$ )r r$ r r r r
� s z UploadProgressReporter.__init__c C s |� dd �|� dd�kr dS dS )NZ REMOTE_USERr TF)r* )r Zsearch_environZupload_environr r r �match� s
z UploadProgressReporter.matchc C sd t �dt �|t ��d|�d�|t |�dd�|�dd�d�}|t }|r`t �dt �|��|d< |S ) Nz%Y-%m-%d %H:%M:%S� r% Z PATH_INFOZ QUERY_STRING)�started�finishedZcontent_lengthZbytes_receivedZ path_infoZ query_stringz%Y:%m:%d %H:%M:%Sr6 )r& �strftime�gmtimer- r* r r' )r r
�retvalr6 r r r �report� s
z UploadProgressReporter.reportc s� g }xz� �fdd��j �� D �D ]\}g }x>|�� D ]2\}}t|��dd��dd�}|�d||f � q2W |�dd �|� � q W d
d �|� }|d d d
t|�fg� |gS )Nc s" g | ]}�� � |�r��|��qS r )r3 r: )�.0�env)r
r r r �
<listcomp>� s z3UploadProgressReporter.__call__.<locals>.<listcomp>�\z\\�"z\"z%s: "%s"z{ %s }z, z[ %s ]z200 OK)z Content-Typez
text/plainzContent-Length)r$ r1 �items�str�replacer, �joinr )r r
r. �body�map�parts�k�vr )r
r r r0 � s z UploadProgressReporter.__call__N)r r r r r
r3 r: r0 r r r r r2 � s
r2 �__main__)Z optionflags)r r& Z
paste.wsgilibr r r" r r- r' �objectr r r2 �__all__r ZdoctestZtestmod�ELLIPSISr r r r �<module>! s %M=
Mini Shell