a
    ΝGd ^                     @   s  d dl Z d dlZd dlZd dlZd dlmZ d dlmZ d dlZd dl	m
Z
 d dlmZ d dlmZmZ d dlmZmZmZmZ d dlmZ d d	lmZ d d
lmZ d dlmZmZ d dlm Z  ddl!m"Z" e#e$Z%G dd de&Z'G dd de&Z(dde(dddddddddddddfddZ)dd Z*dd Z+dd Z,dd Z-dd Z.dd  Z/d!d" Z0d#d$ Z1d:d%d&Z2d'd( Z3d)d* Z4d+d, Z5d-d. Z6d;d2d3Z7d4d5 Z8d6d7 Z9d8d9 Z:dS )<    N)OrderedDict)Decimal)models)	force_str)serializersstatus)DestroyModelMixinListModelMixinRetrieveModelMixinUpdateModelMixin)FileUploadParseris_form_media_type)api_settings)encodersjson)APIView   )swagger_settingsc                   @   s   e Zd ZdZdS )no_bodyzcUsed as a sentinel value to forcibly remove the body of a request via :func:`.swagger_auto_schema`.N__name__
__module____qualname____doc__ r   r   J/var/www/html/django/DPS/env/lib/python3.9/site-packages/drf_yasg/utils.pyr      s   r   c                   @   s   e Zd ZdZdS )unsetzmUsed as a sentinel value for function parameters not set by the caller where ``None`` would be a valid value.Nr   r   r   r   r   r      s   r   c                    s0    	
fdd}|S )a  Decorate a view method to customize the :class:`.Operation` object generated from it.

    `method` and `methods` are mutually exclusive and must only be present when decorating a view method that accepts
    more than one HTTP request method.

    The `auto_schema` and `operation_description` arguments take precedence over view- or method-level values.

    :param str method: for multi-method views, the http method the options should apply to
    :param list[str] methods: for multi-method views, the http methods the options should apply to
    :param drf_yasg.inspectors.SwaggerAutoSchema auto_schema: custom class to use for generating the Operation object;
        this overrides both the class-level ``swagger_schema`` attribute and the ``DEFAULT_AUTO_SCHEMA_CLASS``
        setting, and can be set to ``None`` to prevent this operation from being generated
    :param request_body: custom request body which will be used as the ``schema`` property of a
        :class:`.Parameter` with ``in: 'body'``.

        A Schema or SchemaRef is not valid if this request consumes form-data, because ``form`` and ``body`` parameters
        are mutually exclusive in an :class:`.Operation`. If you need to set custom ``form`` parameters, you can use
        the `manual_parameters` argument.

        If a ``Serializer`` class or instance is given, it will be automatically converted into a :class:`.Schema`
        used as a ``body`` :class:`.Parameter`, or into a list of ``form`` :class:`.Parameter`\ s, as appropriate.
    :type request_body: drf_yasg.openapi.Schema or drf_yasg.openapi.SchemaRef  or rest_framework.serializers.Serializer
        or type[no_body]

    :param rest_framework.serializers.Serializer query_serializer: if you use a ``Serializer`` to parse query
        parameters, you can pass it here and have :class:`.Parameter` objects be generated automatically from it.

        If any ``Field`` on the serializer cannot be represented as a ``query`` :class:`.Parameter`
        (e.g. nested Serializers, file fields, ...), the schema generation will fail with an error.

        Schema generation will also fail if the name of any Field on the `query_serializer` conflicts with parameters
        generated by ``filter_backends`` or ``paginator``.

    :param list[drf_yasg.openapi.Parameter] manual_parameters: a list of manual parameters to override the
        automatically generated ones

        :class:`.Parameter`\ s are identified by their (``name``, ``in``) combination, and any parameters given
        here will fully override automatically generated parameters if they collide.

        It is an error to supply ``form`` parameters when the request does not consume form-data.

    :param str operation_id: operation ID override; the operation ID must be unique across the whole API
    :param str operation_description: operation description override
    :param str operation_summary: operation summary string
    :param list[dict] security: security requirements override; used to specify which authentication mechanism
        is required to call this API; an empty list marks the endpoint as unauthenticated (i.e. removes all accepted
        authentication schemes), and ``None`` will inherit the top-level security requirements
    :param bool deprecated: deprecation status for operation
    :param responses: a dict of documented manual responses
        keyed on response status code. If no success (``2xx``) response is given, one will automatically be
        generated from the request body and http method. If any ``2xx`` response is given the automatic response is
        suppressed.

        * if a plain string is given as value, a :class:`.Response` with no body and that string as its description
          will be generated
        * if ``None`` is given as a value, the response is ignored; this is mainly useful for disabling default
          2xx responses, i.e. ``responses={200: None, 302: 'something'}``
        * if a :class:`.Schema`, :class:`.SchemaRef` is given, a :class:`.Response` with the schema as its body and
          an empty description will be generated
        * a ``Serializer`` class or instance will be converted into a :class:`.Schema` and treated as above
        * a :class:`.Response` object will be used as-is; however if its ``schema`` attribute is a ``Serializer``,
          it will automatically be converted into a :class:`.Schema`
    :type responses: dict[int or str, (drf_yasg.openapi.Schema or drf_yasg.openapi.SchemaRef or
        drf_yasg.openapi.Response or str or rest_framework.serializers.Serializer)]

    :param list[type[drf_yasg.inspectors.FieldInspector]] field_inspectors: extra serializer and field inspectors; these
        will be tried before :attr:`.ViewInspector.field_inspectors` on the :class:`.inspectors.SwaggerAutoSchema`
    :param list[type[drf_yasg.inspectors.FilterInspector]] filter_inspectors: extra filter inspectors; these will be
        tried before :attr:`.ViewInspector.filter_inspectors` on the :class:`.inspectors.SwaggerAutoSchema`
    :param list[type[drf_yasg.inspectors.PaginatorInspector]] paginator_inspectors: extra paginator inspectors; these
        will be tried before :attr:`.ViewInspector.paginator_inspectors` on the :class:`.inspectors.SwaggerAutoSchema`
    :param list[str] tags: tags override
    :param extra_overrides: extra values that will be saved into the ``overrides`` dict; these values will be available
        in the handling :class:`.inspectors.SwaggerAutoSchema` instance via ``self.overrides``
    c                    s  t fddtjD r J d
	r>t	nd rLtnd rZtnd rhtnd dtturd<  sS tdg }tdi }fdd	| D }|| }td
d fdd	tdg D }||  tdi }sr s(J dt	t	ksBJ dt
trVJ drh g}ndd	 D }t fdd|D sJ dt fdd|D rJ d r^t	|t	|ksJ dt dkr|sJ dn
|p }t fdd|D rJ dt fdd|D r>J dfdd|D  _n"|rlJ drzJ d_S )Nc                 3   s   | ]}| v V  qd S Nr   ).0Zhm)extra_overridesr   r   	<genexpr>s       z9swagger_auto_schema.<locals>.decorator.<locals>.<genexpr>z"HTTP method names not allowed here)request_bodyquery_serializermanual_parametersoperation_idoperation_summary
deprecatedoperation_descriptionsecurity	responsesfilter_inspectorspaginator_inspectorsfield_inspectorstagsauto_schemabind_to_methodsmappingc                    s   g | ]\}}| j kr|qS r   )r   )r   mthname)view_methodr   r   
<listcomp>   r"   z:swagger_auto_schema.<locals>.decorator.<locals>.<listcomp>clsc                    s   g | ]}t  |r|qS r   )hasattr)r   mview_clsr   r   r6      r"   http_method_names_swagger_auto_schemazI`method` or `methods` can only be specified on @action or @api_view viewsz specify either method or methodszR`methods` expects to receive a list of methods; use `method` for a single argumentc                 S   s   g | ]}|  qS r   lowerr   r3   r   r   r   r6      r"   c                 3   s   | ]}| v V  qd S r   r   r@   )available_http_methodsr   r   r!      r"   zhttp method not bound to viewc                 3   s   | ]}| v V  qd S r   r   r@   existing_datar   r   r!      r"   z"http method defined multiple timeszthis should never happenr   zon multi-method api_view or action, you must specify swagger_auto_schema on a per-method basis using one of the `method` or `methods` argumentsc                 3   s    | ]}t t |d dV  qd S )Nr=   )r8   getattrr@   r:   r   r   r!      r"   z+swagger_auto_schema applied twice to methodc                 3   s   | ]}| v V  qd S r   r   r@   rB   r   r   r!      r"   c                 3   s   | ]}|   fV  qd S r   r>   r@   )datar   r   r!      r"   zthe methods argument should only be specified when decorating an action; you should also ensure that you put the swagger_auto_schema decorator AFTER (above) the _route decorator)anyr   r<   listfilter_noner   updaterD   itemsbool
isinstancestrr?   alllenr=   )r5   r1   r2   Zmapping_methodsZaction_http_methodsZapi_view_http_methodsZ_methodsr0   r(   r    r.   r,   r%   methodmethodsr)   r&   r'   r-   r$   r#   r+   r*   r/   )rA   rE   rC   r;   r5   r   	decoratorr   st     
  
 z&swagger_auto_schema.<locals>.decoratorr   )rQ   rR   r0   r#   r$   r%   r&   r)   r'   r*   r(   r+   r.   r,   r-   r/   r    rS   r   rP   r   swagger_auto_schema"   s    P,NrT   c                    s    fdd}|S )z
    Decorates the method of a serializers.SerializerMethodField
    to hint as to how Swagger should be generated for this field.

    :param serializer_or_field: ``Serializer``/``Field`` class or instance
    :return:
    c                    s
    | _ | S r   )Z_swagger_serializer)Zserializer_methodserializer_or_fieldr   r   rS      s    z,swagger_serializer_method.<locals>.decoratorr   )rV   rS   r   rU   r   swagger_serializer_method   s    	rW   c                 C   s   t |dd}t ||dp|}t |dd}t |dd}|dv sL|du sL|dkrPd	S |d
v sh|d	u sh|dkrldS t|trzd	S t|tttfrdS | dd}|rd|d v rdS d	S )zCheck if the given path/method appears to represent a list view (as opposed to a detail/instance view).

    :param str path: view path
    :param str method: http method
    :param APIView view: target view
    :rtype: bool
    action Ndetailsuffix)rG   createFListT)retrieverI   Zpartial_updatedestroyInstance/{)rD   rL   r	   r
   r   r   stripsplit)pathrQ   viewrX   rZ   r[   Zpath_componentsr   r   r   is_list_view   s     	
rh   c                 C   s&   | dkrt jS | dkrt jS t jS d S )Npostdelete)r   ZHTTP_201_CREATEDZHTTP_204_NO_CONTENTZHTTP_200_OK)rQ   r   r   r   guess_response_status   s
    rk   c                 C   s.   t dd | D }t|t| ks*J d|S )a  Transform a list of :class:`.Parameter` objects into an ``OrderedDict`` keyed on the ``(name, in_)`` tuple of
    each parameter.

    Raises an ``AssertionError`` if `parameters` contains duplicate parameters (by their name + in combination).

    :param list[drf_yasg.openapi.Parameter] parameters: the list of parameters
    :return: `parameters` keyed by ``(name, in_)``
    :rtype: dict[(str,str),drf_yasg.openapi.Parameter]
    c                 s   s   | ]}|j |jf|fV  qd S r   )r4   Zin_)r   paramr   r   r   r!     r"   z&param_list_to_odict.<locals>.<genexpr>zduplicate Parameters found)r   rO   )
parametersresultr   r   r   param_list_to_odict  s    
ro   c                 C   s"   t | } | t | t|  S )a  Merge `overrides` into `parameters`. This is the same as appending `overrides` to `parameters`, but any element
    of `parameters` whose ``(name, in_)`` tuple collides with an element in `overrides` is replaced by it.

    Raises an ``AssertionError`` if either list contains duplicate parameters.

    :param list[drf_yasg.openapi.Parameter] parameters: initial parameters
    :param list[drf_yasg.openapi.Parameter] overrides: overriding parameters
    :return: merged list
    :rtype: list[drf_yasg.openapi.Parameter]
    )ro   rI   rG   values)rm   Z	overridesr   r   r   merge_params  s    rq   c                 C   sx   | du rdS d}t | tr4t| dd |  D }t | ttfrXt| dd | D }|durtt|t| krt|S | S )zRemove ``None`` values from tuples, lists or dictionaries. Return other objects as-is.

    :param obj: the object
    :return: collection with ``None`` values removed
    Nc                 s   s*   | ]"\}}|d ur|d ur||fV  qd S r   r   )r   kvr   r   r   r!   *  r"   zfilter_none.<locals>.<genexpr>c                 s   s   | ]}|d ur|V  qd S r   r   )r   rs   r   r   r   r!   ,  r"   )rL   dicttyperJ   rG   tuplerO   )objZnew_objr   r   r   rH      s    
rH   c                 C   sL   t | r*t| tjs$J d| j |  S t| tjsHJ dt| j | S )aY  Force `serializer` into a ``Serializer`` instance. If it is not a ``Serializer`` class or instance, raises
    an assertion error.

    :param serializer: serializer class or instance
    :type serializer: serializers.BaseSerializer or type[serializers.BaseSerializer]
    :return: serializer instance
    :rtype: serializers.BaseSerializer
    Serializer required, not %s-Serializer class or instance required, not %sinspectisclass
issubclassr   ZBaseSerializerr   rL   ru   
serializerr   r   r   force_serializer_instance2  s    	
r   c                 C   sZ   | du rdS t | r4t| tjs0J d| j | S t| tjsRJ dt| j t| S )a6  Given a ``Serializer`` class or instance, return the ``Serializer`` class. If `serializer` is not a ``Serializer``
    class or instance, raises an assertion error.

    :param serializer: serializer class or instance, or ``None``
    :return: serializer class
    :rtype: type[serializers.BaseSerializer]
    Nrx   ry   rz   r~   r   r   r   get_serializer_classD  s    
r   c                 C   sZ   | pg } g }| D ]D}t |r8|r,t||rT|| q|rFt||r|t| q|S )at  Given a list of instances or class objects, return the list of their classes.

    :param classes_or_instances: mixed list to parse
    :type classes_or_instances: list[type or object]
    :param expected_base_class: if given, only subclasses or instances of this type will be returned
    :type expected_base_class: type
    :return: list of classes
    :rtype: list
    )r{   r|   r}   appendrL   ru   )Zclasses_or_instancesZexpected_base_classrn   rw   r   r   r   get_object_classesX  s    

r   c                 C   sJ   t | } dd | D } dd | p"g D }dd |D }t|dkrF|S |S )a  Extract ``consumes`` MIME types from a list of parser classes.

    :param list parser_classes: parser classes
    :type parser_classes: list[rest_framework.parsers.BaseParser or type[rest_framework.parsers.BaseParser]]
    :return: MIME types for ``consumes``
    :rtype: list[str]
    c                 S   s   g | ]}t |ts|qS r   )r}   r   )r   Zpcr   r   r   r6   x  r"   z get_consumes.<locals>.<listcomp>c                 S   s   g | ]
}|j qS r   
media_type)r   parserr   r   r   r6   y  r"   c                 S   s   g | ]}t |s|qS r   r   )r   encodingr   r   r   r6   z  r"   r   )r   rO   )Zparser_classesmedia_typesZnon_form_media_typesr   r   r   get_consumeso  s    r   c                 C   s,   t | } dd | pg D }dd |D }|S )a/  Extract ``produces`` MIME types from a list of renderer classes.

    :param list renderer_classes: renderer classes
    :type renderer_classes: list[rest_framework.renderers.BaseRenderer or type[rest_framework.renderers.BaseRenderer]]
    :return: MIME types for ``produces``
    :rtype: list[str]
    c                 S   s   g | ]
}|j qS r   r   )r   rendererr   r   r   r6     r"   z get_produces.<locals>.<listcomp>c                    s(   g | ]  t  fd dtjD s qS )c                 3   s   | ]}| v V  qd S r   r   )r   excludedr   r   r   r!     r"   z*get_produces.<locals>.<listcomp>.<genexpr>)rF   r   ZEXCLUDED_MEDIA_TYPES)r   r   r   r   r6     s   )r   )Zrenderer_classesr   r   r   r   get_produces  s    r   c                 C   s,   t | tjst | tjr(t| dtj S dS )zReturns true if ``field`` is a django-rest-framework DecimalField and its ``coerce_to_string`` attribute or the
    ``COERCE_DECIMAL_TO_STRING`` setting is set to ``False``.

    :rtype: bool
    Zcoerce_to_stringF)rL   r   ZDecimalFieldr   rD   rest_framework_settingsZCOERCE_DECIMAL_TO_STRING)fieldr   r   r   decimal_as_float  s    r   c                 C   sx   t | dd}t| j}t|dr(|j}nL|dkrTt| tjrTt	dt
|   d}n |}|drt|dtd  }|S )zGet serializer's ref_name (or None for ModelSerializer if it is named 'NestedSerializer')

    :param serializer: Serializer instance
    :return: Serializer's ``ref_name`` or ``None`` for inline serializer
    :rtype: str or None
    ZMetaNref_nameZNestedSerializerzDForcing inline output for ModelSerializer named 'NestedSerializer':

Serializer)rD   ru   r   r8   r   rL   r   ZModelSerializerloggerdebugrM   endswithrO   )r   Zserializer_metaZserializer_namer   r   r   r   get_serializer_ref_name  s    


r   utf-8Fstrictc                 C   s8   | dur4t | |||} t| tkr*d|  } t| } | S )zi
    Force `s` into a ``str`` instance.

    Fix for https://github.com/axnsan12/drf-yasg/issues/159
    NrY   )r   ru   rM   textwrapdedent)sr   Zstrings_onlyerrorsr   r   r   force_real_str  s    
r   c                 C   sX   |  |}t|tr.t| r&t|}nt|}t|tjrBt|}t	tj
|tjdS )zConvert a python value related to a field (default, choices, etc.) into its OpenAPI-compatible representation.

    :param serializers.Field field: field associated with the value
    :param object value: value
    :return: the converted value
    )r7   )Zto_representationrL   r   r   floatrM   pytzZ
BaseTzInfor   loadsdumpsr   JSONEncoder)r   valuer   r   r   field_value_to_representation  s    


r   c                 C   s   t | dtj}|tjurt|r~z4t|dr6||  t |ddrL|| }n| }W n( ty|   tjd| dd tj}Y n0 |tjur|durzt	| |}W n( ty   tjd	| dd tj}Y n0 |S )
z
    Get the default value for a field, converted to a JSON-compatible value while properly handling callables.

    :param field: field instance
    :return: default value
    defaultset_contextZrequires_contextFzfdefault for %s is callable but it raised an exception when called; 'default' will not be set on schemaT)exc_infoNzX'default' on schema for %s will not be set because to_representation raised an exception)
rD   r   emptycallabler8   r   	Exceptionr   warningr   )r   r   r   r   r   get_field_default  s.    




r   c                 C   s   t jdkrt| tS t| tS )zCheck if a given object is a dict that maintains insertion order.

    :param obj: the dict object to check
    :rtype: bool
    )      )sysversion_inforL   rt   r   )rw   r   r   r   dict_has_ordered_keys  s    

r   )N)r   Fr   );r{   loggingr   r   collectionsr   decimalr   r   Z	django.dbr   Zdjango.utils.encodingr   Zrest_frameworkr   r   Zrest_framework.mixinsr   r	   r
   r   Zrest_framework.parsersr   Zrest_framework.requestr   Zrest_framework.settingsr   r   Zrest_framework.utilsr   r   Zrest_framework.viewsr   Zapp_settingsr   	getLoggerr   r   objectr   r   rT   rW   rh   rk   ro   rq   rH   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   <module>   sT   



 "$	

!