U
    f!                     @   s   d dl mZ d dlmZ d dlmZ d dlmZ d dlm	Z	m
Z
mZ d dlmZ d dlmZ d dlmZ d d	lmZ ed
ZG dd deZG dd de	ZG dd dejZdd Zdd Zdd Zdd ZdS )    )IOBase)settings)signals)base)HttpRequest	QueryDictparse_cookie)set_script_prefix)repercent_broken_unicode)cached_property)_lazy_re_compiles   /+c                   @   s,   e Zd ZdZdd Zd
ddZdddZd	S )LimitedStreamz
    Wrap another stream to disallow reading it past a number of bytes.

    Based on the implementation from werkzeug.wsgi.LimitedStream
    See https://github.com/pallets/werkzeug/blob/dbf78f67/src/werkzeug/wsgi.py#L828
    c                 C   s    |j | _|j| _d| _|| _d S )Nr   )read_readreadline	_readline_poslimit)selfstreamr    r   =/tmp/pip-unpacked-wheel-_jot26k8/django/core/handlers/wsgi.py__init__   s    zLimitedStream.__init__c                C   s`   | j }| j}||krdS |dks(|d kr2|| }nt||| }| |}|  j t|7  _ |S N    r   )r   r   minr   len)r   sizer   r   datar   r   r   r      s    

zLimitedStream.readc                C   s`   | j }| j}||krdS |dks(|d kr2|| }nt||| }| |}|  j t|7  _ |S r   )r   r   r   r   r   )r   r   r   r   liner   r   r   r   *   s    

zLimitedStream.readlineN)r   )r   )__name__
__module____qualname____doc__r   r   r   r   r   r   r   r      s   
r   c                       s   e Zd ZejedgB ZeddgZdd Z fddZdd	 Z	e
d
d Zdd Zdd Ze
dd Zedd ZeeeZ  ZS )WSGIRequestenvironzwsgi.errors
wsgi.inputc              	   C   s   t |}t|pd}|| _|| _d|d|dddf | _|| _|| jd< || jd< |d  | _	| 
| zt|d}W n ttfk
r   d	}Y nX t| jd
 || _d| _d | _d S )N/z%s/%s    	PATH_INFOSCRIPT_NAMEREQUEST_METHODCONTENT_LENGTHr   r'   F)get_script_nameget_path_infor&   	path_inforstripreplacepathMETAuppermethodZ_set_content_type_paramsintget
ValueError	TypeErrorr   _streamZ_read_startedZresolver_match)r   r&   script_namer1   content_lengthr   r   r   r   <   s"    



zWSGIRequest.__init__c                    s0   t   }| jD ]}||d kr|d |= q|S )Nr5   )super__getstate__meta_non_picklable_attrs)r   stateattr	__class__r   r   r@   U   s
    

zWSGIRequest.__getstate__c                 C   s   | j dS )Nzwsgi.url_scheme)r&   r9   r   r   r   r   _get_scheme\   s    zWSGIRequest._get_schemec                 C   s   t | jdd}t|| jdS )NQUERY_STRINGr)   )encoding)get_bytes_from_wsgir&   r   	_encoding)r   Zraw_query_stringr   r   r   GET_   s    zWSGIRequest.GETc                 C   s   t | ds|   | jS )N_post)hasattr_load_post_and_filesrM   rF   r   r   r   	_get_poste   s    
zWSGIRequest._get_postc                 C   s
   || _ d S N)rM   )r   postr   r   r   	_set_postj   s    zWSGIRequest._set_postc                 C   s   t | jdd}t|S )NZHTTP_COOKIEr)   )get_str_from_wsgir&   r   )r   Z
raw_cookier   r   r   COOKIESm   s    zWSGIRequest.COOKIESc                 C   s   t | ds|   | jS )N_files)rN   rO   rV   rF   r   r   r   FILESr   s    
zWSGIRequest.FILES)r!   r"   r#   r   Znon_picklable_attrs	frozensetrA   r   r@   rG   r   rL   rP   rS   rU   propertyrW   POST__classcell__r   r   rD   r   r%   8   s   


r%   c                       s(   e Zd ZeZ fddZdd Z  ZS )WSGIHandlerc                    s   t  j|| |   d S rQ   )r?   r   Zload_middleware)r   argskwargsrD   r   r   r   ~   s    zWSGIHandler.__init__c                 C   s   t t| tjj| j|d | |}| |}| j|_d|j	|j
f }| dd |j D }||| t|dd d k	r|dr|j|j_|d |j|j}|S )N)Zsenderr&   z%d %sc                 s   s   | ]}d |j ddfV  qdS )z
Set-Cookier)   )headerN)output).0cr   r   r   	<genexpr>   s     z'WSGIHandler.__call__.<locals>.<genexpr>file_to_streamzwsgi.file_wrapper)r	   r/   r   Zrequest_startedsendrE   request_classZget_responseZ_handler_classstatus_codeZreason_phraseitemscookiesvaluesgetattrr9   closerd   
block_size)r   r&   Zstart_responserequestresponsestatusresponse_headersr   r   r   __call__   s&    



 zWSGIHandler.__call__)r!   r"   r#   r%   rf   r   rr   r[   r   r   rD   r   r\   {   s   r\   c                 C   s   t | dd}t| S )z0Return the HTTP request's PATH_INFO as a string.r+   r(   )rJ   r
   decode)r&   r1   r   r   r   r0      s    r0   c                 C   s|   t jdk	rt jS t| ddp&t| dd}|rhd|kr@td|}t| dd}|rb|dt|  n|}nt| dd}| S )	aE  
    Return the equivalent of the HTTP request's SCRIPT_NAME environment
    variable. If Apache mod_rewrite is used, return what would have been
    the script name prior to any rewriting (so it's the script name as seen
    from the client's perspective), unless the FORCE_SCRIPT_NAME setting is
    set (to anything).
    NZ
SCRIPT_URLr)   ZREDIRECT_URLs   //   /r+   r,   )r   ZFORCE_SCRIPT_NAMErJ   _slashes_resubr   rs   )r&   Z
script_urlr1   r=   r   r   r   r/      s    
  r/   c                 C   s   |  ||}|dS )zh
    Get a value from the WSGI environ dictionary as bytes.

    key and default should be strings.
    z
iso-8859-1)r9   encoder&   keydefaultvaluer   r   r   rJ      s    rJ   c                 C   s   t | ||}|jddS )zj
    Get a value from the WSGI environ dictionary as str.

    key and default should be str objects.
    r3   )errors)rJ   rs   rx   r   r   r   rT      s    rT   N)ior   Zdjango.confr   Zdjango.corer   Zdjango.core.handlersr   Zdjango.httpr   r   r   Zdjango.urlsr	   Zdjango.utils.encodingr
   Zdjango.utils.functionalr   Zdjango.utils.regex_helperr   ru   r   r%   BaseHandlerr\   r0   r/   rJ   rT   r   r   r   r   <module>   s    )C"!