a
    amgZ                     @   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g d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 d.d/d0ZBd1d2eee0eeee<d3 f ef eCeee@ef  d4d5d6ZDd1d2ee ee0eeee<d3 f ef eCeee@ef  d7d8d9ZEejFe@e@d:d;d<ZGejFe@e@d:d=d>ZHejFe@e
e@ ee@ef d?d@dAZId1d2ejFe
e@ ee0eeee<d3 f ef eCeee@ef ee@ef ee@ef f dBdCdDZJe	e8 ee dEdFdGZKdHddddddddd1dI
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 dJdKdLZLdS )M    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-   )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modelscheme_nameappendZscopes)r<   security_definitionsoperation_securityZsecurity_requirementZsecurity_definitionZsecurity_name rD   L/var/www/html/idle/venv/lib/python3.9/site-packages/fastapi/openapi/utils.py get_openapi_security_definitionsM   s    
rF   T)separate_input_output_schemas)Z
validationZserialization)	dependantschema_generatormodel_name_mapfield_mappingrG   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fieldrI   rJ   rK   rG   )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   rS   r   r   rA   )rH   rI   rJ   rK   rG   
parametersr<   rX   rY   rZ   r[   Zparameter_groupsZ
param_typeZparam_groupparamr`   Zparam_schemaZ	parameterrT   rU   rD   rD   rE   !_get_openapi_operation_parameters^   sP    





rf   )
body_fieldrI   rJ   rK   rG   r=   c                 C   s   | sd S t | tsJ 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 )NrN   r6   rR   rV   rU   content)
isinstancer   r   r   r   r`   
media_typer6   rT   r   rU   r   )rg   rI   rJ   rK   rG   Zbody_schemar`   Zrequest_media_typer6   request_body_oaiZrequest_media_contentrD   rD   rE   "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)rP   r\   rn   )warningswarnDeprecationWarningoperation_idpath_formatr$   rP   )rm   rn   r\   rD   rD   rE   generate_operation_id   s    rw   c                 C   s   | j r| j S | jdd S )N_ )summaryrP   replacer0   rm   rn   rD   rD   rE   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}||v 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   rS   zDuplicate Operation ID z for function __globals____file__z at    rp   ZoperationIdrW   )r   r}   rS   ru   Z	unique_idZendpoint__name__ra   getrr   rs   addrW   )rm   rn   r~   	operationru   message	file_namerD   rD   rE   get_openapi_operation_metadata   s*    




r   )rm   r~   rI   rJ   rK   rG   r=   c           +   
      s@  i }i }i }| j d usJ dt| jtr4| jj}	n| j}	|	sFJ 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v r8t| j||||d}|r8| d< | jri }| jD ]>}t|tjrJt||||||d\}}}|j|i||j< qJ| d< | jd urt| j}n:t|	j}|j d}|d u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.sJ d| j/ |}#d }$|#r<t(|#||||d}$|
pd}%| di |%i di }&t0|&|$ t1 t|- p`t2j3j) t"|}'|  dp|" dp|'pd}(t0|"|  |(|"d< qtt4})t5| j	}*|*s| jrt6 fdd |)d!dfD sd"ddd#t7d$ iiid% d |)< d&|vr|t8t9d' | j:r(t0 | j:  ||; < qZ|||fS )(NzMethods must be a listz.A response class is needed to generate OpenAPI)rm   rn   r~   TrL   )r<   security)rH   rI   rJ   rK   rG   c                 S   s   i | ]}|d  |d f|qS )rQ   rP   rD   .0re   rD   rD   rE   
<dictcomp>  s   z$get_openapi_path.<locals>.<dictcomp>c                 S   s(   i | ] }| d r|d |d f|qS )r6   rQ   rP   )r   r   rD   rD   rE   r     s   
rd   )rg   rI   rJ   rK   rG   ZrequestBodyrm   r~   rI   rJ   rK   rG   	callbacksstatus_code	responsesrS   r-   r.   rN   rh   rR   r?   r:   defaultz%An additional response must be a dictzapplication/jsonzAdditional Responsec                 3   s   | ]}| d  v V  qdS )r   NrD   )r   statusr   rD   rE   	<genexpr>  s   z#get_openapi_path.<locals>.<genexpr>r9   zValidation Errorr8   r7   )rS   rh   r*   )r*   r7   )<methodsri   response_classr   rc   rj   rM   r   r   rH   rF   
setdefaultextendupdaterf   listvaluesr   rl   rg   r   r   APIRouteget_openapi_pathr\   rP   r   strinspect	signature__init__rd   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~   rI   rJ   rK   rG   r\   security_schemesdefinitionsZcurrent_response_classZroute_response_media_typern   rd   r<   rB   rC   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_responserO   Zadditional_field_schemarj   Zadditional_schemaZstatus_textrS   Zhttp422Zall_route_paramsrD   r   rE   r      s^   






	









r   )routesr=   c                 C   s   g }g }g }g }| D ]}t |dd rt|tjr|jrVt|jtsJJ d||j |jrh||j |jr~|	|j
  |jr|	t|j t|j}|	| q|t|| |  }|S )NrM   z'A request body must be a Pydantic Field)ra   ri   r   r   rg   r   rA   r   r   r   r   r   get_fields_from_routesr   rH   r   )r   Zbody_fields_from_routesZresponses_from_routesZrequest_fields_from_routesZcallback_flat_modelsrm   paramsZflat_modelsrD   rD   rE   r     s6    

r   z3.1.0)
openapi_versionrz   rS   webhooksr   serversterms_of_servicecontactlicense_inforG   )r0   versionr   rz   rS   r   r   r   r   r   r   r   rG   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 i |dddS )N)r0   r   rz   rS   ZtermsOfServicer   license)Zopenapiinfor   )Zref_template)fieldsrI   rJ   rG   r   ZsecuritySchemesc                    s   i | ]}| | qS rD   rD   )r   kr   rD   rE   r         zget_openapi.<locals>.<dictcomp>Zschemas
componentspathsr   r   Tr>   )setr   r   r   r   r   r   ri   r   r   r   r   rv   r   sortedr   r   )r0   r   r   rz   rS   r   r   r   r   r   r   r   rG   r   outputr   r   Zwebhook_pathsr~   Z
all_fieldsrJ   rI   rK   rm   resultr\   r   Zpath_definitionsZwebhookrD   r   rE   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__rF   boolrf   rl   r   rw   r}   r   r   r   r   rD   rD   rD   rE   <module>   s   
0(
	
?&
#$ B#

