U
    diZ                     @   s  U d dl Zd dlZd dlZd dlmZmZmZmZm	Z	m
Z
mZmZmZmZ d dlmZ d dlmZmZmZmZmZmZmZ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%m&Z&m'Z' d d
l(m)Z) d dl*m+Z+m,Z, d dl-m.Z. d dl/m0Z0 d dl1m2Z2m3Z3m4Z4 d dl5m6Z6 d dl7m8Z8 d dl9m:Z: d dl;m<Z< dddddddiddigiddddddddd d!dgd"Z=d#dd$d%dd&e&d idid'Z>d(d)d*d+d,d-d.Z?ee@e@f eAd/< eeee@ef eee@ef  f d0d1d2ZBd3d4eee0eeee<d5 f ef eCeee@ef  d6d7d8ZDd3d4ee ee0eeee<d5 f ef eCeee@ef  d9d:d;ZEejFe@e@d<d=d>ZGejFe@e@d<d?d@ZHejFe@e
e@ ee@ef dAdBdCZId3d4ejFe
e@ ee0eeee<d5 f ef eCeee@ef ee@ef ee@ef f dDdEdFZJe	e8 ee dGdHdIZKdJddddddddd3dK
e@e@e@ee@ ee@ e	e8 ee	e8  eeee@ef   eeee@ee@ef f   ee@ eee@ee@ef f  eee@ee@ef f  eCee@ef dLdMdNZLdS )O    N)
AnyDictListOptionalSequenceSetTupleTypeUnioncast)routing)GenerateJsonSchemaJsonSchemaValue
ModelField	Undefinedget_compat_model_name_mapget_definitionsget_schema_from_model_fieldlenient_issubclass)DefaultPlaceholder)	Dependant)_get_flat_fields_from_paramsget_flat_dependantget_flat_params)jsonable_encoder)METHODS_WITH_BODY
REF_PREFIXREF_TEMPLATE)OpenAPI)Body
ParamTypes)Response)ModelNameMap)deep_dict_updategenerate_operation_id_for_pathis_body_allowed_for_status_code)JSONResponse)	BaseRoute)HTTP_422_UNPROCESSABLE_ENTITY)LiteralValidationErrorobjectZLocationarrayZanyOftypestringinteger)titler-   itemsMessage)r0   r-   z
Error Type)locmsgr-   r3   r4   )r0   r-   
propertiesrequiredHTTPValidationErrorZdetailZDetail$ref)r0   r-   r5   ZInformationZSuccessZRedirectionzClient ErrorzServer ErrorzDefault Response)Z1XXZ2XXZ3XX4XXZ5XXDEFAULTstatus_code_ranges)flat_dependantreturnc                 C   sN   i }g }| j D ]6}t|jjddd}|jj}|||< |||ji q||fS )NTZby_aliasZexclude_none)Zsecurity_requirementsr   Zsecurity_schememodelZscheme_nameappendZscopes)r<   security_definitionsoperation_securityZsecurity_requirementZsecurity_definitionZsecurity_name rC   9/tmp/pip-unpacked-wheel-t6atx7xj/fastapi/openapi/utils.py get_openapi_security_definitionsM   s    
rE   T)separate_input_output_schemas)Z
validationZserialization)	dependantschema_generatormodel_name_mapfield_mappingrF   r=   c              	   C   s   g }t | dd}t|j}t|j}t|j}	t|j}
tj|ftj|ftj	|	ftj
|
fg}|D ]\}}|D ]}|j}t|ddsqlt|||||d}|j|j|j|d}|jr|j|d< t|dd }t|dd }|rt||d	< n|tkrt||d< t|d
d rd|d
< || qlq`|S )NTZskip_repeatsinclude_in_schemafieldrH   rI   rJ   rF   )nameinr6   schemadescriptionopenapi_examplesexampleexamples
deprecated)r   r   path_paramsquery_paramsheader_paramscookie_paramsr    pathqueryheadercookie
field_infogetattrr   aliasvaluer6   rR   r   r   r@   )rG   rH   rI   rJ   rF   
parametersr<   rW   rX   rY   rZ   Zparameter_groupsZ
param_typeZparam_groupparamr_   Zparam_schemaZ	parameterrS   rT   rC   rC   rD   !_get_openapi_operation_parameters^   sP    





re   )
body_fieldrH   rI   rJ   rF   r=   c                 C   s   | sd S t | tstt| ||||d}tt| j}|j}| j}i }	|rP||	d< d|i}
|j	rnt
|j	|
d< n|jtkrt
|j|
d< ||
i|	d< |	S )NrM   r6   rQ   rU   rT   content)
isinstancer   AssertionErrorr   r   r   r_   
media_typer6   rS   r   rT   r   )rf   rH   rI   rJ   rF   Zbody_schemar_   Zrequest_media_typer6   request_body_oaiZrequest_media_contentrC   rC   rD   "get_openapi_operation_request_body   s2    


rl   )routemethodr=   c                 C   s2   t jdtdd | jr| jS | j}t| j||dS )Nzqfastapi.openapi.utils.generate_operation_id() was deprecated, it is not used internally, and will be removed soon   
stacklevel)rO   r[   rn   )warningswarnDeprecationWarningoperation_idpath_formatr$   rO   )rm   rn   r[   rC   rC   rD   generate_operation_id   s    rw   c                 C   s   | j r| j S | jdd S )N_ )summaryrO   replacer0   rm   rn   rC   rC   rD   generate_operation_summary   s    r}   )rm   rn   operation_idsr=   c                 C   s   i }| j r| j |d< t| |d|d< | jr4| j|d< | jp>| j}||krd| d| jj  }t| jdi d}|r|d	| 7 }t	j
|d
d || ||d< | jr| j|d< |S )Ntagsr|   rz   rR   zDuplicate Operation ID z for function __globals____file__z at    rp   ZoperationIdrV   )r   r}   rR   ru   Z	unique_idZendpoint__name__r`   getrr   rs   addrV   )rm   rn   r~   	operationru   message	file_namerC   rC   rD   get_openapi_operation_metadata   s*    




r   )rm   r~   rH   rI   rJ   rF   r=   c           +   
      s@  i }i }i }| j d k	stdt| jtr4| jj}	n| j}	|	sFtd|	j}
| jr6| j D ]}t| ||d g }t	| j
dd}t|d\}}|r dg | |r|| t| j
||||d}|| |rd	d
 |D }dd
 |D }|| t|  d< |tkr8t| j||||d}|r8| d< | jri }| jD ]>}t|tjrJt||||||d\}}}|j|i||j< qJ| d< | jd k	rt| j}n:t|	j}|j !d}|d k	rt|j"t#rt|j"}| j$ di |i d< |
rnt%| jrnddi}t&|	t'rF| j(rBt)| j(||||d}ni }| di |i di |
i d< | j*r di }| j*+ D ]
\}}|, } | -dd  t|. }!|!dkrd}!||!i }"t| t/std| j0!|}#d }$|#r<t)|#||||d}$|
pd}%| di |%i di }&t1|&|$ t2!t|. p`t3j4j*!t#|}'| !dp|"!dp|'pd}(t1|"|  |(|"d< qtt5})t6| j
}*|*s| jrt7 fdd |)d!dfD sd"ddd#t8d$ iiid% d |)< d&|kr|t9t:d' | j;r(t1 | j;  ||< < qZ|||fS )(NzMethods must be a listz.A response class is needed to generate OpenAPI)rm   rn   r~   TrK   )r<   security)rG   rH   rI   rJ   rF   c                 S   s   i | ]}|d  |d f|qS )rP   rO   rC   .0rd   rC   rC   rD   
<dictcomp>  s     z$get_openapi_path.<locals>.<dictcomp>c                 S   s(   i | ] }| d r|d |d f|qS )r6   rP   rO   )r   r   rC   rC   rD   r     s   
 rc   )rf   rH   rI   rJ   rF   ZrequestBodyrm   r~   rH   rI   rJ   rF   	callbacksstatus_code	responsesrR   r-   r.   rM   rg   rQ   r?   r:   defaultz%An additional response must be a dictzapplication/jsonzAdditional Responsec                 3   s   | ]}| d  kV  qdS )r   NrC   )r   statusr   rC   rD   	<genexpr>  s   z#get_openapi_path.<locals>.<genexpr>r9   zValidation Errorr8   r7   )rR   rg   r*   )r*   r7   )=methodsri   rh   response_classr   rb   rj   rL   r   r   rG   rE   
setdefaultextendupdatere   listvaluesr   rl   rf   r   r   APIRouteget_openapi_pathr[   rO   r   strinspect	signature__init__rc   r   r   intZresponse_descriptionr%   r   r&   response_fieldr   r   r1   copypopupperdictresponse_fieldsr#   r;   httpclientr(   r   anyr   validation_error_definition$validation_error_response_definitionZopenapi_extralower)+rm   r~   rH   rI   rJ   rF   r[   security_schemesdefinitionsZcurrent_response_classZroute_response_media_typern   rc   r<   rA   rB   Zoperation_parametersZall_parametersZrequired_parametersrk   r   callbackZcb_pathZcb_security_schemesZcb_definitionsr   Zresponse_signatureZstatus_code_paramZresponse_schemaZoperation_responsesZadditional_status_codeZadditional_responseprocess_responseZstatus_code_keyZopenapi_responserN   Zadditional_field_schemarj   Zadditional_schemaZstatus_textrR   Zhttp422Zall_route_paramsrC   r   rD   r      sr   
  





	

   
  
  




 

r   )routesr=   c                 C   s   g }g }g }g }| D ]}t |dd rt|tjr|jrVt|jtsJtd||j |jrh||j |j	r~|
|j	  |jr|
t|j t|j}|
| q|t|| |  }|S )NrL   z'A request body must be a Pydantic Field)r`   rh   r   r   rf   r   ri   r@   r   r   r   r   r   get_fields_from_routesr   rG   r   )r   Zbody_fields_from_routesZresponses_from_routesZrequest_fields_from_routesZcallback_flat_modelsrm   paramsZflat_modelsrC   rC   rD   r     s:      

r   z3.1.0)
openapi_versionrz   rR   webhooksr   serversterms_of_servicecontactlicense_inforF   )r0   versionr   rz   rR   r   r   r   r   r   r   r   rF   r=   c              	      s  | |d}|r||d< |r"||d< |	r.|	|d< |
r:|
|d< |rF||d< ||d}|r\||d< i }i }i }t  }tt|pxg t|pg  }t|}ttd	}t||||d
\} |pg D ]r}t|tj	rt
||||||d}|r|\}}}|r||ji | |r|di | |r | q|p4g D ]z}t|tj	r6t
||||||d}|r6|\}}}|r||ji | |r|di | |r6 | q6 r҇ fddt D |d< |r||d< ||d< |r||d< |r||d< ttf |dddS )N)r0   r   rz   rR   ZtermsOfServicer   license)Zopenapiinfor   )Zref_template)fieldsrH   rI   rF   r   ZsecuritySchemesc                    s   i | ]}| | qS rC   rC   )r   kr   rC   rD   r     s      zget_openapi.<locals>.<dictcomp>Zschemas
componentspathsr   r   Tr>   )setr   r   r   r   r   r   rh   r   r   r   r   rv   r   sortedr   r   )r0   r   r   rz   rR   r   r   r   r   r   r   r   rF   r   outputr   r   Zwebhook_pathsr~   Z
all_fieldsrI   rH   rJ   rm   resultr[   r   Zpath_definitionsZwebhookrC   r   rD   get_openapi  s    





r   )Mhttp.clientr   r   rr   typingr   r   r   r   r   r   r   r	   r
   r   Zfastapir   Zfastapi._compatr   r   r   r   r   r   r   r   Zfastapi.datastructuresr   Zfastapi.dependencies.modelsr   Zfastapi.dependencies.utilsr   r   r   Zfastapi.encodersr   Zfastapi.openapi.constantsr   r   r   Zfastapi.openapi.modelsr   Zfastapi.paramsr   r    Zfastapi.responsesr!   Zfastapi.typesr"   Zfastapi.utilsr#   r$   r%   Zstarlette.responsesr&   Zstarlette.routingr'   Zstarlette.statusr(   Ztyping_extensionsr)   r   r   r;   r   __annotations__rE   boolre   rl   r   rw   r}   r   r   r   r   rC   rC   rC   rD   <module>   s    0(
	
?&   
#$ B#

