Format: application/x-python-code
1 o
2
3 C'hs& � @ s d dl mZ d dlZd dlZd dlZd dlZd dlZd dlmZ ddl m
4 Z
5 m Z m Z m
6 Z
7 mZmZ ddlmZmZmZ ddd �Ze�d
8 d � ed �D �� e�d
9 �dd� e�� D ���Zd%dd�Zd&dd�Z d'd d �Z G d d � d �Z G d!d"� d"�Z G d#d$� d$ee
10 �Z dS )(� )� annotationsN)�Path� )�AsyncByteStream� FileContent� FileTypes� RequestData� RequestFiles�SyncByteStream)�peek_filelike_length�primitive_value_to_str�to_bytesz%22z\\)�"�\c C s$ i | ]}|d krt |�d�|��qS )� z%{:02X})�chr�format��.0�c� r �FC:\Users\shubh\Videos\agent\nova\Lib\site-packages\httpx/_multipart.py�
11 <dictcomp> s $ r � �|c C s g | ]}t �|��qS r )�re�escaper r r r �
12 <listcomp> s r �name�str�value�return�bytesc C s* d
13 dd�}t �||�}| � d|� d��� S ) z;
14 Encode a name/value pair within a multipart form.
15 �match�typing.Match[str]r! r c S s t | �d� S )Nr )�!_HTML5_FORM_ENCODING_REPLACEMENTS�group)r# r r r �replacer& s z$_format_form_param.<locals>.replacerz="r N)r# r$ r! r )�_HTML5_FORM_ENCODING_RE�sub�encode)r r r' r r r �_format_form_param! s
16 r+ �filename�
17 str | Nonec C s | r t �| �d p
18 dS dS )z�
19 Guesses the mimetype based on a filename. Defaults to `application/octet-stream`.
20
21 Returns `None` if `filename` is `None` or empty.
22 r zapplication/octet-streamN)� mimetypes�
23 guess_type)r, r r r �_guess_content_type- s r0 � content_type� bytes | Nonec C s^ | r| � d�s d S d| v r-| �d�D ]}|�� �� � d�r,|�� td�d � �d� S qd S )Ns multipart/form-data� ;s boundary=� ")�
24 startswith�split�strip�lower�len)r1 �sectionr r r �(get_multipart_boundary_from_content_type8 s �r; c @ sB e Zd ZdZddd �Zdd d �Zdd
25 d�Zddd�Zddd�ZdS )� DataFieldzB
26 A single form field item, within a multipart form field.
27 r r r � str | bytes | int | float | Noner! �Nonec C s| t |t�stdt|�� d|����|d ur*t |ttttf�s*tdt|�� d|����|| _t |t�r7|| _ d S t|�| _ d S )Nz)Invalid type for name. Expected str, got �: z5Invalid type for value. Expected primitive type, got )
28 �
29 isinstancer � TypeError�typer" �int�floatr r r )�selfr r r r r �__init__K s"
30 ���� ��zDataField.__init__r" c C s. t | d�std| j�}d�d|dg�| _| jS )N�_headersr � � Content-Disposition: form-data; �
31
32 )�hasattrr+ r �joinrG )rE r r r r �render_headersZ s
33 �zDataField.render_headersc C s t | d�s t| j�| _| jS )N�_data)rK r
34 r rN �rE r r r � render_datac s
35 zDataField.render_datarC c C s | � � }| �� }t|�t|� S �N)rM rP r9 )rE �headers�datar r r �
36 get_lengthi s zDataField.get_length�typing.Iterator[bytes]c c s � | � � V | �� V d S rQ �rM rP rO r r r �rendern s �
37 zDataField.renderN)r r r r= r! r> �r! r" )r! rC �r! rU ) �__name__�
38 __module__� __qualname__�__doc__rF rM rP rT rW r r r r r< F s
39
40
41
42 r< c @ sF e Zd ZdZdZdd d
43 �Zdd d
44 �Zddd�Zddd�Zddd�Z dS )� FileFieldzB
45 A single file field item, within a multipart form field.
46 i r r r r r! r> c C s� || _ i }d }t|t�r*t|�dkr|\}}n t|�dkr#|\}}}n|\}}}}n
47 ttt|dd���j }|}|d u r?t|�}tdd� |D ��}|d urR|sR||d< t|t j
48 �r\t d��t|t j �rft d ��|| _
49 || _|| _d S )
50 N� � r Zuploadc s s � | ] }d |� � v V qdS )z content-typeN)r8 )r �keyr r r � <genexpr>� s � z%FileField.__init__.<locals>.<genexpr>� Content-Typez?Multipart file uploads require 'io.BytesIO', not 'io.StringIO'.zDMultipart file uploads must be opened in binary mode, not text mode.)r r@ �tupler9 r r �getattrr0 �any�io�StringIOrA �
51 TextIOBaser, �filerR )rE r r rR r1 r, �fileobjZhas_content_type_headerr r r rF z s6
52
53 � �
54 zFileField.__init__�
55 int | Nonec C sP | � � }t| jttf�rt|�tt| j�� S t| j�}|d u r"d S t|�| S rQ )rM r@ rj r r" r9 r
56 r )rE rR Z file_lengthr r r rT � s
57 zFileField.get_lengthr" c C s� t | d�sFdtd| j�g}| jr td| j�}|�d|g� | j�� D ]\}}d|� d��� |�� }}|�||g� q"|�d� d � |�| _
58 | j
59 S )
60 NrG rI r r, s ; z
61 r? rJ rH ) rK r+ r r, �extendrR �itemsr* �appendrL rG )rE �partsr, � header_name� header_valuera �valr r r rM � s
62
63 �
64 zFileField.render_headersrU c c s� � t | jttf�rt| j�V d S t| jd�r*z| j�d� W n
65 tjy) Y nw | j� | j
66 �}|rCt|�V | j� | j
67 �}|s3d S d S )N�seekr ) r@ rj r r" r
68 rK rt rg �UnsupportedOperation�read�
69 CHUNK_SIZE�rE �chunkr r r rP � s � �
70 �zFileField.render_datac c s � | � � V | �� E d H d S rQ rV rO r r r rW � s �
71 zFileField.renderN)r r r r r! r> �r! rl rX rY )
72 rZ r[ r\ r] rw rF rT rM rP rW r r r r r^ s s
73
74 1
75
76 r^ c @ sZ e Zd ZdZ d d d d �Zd dd�Zd!dd�Zd"dd�Zd#dd�Zd!dd�Z d$d d �Z
77 dS )%�MultipartStreamzC
78 Request content as streaming multipart encoded form data.
79 NrS r �filesr �boundaryr2 r! r> c C sH |d u rt �d��� �d�}|| _d|�d� | _t| �||��| _ d S )N� �asciiz multipart/form-data; boundary=%s)
80 �os�urandom�hexr* r} �decoder1 �list� _iter_fields�fields)rE rS r| r} r r r rF � s �zMultipartStream.__init__�&typing.Iterator[FileField | DataField]c c s� � |� � D ] \}}t|ttf�r |D ] }t||d�V qqt||d�V qt|tj�r/|� � n|}|D ] \}}t||d�V q3d S )N)r r )rn r@ rd r� r< �typing�Mappingr^ )rE rS r| r r �itemZ
81 file_itemsr r r r� � s �� �z MultipartStream._iter_fieldsrU c c s> � | j D ]}d| j V |�� E d H dV qd| j V d S )Ns --%s
82 s
83 s --%s--
84 )r� r} rW )rE �fieldr r r � iter_chunks s �
85 zMultipartStream.iter_chunksrl c C sd t | j�}d}| jD ] }|�� }|du r dS |d| d 7 }||7 }|d7 }q
86 |d| d 7 }|S )z�
87 Return the length of the multipart encoded content, or `None` if
88 any of the files have a length that cannot be determined upfront.
89 r Nr_ � )r9 r} r� rT )rE Zboundary_length�lengthr� Z field_lengthr r r �get_content_length s
90
91
92 z"MultipartStream.get_content_length�dict[str, str]c C s. | � � }| j}|d u rd|d�S t|�|d�S )N�chunked)zTransfer-Encodingrc )zContent-Lengthrc )r� r1 r )rE �content_lengthr1 r r r � get_headers s
93
94 zMultipartStream.get_headersc c s � | � � D ]}|V qd S rQ �r� rx r r r �__iter__& � � �zMultipartStream.__iter__�typing.AsyncIterator[bytes]c C s �| � � D ]}|V qd S rQ r� rx r r r � __aiter__* r� zMultipartStream.__aiter__rQ )rS r r| r r} r2 r! r> )rS r r| r r! r� rY rz )r! r� )r! r� ) rZ r[ r\ r] rF r� r� r� r� r� r� r r r r r{ � s �
95
96
97
98
99 r{ )r r r r r! r" )r, r- r! r- )r1 r2 r! r2 )!�
100 __future__r rg r. r� r r� �pathlibr �_typesr r r r r r
101 �_utilsr r r
102 r% �update�range�compilerL �keysr( r+ r0 r; r< r^ r{ r r r r �<module> s,
103 ��
104
105
106 -m