o
    3IhZ                     @   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-< d.ed/eee@ef eee@ef  f fd0d1ZBd2d3d4ed5ed6e0d7eeee<d8 f ef d9eCd/eee@ef  fd:d;ZDd2d3d<ee d5ed6e0d7eeee<d8 f ef d9eCd/eee@ef  fd=d>ZEd?ejFd@e@d/e@fdAdBZGd?ejFd@e@d/e@fdCdDZHd?ejFd@e@dEe
e@ d/ee@ef fdFdGZId2d3d?ejFdEe
e@ d5ed6e0d7eeee<d8 f ef d9eCd/eee@ef ee@ef ee@ef f fdHdIZJdJe	e8 d/ee fdKdLZKdMddddddddd2dN
dOe@dPe@dQe@dRee@ dSee@ dJe	e8 dTee	e8  dUeeee@ef   dVeeee@ee@ef f   dWee@ dXeee@ee@ef f  dYeee@ee@ef f  d9eCd/ee@ef fdZd[ZLdS )\    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objectLocationarrayanyOftypestringinteger)titler/   itemsMessage)r2   r/   z
Error Type)locmsgr/   )r2   r/   
propertiesrequiredHTTPValidationErrordetailDetail$ref)r2   r/   r7   InformationSuccessRedirectionzClient ErrorzServer ErrorzDefault Response)1XX2XX3XX4XX5XXDEFAULTstatus_code_rangesflat_dependantreturnc                 C   sN   i }g }| j D ]}t|jjddd}|jj}|||< |||ji q||fS )NTby_aliasexclude_none)security_requirementsr   security_schememodelscheme_nameappendscopes)rG   security_definitionsoperation_securitysecurity_requirementsecurity_definitionsecurity_name rW   U/home/air/sanwanet/gpt-api/venv/lib/python3.10/site-packages/fastapi/openapi/utils.py get_openapi_security_definitionsM   s   
rY   T)separate_input_output_schemas	dependantschema_generatormodel_name_mapfield_mapping)
validationserializationrZ   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 ]U}|j}t|ddsBq6t|||||d}|j|j|j|d}|jr]|j|d< t|dd }t|dd }|rrt||d	< n
|tkr|t||d< t|d
d rd|d
< || q6q0|S )NTskip_repeatsinclude_in_schemafieldr\   r]   r^   rZ   )nameinr8   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valuer8   ri   r   r   rP   )r[   r\   r]   r^   rZ   
parametersrG   rn   ro   rp   rq   parameter_groups
param_typeparam_groupparamrv   param_schema	parameterrj   rk   rW   rW   rX   !_get_openapi_operation_parameters^   sR   





r   
body_fieldc                 C   s   | sd S t | tsJ t| ||||d}tt| j}|j}| j}i }	|r(||	d< d|i}
|jr7t	|j|
d< n|j
tkrCt	|j
|
d< ||
i|	d< |	S )Nrd   r8   rh   rl   rk   content)
isinstancer   r   r   r   rv   
media_typer8   rj   r   rk   r   )r   r\   r]   r^   rZ   body_schemarv   request_media_typer8   request_body_oairequest_media_contentrW   rW   rX   "get_openapi_operation_request_body   s2   


r   routemethodc                 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)rf   rr   r   )warningswarnDeprecationWarningoperation_idpath_formatr$   rf   )r   r   rr   rW   rW   rX   generate_operation_id   s   r   c                 C   s   | j r| j S | jdd S )N_ )summaryrf   replacer2   r   r   rW   rW   rX   generate_operation_summary   s   r   operation_idsc                 C   s   i }| j r
| j |d< t| |d|d< | jr| j|d< | jp| j}||v rId| d| jj  }t| jdi d}|rB|d	| 7 }t	j
|d
d || ||d< | jrZ| j|d< |S )Ntagsr   r   ri   zDuplicate Operation ID z for function __globals____file__z at    r   operationIdrm   )r   r   ri   r   	unique_idendpoint__name__rw   getr   r   addrm   )r   r   r   	operationr   message	file_namerW   rW   rX   get_openapi_operation_metadata   s*   




r   c           +   
      s,  i }i }i }| j d usJ dt| jtr| jj}	n| j}	|	s#J d|	j}
| jr| j D ]}t| ||d g }t| j	dd}t
|d\}}|rR dg | |rY|| t| j	||||d}|| |rd	d
 |D }dd
 |D }|| t|  d< |tv rt| j||||d}|r| d< | jri }| jD ]}t|tjrt||||||d\}}}|j|i||j< q| d< | jd urt| j}nt|	j}|j d}|d urt|j!t"rt|j!}| j# di |i d< |
r.t$| jr.ddi}t%|	t&r| j'rt(| j'||||d}ni }| di |i di |
i d< | j)rÈ di }| j)* D ]\}}|+ } | ,dd  t|- }!|!dkrXd}!||!i }"t| t.shJ d| j/ |}#d }$|#rt(|#||||d}$|
pd}%| di |%i di }&t0|&|$ t1 t|- pt2j3j) t"|}'|  dp|" dp|'pd}(t0|"|  |(|"d< q=t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:  ||; < q-|||fS )(NzMethods must be a listz.A response class is needed to generate OpenAPI)r   r   r   Tra   )rG   security)r[   r\   r]   r^   rZ   c                 S   s   i | ]}|d  |d f|qS )rg   rf   rW   .0r~   rW   rW   rX   
<dictcomp>  s    z$get_openapi_path.<locals>.<dictcomp>c                 S   s(   i | ]}| d r|d |d f|qS )r8   rg   rf   )r   r   rW   rW   rX   r     s    rz   )r   r\   r]   r^   rZ   requestBodyr   r   r\   r]   r^   rZ   	callbacksstatus_code	responsesri   r/   r0   rd   r   rh   rN   rE   defaultz%An additional response must be a dictzapplication/jsonzAdditional Responsec                 3   s    | ]	}| d  v V  qdS )r   NrW   )r   statusr   rW   rX   	<genexpr>  s
    

z#get_openapi_path.<locals>.<genexpr>rC   zValidation Errorr<   r9   )ri   r   r*   )r*   r9   )<methodsr   response_classr   ry   r   rc   r   r   r[   rY   
setdefaultextendupdater   listvaluesr   r   r   r   r   APIRouteget_openapi_pathrr   rf   r   strinspect	signature__init__rz   r   r   intresponse_descriptionr%   r   r&   response_fieldr   r   r3   copypopupperdictresponse_fieldsr#   rF   httpclientr(   r   anyr   validation_error_definition$validation_error_response_definitionopenapi_extralower)+r   r   r\   r]   r^   rZ   rr   security_schemesdefinitionscurrent_response_classroute_response_media_typer   rz   rG   rR   rS   operation_parametersall_parametersrequired_parametersr   r   callbackcb_pathcb_security_schemescb_definitionsr   response_signaturestatus_code_paramresponse_schemaoperation_responsesadditional_status_codeadditional_responseprocess_responsestatus_code_keyopenapi_responsere   additional_field_schemar   additional_schemastatus_textri   http422all_route_paramsrW   r   rX   r      sV  





	











r   routesc                 C   s   g }g }g }g }| D ]J}t |dd rTt|tjrT|jr+t|jts%J d||j |jr4||j |jr?|	|j
  |jrJ|	t|j t|j}|	| q
|t|| |  }|S )Nrc   z'A request body must be a Pydantic Field)rw   r   r   r   r   r   rP   r   r   r   r   r   get_fields_from_routesr   r[   r   )r   body_fields_from_routesresponses_from_routesrequest_fields_from_routescallback_flat_modelsr   paramsflat_modelsrW   rW   rX   r     s8   


r   z3.1.0)
openapi_versionr   ri   webhooksr   serversterms_of_servicecontactlicense_inforZ   r2   versionr   r   ri   r   r   r   r   r   r   c              	      s  | |d}|r||d< |r||d< |	r|	|d< |
r|
|d< |r#||d< ||d}|r.||d< i }i }i }t  }tt|p<g t|pAg  }t|}ttd	}t||||d
\} |p[g D ]7}t|tj	rt
||||||d}|r|\}}}|r||ji | |r|di | |r | q\|pg D ]7}t|tj	rt
||||||d}|r|\}}}|r||ji | |r|di | |rψ | q r߇ fddt D |d< |r||d< ||d< |r||d< |r||d< ttdi |dddS )N)r2   r   r   ri   termsOfServicer   license)openapiinfor   )ref_template)fieldsr\   r]   rZ   r   securitySchemesc                    s   i | ]}| | qS rW   rW   )r   kr   rW   rX   r     s    zget_openapi.<locals>.<dictcomp>schemas
componentspathsr   r   TrI   rW   )setr   r   r   r   r   r   r   r   r   r   r   r   r   sortedr   r   )r2   r   r   r   ri   r   r   r   r   r   r   r   rZ   r   outputr  r  webhook_pathsr   
all_fieldsr]   r\   r^   r   resultrr   r   path_definitionswebhookrW   r  rX   get_openapi  s   







r  )Mhttp.clientr   r   r   typingr   r   r   r   r   r   r   r	   r
   r   fastapir   fastapi._compatr   r   r   r   r   r   r   r   fastapi.datastructuresr   fastapi.dependencies.modelsr   fastapi.dependencies.utilsr   r   r   fastapi.encodersr   fastapi.openapi.constantsr   r   r   fastapi.openapi.modelsr   fastapi.paramsr   r    fastapi.responsesr!   fastapi.typesr"   fastapi.utilsr#   r$   r%   starlette.responsesr&   starlette.routingr'   starlette.statusr(   typing_extensionsr)   r   r   rF   r   __annotations__rY   boolr   r   r   r   r   r   r   r   r  rW   rW   rW   rX   <module>   sH  
 0(
	


	
?	
%


#	$

 A
#
	

