U
|
Þ=®dO+ ã @ sp d dl mZ d dlZd dlZd dlmZ d dlmZ d dlm Z ddl
|
m Z ddl m Z dd lmZ G d
|
d d Zed d G dd deZed d G dd deZed d G dd deZed d G dd deZed d G dd deZG dd deZe ZG dd de Zd Ze d ej¡Z e eej¡Z e d e ej¡Z d Z G d d! d!Z G d"d# d#Z!dS )$é )Ú annotationsN)Ú dataclass)Úauto)ÚEnumé )ÚHeaders)ÚRequestEntityTooLarge)Úparse_options_headerc @ s e Zd ZdS )ÚEventN©Ú__name__Ú
|
__module__Ú __qualname__© r r úPd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\werkzeug/sansio/multipart.pyr
|
s r
|
T)Úfrozenc @ s e Zd ZU ded< dS )ÚPreambleÚbytesÚdataN©r r r Ú__annotations__r r r r r s
|
r c @ s e Zd ZU ded< ded< dS )ÚFieldÚstrÚnamer ÚheadersNr r r r r r s
|
r c @ s& e Zd ZU ded< ded< ded< dS )ÚFiler r Úfilenamer r Nr r r r r r s
|
r c @ s e Zd ZU ded< ded< dS )ÚDatar r ÚboolÚ more_dataNr r r r r r $ s
|
r c @ s e Zd ZU ded< dS )ÚEpiloguer r Nr r r r r r * s
|
r c @ s e Zd ZdS )ÚNeedDataNr r r r r r! / s r! c @ s0 e Zd Ze Ze Ze Ze Ze Ze Z dS )ÚStateN)
|
r r r r ÚPREAMBLEÚPARTÚDATAÚ
|
DATA_STARTÚEPILOGUEÚCOMPLETEr r r r r" 6 s r" s
|
(?:
|
|
|
| )s (?:
|
|
| |
|
|
)s %s[ ]é c @ s| e Zd ZdZd ddddddddd Zdd
|
d d d Zddd ddZddddZddd ddZdddddd Z dS ) ÚMultipartDecoderz³Decodes a multipart message as bytes into Python events.
|
|
The part data is returned as available to allow the caller to save
|
the data from memory to disk, if desired.
|
N)Ú max_partsr z
|
int | NoneÚNone)ÚboundaryÚmax_form_memory_sizer+ Úreturnc C s| t | _d| _|| _|| _tj| _|| _t
|
dt t |¡t t f t j ¡| _t
|
dt t |¡t t f t j ¡| _d| _d| _d S )NFs% %s?--%s(--[^\S\n\r]*%s?|[^\S\n\r]*%s)s$ %s--%s(--[^\S\n\r]*%s?|[^\S\n\r]*%s)r )Ú bytearrayÚbufferÚcompleter. r+ r" r# Ústater- ÚreÚcompileÚ
|
LINE_BREAKÚescapeÚ MULTILINEÚ preamble_reÚ boundary_reÚ_search_positionÚ_parts_decoded)Úselfr- r. r+ r r r Ú__init__T s( ÿý ÿýzMultipartDecoder.__init__Úint)r r/ c C sb z| d¡}W n tk
|
r* t|}Y nX z| d¡}W n tk
|
rV t|}Y nX t||S )Nó
|
ó )ÚrindexÚ
|
ValueErrorÚlenÚmin)r= r Zlast_nlZlast_crr r r Ú last_newline| s z MultipartDecoder.last_newlinez bytes | Nonec C sJ |d krd| _ n6| jd k r:t| jt| | jkr:t n | j |¡ d S )NT)r2 r. rD r1 r Úextend)r= r r r r Ú receive_data s ÿþz MultipartDecoder.receive_datar
|
)r/ c C sº t }| jtjkr¨| j | j| j¡}|d k r| d¡ d¡rDtj
|
| _ntj | _t | jd | ¡
}| jd | ¡
= t|d}d| _n tdt| jt| j t | _nê| jtj krÂt | j| j¡}|d k rª| | jd | ¡
¡}| ¡ | ¡ d }| jd |
= d|krtdt|d \}}t t| d¡¡}| d ¡} | d k r`t | ||d
|
}n t ||d }tj | _d| _| j d7 _ | j d k rÀ| j | j krÀt! ntdt| jt | _nÐ| jtj kr| j"| jd d \}}
|
} | jd |
|
= t#|| d}| rtj$| _n| jtj$krX| j"| jdd \}}
|
} | jd |
|
= |sJ| st#|| d}n:| jtj
|
kr| j%rt&t | jd}| jd d
= tj'| _| j%r¶t(|t)r¶td| j |S )Né ó --)r r r úcontent-dispositionz"Missing Content-Disposition headerr r )r r r )r r T)Ústart)r r Fz&Invalid form-data cannot parse beyond )*Ú NEED_DATAr3 r" r# r9 Úsearchr1 r; ÚgroupÚ
|
startswithr' r$ r rL Úendr ÚmaxrD r- ÚSEARCH_EXTRA_LENGTHÚ BLANK_LINE_REÚ_parse_headersrC r ÚtÚcastr Úgetr r r& r< r+ r Ú _parse_datar r% r2 r r( Ú
|
isinstancer! ) r= ÚeventÚmatchr r Z headers_endÚ dispositionÚextrar r Ú del_indexr r r r Ú
|
next_event s|
|
|
ÿ
|
|
|
ÿ
|
|
ýþ
|
zMultipartDecoder.next_eventr c C s` g }t d|¡}| ¡ D ]>}| ¡ }|dkr| ¡ d¡\}}}| | ¡ | ¡ f¡ qt|S )Nó ó ú:)ÚHEADER_CONTINUATION_REÚsubÚ
|
splitlinesÚstripÚdecodeÚ partitionÚappendr )r= r r Úliner Ú_Úvaluer r r rU é s z MultipartDecoder._parse_headersr ztuple[bytes, int, bool])r rL r/ c C sØ |r&t |¡}t tjt |¡ ¡ }nd}| j d| j ¡dkr\|
|
||d
¡ }}d}nf| j |¡}|d k r¤| d¡ d¡rtj| _ntj| _| ¡ }| ¡ }n|
|
||d
¡ }}|d k}t|||
||fS )Nr rJ éÿÿÿÿTrI )Ú LINE_BREAK_REr\ rV rW ÚMatchr rQ r1 Úfindr- rF r: rN rO rP r" r' r3 r$ rL )r= r rL r\ Z
|
data_startZdata_endr_ r r r r rY ö s"
|
|
|
z MultipartDecoder._parse_data)N)
|
r r r Ú__doc__r> rF rH r` rU rY r r r r r* M s ýû( V r* c @ s, e Zd ZdddddZddddd Zd
|
S ) ÚMultipartEncoderr r, )r- r/ c C s || _ tj| _d S )N)r- r" r# r3 )r= r- r r r r> s zMultipartEncoder.__init__r
|
)r[ r/ c C sz t |tr$| jtjkr$tj| _|jS t |ttfrÔ| jtjtjtj hkrÔd| j
|
d }|d|j ¡ 7 }t |tr|d|j ¡ 7 }|d7 }t t|¡jD ],\}}| ¡ dkr|| d| d ¡ 7 }qtj| _|S t |tr| jtjkrtj | _t|jdkrd|j S |jS n^t |tr8| jtj kr8|jS t |tr`tj| _d| j
|
d |j S td
|
| d | j d S ) Ns
|
--s
|
s) Content-Disposition: form-data; name="%s"s ; filename="%s"rK z: z
|
r s --
|
zCannot generate z in state: )rZ r r3 r" r# r$ r r r r% r- r Úencoder rV rW r Úlowerr& r rD r r( rC )r= r[ r r rm r r r Ú
|
send_event s: ý
|
|
zMultipartEncoder.send_eventN)r r r r> rv r r r r rs s rs )"Ú
|
__future__r r4 ÚtypingrV Z dataclassesr Úenumr r Zdatastructuresr Ú
|
exceptionsr Úhttpr r
|
r r r r r r! rM r" r6 r5 r8 rT ro rd rS r* rs r r r r Ú<module> s< I
|
