o MfD@sddlZddlmZddlmZddlmZddlmZddl m Z ddl m Z dd l mZmZmZmZmZmZmZmZmZd d lmZmZeeZGd d d eZdS)N) OrderedDictis_form_media_type) AutoSchema) is_success)openapi)SwaggerGenerationError) filter_noneforce_real_strforce_serializer_instance get_consumes get_producesguess_response_status merge_paramsno_bodyparam_list_to_odict) ViewInspectorcall_view_methodcseZdZd4fdd Zd4ddZddZdd Zd d Zd d ZddZ ddZ ddZ ddZ ddZ ddZddZddZddZd d!Zd"d#Zd4d$d%Zd&d'Zd(d)Zd*d+Zd,d-Zd4d.d/Zd0d1Zd2d3ZZS)5SwaggerAutoSchemaNcs4tt|||||||t|_||j_||_dSN)superr__init__r_schviewoperation_keys)selfrpathmethod componentsrequest overridesr __class__Z/var/www/html/analyze/labelStudio/lib/python3.10/site-packages/drf_yasg/inspectors/view.pyrs zSwaggerAutoSchema.__init__c Cs|p|j}|}|}||}|}||}t|}||}||}|\}} | } | dus?t | t s?Jd| } | |} |} tj|t| t|| |||| | | d S)Nz7security must be a list of security requirement objects) operation_id descriptionsummary responses parametersconsumesproducestagssecurity deprecated)rr rget_request_body_parametersget_query_parametersr add_manual_parametersget_operation_idget_summary_and_description get_security isinstancelist is_deprecatedget_tags get_responsesr Operationr )rrr,r-bodyqueryr+r'r)r(r/r0r.r*r%r%r& get_operations6      zSwaggerAutoSchema.get_operationcCs|}d}|dur gSt|tjjr|}tdd|Dr+|dur&td||S|dur4||}|dur>| |gSgS)aReturn the request body parameters for this view. |br| This is either: - a list with a single object Parameter with a :class:`.Schema` derived from the request serializer - a list of primitive Parameters parsed as form data :param list[str] consumes: a list of accepted MIME types as returned by :meth:`.get_consumes` :return: a (potentially empty) list of :class:`.Parameter`\ s either ``in: body`` or ``in: formData`` :rtype: list[openapi.Parameter] Ncs|]}t|VqdSrr.0encodingr%r%r& Oz@SwaggerAutoSchema.get_request_body_parameters..z$form request body cannot be a Schema) get_request_serializerr7rSchemaOR_REFanyr get_request_form_parametersget_request_body_schemamake_body_parameter)rr, serializerschemar%r%r&r1<s   z-SwaggerAutoSchema.get_request_body_parameterscCs t|jdS)zReturn the serializer as defined by the view's ``get_serializer()`` method. :return: the view's ``Serializer`` :rtype: rest_framework.serializers.Serializer get_serializer)rrrr%r%r&get_view_serializerXs z%SwaggerAutoSchema.get_view_serializercCsd|jdd}|dur0|turtS|j|jvr#tdd|jdt|tj j r,|St |S|S)zOParse the request_body key in the override dict. This method is not public API. request_bodyNz%request_body can only be applied to (,z=); are you looking for query_serializer or manual_parameters?) r"getrr body_methodsr joinr7rrGrHr r body_overrider%r%r&_get_request_body_override`s z,SwaggerAutoSchema._get_request_body_overridecCs4|}|dur|j|jvr|S|turdS|S)zReturn the request serializer (used for parsing the request payload) for this endpoint. :return: the request serializer, or one of :class:`.Schema`, :class:`.SchemaRef`, ``None`` :rtype: rest_framework.serializers.Serializer N)rYrimplicit_body_methodsrQrrWr%r%r&rFps z(SwaggerAutoSchema.get_request_serializercCs|j|tjdS)zGiven a Serializer, return a list of ``in: formData`` :class:`.Parameter`\ s. :param serializer: the view's request serializer as returned by :meth:`.get_request_serializer` :rtype: list[openapi.Parameter] in_)serializer_to_parametersrIN_FORMrrMr%r%r&rJsz-SwaggerAutoSchema.get_request_form_parameterscCs ||S)zReturn the :class:`.Schema` for a given request's body data. Only applies to PUT, PATCH and POST requests. :param serializer: the view's request serializer as returned by :meth:`.get_request_serializer` :rtype: openapi.Schema )serializer_to_schemar_r%r%r&rKs z)SwaggerAutoSchema.get_request_body_schemacCstjdtjd|dS)zGiven a :class:`.Schema` object, create an ``in: body`` :class:`.Parameter`. :param openapi.Schema schema: the request body schema :rtype: openapi.Parameter dataT)namer\requiredrN)r ParameterIN_BODY)rrNr%r%r&rLsz%SwaggerAutoSchema.make_body_parametercCs|jddpg}tdd|Drtdtdd|DrKtdd|D}|s5tdd|Ds9td |j|jvrKtd d |jd t||S) zAdd/replace parameters from the given list of automatically generated request parameters. :param list[openapi.Parameter] parameters: generated parameters :return: modified parameters :rtype: list[openapi.Parameter] manual_parametersNcs|] }|jtjkVqdSrr\rrerBparamr%r%r&rDz:SwaggerAutoSchema.add_manual_parameters..zDspecify the body parameter as a Schema or Serializer in request_bodycsrgr)r\rr^rir%r%r&rDrkcsrgrrhrir%r%r&rDrkcsr@rrrAr%r%r&rDrEz~cannot add form parameters when the request has a request body; did you forget to set an appropriate parser class on the view?z(form parameters can only be applied to (rSz) HTTP methods) r"rTrIr r rrUrVr)rr+rfhas_body_parameterr%r%r&r3s   z'SwaggerAutoSchema.add_manual_parameterscCs|}tj||dS)zGet the possible responses for this view as a swagger :class:`.Responses` object. :return: the documented responses :rtype: openapi.Responses )r*)get_response_serializersr Responsesget_response_schemas)rresponse_serializersr%r%r&r;szSwaggerAutoSchema.get_responsescCs |}|r |tur |S|S)aReturn the default response serializer for this endpoint. This is derived from either the ``request_body`` override or the request serializer (:meth:`.get_view_serializer`). :return: response serializer, :class:`.Schema`, :class:`.SchemaRef`, ``None`` )rYrrQrWr%r%r&get_default_response_serializers z1SwaggerAutoSchema.get_default_response_serializercCs|j}t|}d}|dvr|}|pd}|r&t|tjs&||p%d}|r?|r4tjtj |d}| r?| |p>|}t t ||iS)zGet the default responses determined for this view from the request serializer and request method. :type: dict[str, openapi.Schema] )rTpostputpatch)typeitems)rlowerrrqr7rrGr`has_list_response TYPE_ARRAY should_pageget_paginated_responserstr)rrdefault_statusdefault_schemar%r%r&get_default_responsess z'SwaggerAutoSchema.get_default_responsescCsd|jddpi}tdd|D}t}tdd|Ds$|}|dd|D|S)aReturn the response codes that this view is expected to return, and the serializer for each response body. The return value should be a dict where the keys are possible status codes, and values are either strings, ``Serializer``\ s, :class:`.Schema`, :class:`.SchemaRef` or :class:`.Response` objects. See :func:`@swagger_auto_schema <.swagger_auto_schema>` for more details. :return: the response serializers :rtype: dict r*Ncs |] \}}t||fVqdSrr}rBscrespr%r%r&rDz=SwaggerAutoSchema.get_response_serializers..css$|] }|dkrtt|VqdS)defaultN)rint)rBrr%r%r&rDs"csrrrrr%r%r&rDr)r"rTrrwrIrupdate)rmanual_responsesr*r%r%r&rms z*SwaggerAutoSchema.get_response_serializerscCst}|D]b\}}t|trtjt|d}nJ|sqt|tjr=|}t|drs      ,