o hf@s>ddlZddlZddlZddlmZddlmZmZddl Z ddl m Z m Z ddl mZddlmZddlmZddlmZmZddlmZmZdd lmZd d lmZd d lmZd d lm Z d dl!m"Z"m#Z#m$Z$d dlm%Z%m&Z&d dl'm(Z(m)Z)m*Z*e+e,Z-e.dZ/ddZ0ddZ1GdddeZGdddZ2dS)N) OrderedDict defaultdict) URLPattern URLResolver) versioning)SchemaGenerator)EndpointEnumerator)endpoint_ordering get_pk_name)get_pk_description is_list_view) api_settings)openapi)swagger_settings)SwaggerGenerationError)get_basic_type_infoget_queryset_fieldget_queryset_from_view)ReferenceResolver SwaggerDict)force_real_str get_consumes get_producesz{(?P\w+)}cCs^dd|D}t|}t|}|}t|D]\}}|||kr'|d|}nqdd|S)NcSsg|] }|ddqS)/)stripsplit).0pathrS/var/www/html/premium_crap/venv/lib/python3.10/site-packages/drf_yasg/generators.py szcommon_path..r)minmax enumeratejoin)paths split_pathss1s2commonicrrr common_paths  r-cCs|dvS)N>listcreateupdatedestroyretrievepartial_updater)actionrrr is_custom_action(sr5csZeZdZdfdd ZfddZdfdd Zd d Zdd d Zd dZddZ Z S)rNcstt|||||_dSN)superr__init__request)selfpatternsurlconfr9 __class__rr r8/s zEndpointEnumerator.__init__cs,|dr td||tt||S)N)zEurl pattern does not end in $ ('%s') - unexpected things might happen)endswithloggerwarning unescape_pathr7rget_path_from_regex)r: path_regexr=rr rD3s  z&EndpointEnumerator.get_path_from_regexcsxtt|||s dSt|jdd}t|jdd}|dur.t|tjr.|r.|| dvr.dSt|jdt dur:dSdS)NFversionversioning_class:swagger_schemaT) r7rshould_include_endpointgetattrr9cls issubclassrNamespaceVersioningrobject)r:rcallbackapp_name namespaceurl_namerGrHr=rr rK8sz*EndpointEnumerator.should_include_endpointcCsvt|jdd}|dur9t|tjr9t|jdd}|r9t|dd}d|}||vr3td|j||f|||}|S)a"If ``request.version`` is not ``None`` and `callback` uses ``URLPathVersioning``, this function replaces the ``version`` parameter in `path` with the actual version. :param str path: the templated path :param callback: the view callback :rtype: str rHNrG version_param{%s}z9view %s uses URLPathVersioning but URL %s has no param %s) rLrMrNrURLPathVersioningr9rAinforeplace)r:rrQrHrGrUrrr replace_versionGs   z"EndpointEnumerator.replace_versionc CsV|dur|j}g}|durt}|D]}|t|j}t|trnz=||} |j} |j} | | | |p3d|p6d| r\| | | } | |vrEWq| | | | D] } | | | f} | | qOWqtymtjdddYqwt|tr|j|j||rd||jfn|j|rd||jfn|j|d}||qtdt|qt|td }|S) z Return a list of all available API endpoints by inspecting the URL conf. Copied entirely from super. NrFzfailed to enumerate viewT)exc_infoz%s:%s)r;prefixrRrSignored_endpointszunknown pattern type {}key)r;setstrpattern isinstancerrDrQnamerKrZaddget_allowed_methodsappend ExceptionrArBrget_api_endpoints url_patternsrRrSextendformattypesortedr )r:r;r\rRrSr] api_endpointsrbrErrQrTmethodendpointnested_endpointsrrr ri\sH          z$EndpointEnumerator.get_api_endpointscCstdd|S)zzUnescape all backslash escapes from `s`. :param str s: string with backslash escapes :rtype: str z\\(.)z\1)resub)r:srrr unescapeszEndpointEnumerator.unescapecCshd}|r2t|}|s|||7} |S|||d|7}||7}||d}|s|S)aRemove backslashes escapes from all path components outside {parameters}. This is needed because ``simplify_regex`` does not handle this correctly. **NOTE:** this might destructively affect some url regex patterns that contain metacharacters (e.g. \w, \d) outside path parameter groups; if you are in this category, God help you :param str path: path possibly containing :return: the unescaped path :rtype: str rFN)PATH_PARAMETER_REsearchrvstartgroupend)r:r clean_pathmatchrrr rCs    z EndpointEnumerator.unescape_path)NNN)rFrFN)NrFNNN) __name__ __module__ __qualname__r8rDrKrZrirvrC __classcell__rrr=r r.s  3 rc@seZdZdZeZeZddddddZd-d d Z e d d Z ddZ ddZ d.ddZd/ddZddZddZddZddZdd Zd!d"Zd#d$Zd%d&Zd'd(Zd)d*Zd+d,Zd S)0OpenAPISchemaGeneratorz This class iterates over all registered API endpoints and returns an appropriate OpenAPI 2.0 compliant schema. Method implementations shamelessly stolen and adapted from rest-framework ``SchemaGenerator``. r2r/r0r3r1)getpostputpatchdeleterFNcCst|j||dd|||_||_||_g|_g|_tj |_ |dur*t j dur*t j }|rOt |}|jdvs9|js=td|jrItd|dS||j_dSdS)a :param openapi.Info info: information about the API :param str version: API version string; if omitted, `info.default_version` will be used :param str url: API scheme, host and port; if ``None`` is passed and ``DEFAULT_API_URL`` is not set, the url will be inferred from the request made against the schema view, so you should generally not need to set this parameter explicitly; if the empty string is passed, no host and scheme will be emitted If `url` is not ``None`` or the empty string, it must be a scheme-absolute uri (i.e. starting with http:// or https://), and any path component is ignored; See also: :ref:`documentation on base URL construction ` :param patterns: if given, only these patterns will be enumerated for inclusion in the API spec :param urlconf: if patterns is not given, use this urlconf to enumerate patterns; if not given, the default urlconf is used descriptionrFN)httphttpsz%`url` must be an absolute HTTP(S) urlzKpath component of api base URL %s is ignored; use FORCE_SCRIPT_NAME instead)rtitler_genrXrGconsumesproducesr SCHEMA_COERCE_METHOD_NAMEScoerce_method_namesrDEFAULT_API_URLurlparseschemenetlocrrrArBurl)r:rXrGrr;r< parsed_urlrrr r8s   zOpenAPISchemaGenerator.__init__cCs|jjSr6)rr)r:rrr rszOpenAPISchemaGenerator.urlcCstj}|dur t|i}|S)aGet the security schemes for this API. This determines what is usable in security requirements, and helps clients configure their authorization credentials. :return: the security schemes usable with this API :rtype: dict[str,dict] or None N)rSECURITY_DEFINITIONSr _as_odict)r:security_definitionsrrr get_security_definitionss z/OpenAPISchemaGenerator.get_security_definitionscCs:tj}|durdd|D}dd|D}t|td}|S)ajGet the base (global) security requirements of the API. This is never called if :meth:`.get_security_definitions` returns `None`. :param security_definitions: security definitions as returned by :meth:`.get_security_definitions` :return: the security schemes accepted by default :rtype: list[dict[str,list[str]]] or None NcSsg|]}|giqSrr)rsecurity_schemerrr r!szDOpenAPISchemaGenerator.get_security_requirements..cSsg|]}t|iqSr)rr)rsrrrr r!sr^)rSECURITY_REQUIREMENTSrnr.)r:rsecurity_requirementsrrr get_security_requirementss  z0OpenAPISchemaGenerator.get_security_requirementsFc Cs||}|jtjdd}ttj|_ttj |_ | ||||\}}| }|r/| |}nd}|j} | dur@|dur@|} tjd|j||jpJd|j pNd||| ||jd t|S)aGenerate a :class:`.Swagger` object representing the API schema. :param request: the request used for filtering accessible endpoints and finding the spec URI :type request: rest_framework.request.Request or None :param bool public: if True, all endpoints are included regardless of access through `request` :return: the generated Swagger specification :rtype: openapi.Swagger T) force_initN) rXr&rrrsecurity_url_prefix_versionr) get_endpointsreference_resolver_classrSCHEMA_DEFINITIONSrr DEFAULT_PARSER_CLASSESrrDEFAULT_RENDERER_CLASSESr get_pathsrrrbuild_absolute_uriSwaggerrXrGdict) r:r9public endpoints componentsr&r\rrrrrr get_schemas&    z!OpenAPISchemaGenerator.get_schemacCsh|j|||}t|dd}|dur,|D]\}}t||d}|dur+t|jd|qt|dd|S)aPCreate a view instance from a view callback as registered in urlpatterns. :param callback: view callback registered in urlpatterns :param str method: HTTP method :param request: request to bind to the view :type request: rest_framework.request.Request or None :return: the view instance _swagger_auto_schemaNswagger_fake_viewT)r create_viewrLitemssetattr__func__)r:rQrpr9view overrides_ view_methodrrr r&s   z"OpenAPISchemaGenerator.create_viewcCs>d|vr|Stt|dd}|rt|}nd}|dd|S)aKCoerce {pk} path arguments into the name of the model field, where possible. This is cleaner for an external representation (i.e. "this is an identifier", not "this is a database primary key"). :param str path: the path :param rest_framework.views.APIView view: associated view :rtype: str z{pk}modelNidrV)rLrr rY)r:rrr field_namerrr coerce_path;s z"OpenAPISchemaGenerator.coerce_pathc s|j|jj|jj|d}|}tt}i|D] \}}}||||}|||}|| ||f|j |<qfdd| DS)aIterate over all the registered endpoints in the API and return a fake view with the right parameters. :param request: request to bind to the endpoint views :type request: rest_framework.request.Request or None :return: {path: (view_class, list[(http_method, view_instance)]) :rtype: dict[str,(type,list[(str,rest_framework.views.APIView)])] )r9csi|] \}}|||fqSrr)rrmethodsview_clsrr _sz8OpenAPISchemaGenerator.get_endpoints..) endpoint_enumerator_classrr;r<rirr.rrrgrMr) r:r9 enumeratorr view_pathsrrprQrrrr rMs  z$OpenAPISchemaGenerator.get_endpointscCst|dr |j}nt|||rd}n|j|}dd|ddD}t|rWdd|jD}t |dkrN|j|}||j vrI|j |}||gS|d d |gS||j vra|j |}||gS) aXReturn a list of keys that should be used to group an operation within the specification. :: /users/ ("users", "list"), ("users", "create") /users/{pk}/ ("users", "read"), ("users", "update"), ("users", "delete") /users/enabled/ ("users", "enabled") # custom viewset list action /users/{pk}/star/ ("users", "star") # custom viewset detail action /users/{pk}/groups/ ("users", "groups", "list"), ("users", "groups", "create") /users/{pk}/groups/{pk}/ ("users", "groups", "read"), ("users", "groups", "update") :param str subpath: path to the operation with any common prefix/base path removed :param str method: HTTP method :param view: the view associated with the operation :rtype: list[str] r4r.cSsg|]}d|vr|qS){r)r componentrrr r!zs z=OpenAPISchemaGenerator.get_operation_keys..rcSsh|]}|dkr|qS)headr)rrprrr sz.rN) hasattrr4r default_mappinglowerrrr5 action_maplenr)r:subpathrprr4named_path_componentsmapped_methodsrrr get_operation_keysas*         z)OpenAPISchemaGenerator.get_operation_keyscCsxg}|D]3}|dd}g}|D] }d|vrn||qd|dd}|s.dS|d|dqt|S)a Given a list of all paths, return the common prefix which should be discounted when generating a schema structure. This will be the longest common string that does not include that last component of the URL, or the last component before a path parameter. For example: :: /api/v1/users/ /api/v1/users/{pk}/ The path prefix is ``/api/v1/``. :param list[str] paths: list of paths :rtype: str rrNr)rrrgr%r-)r:r&prefixesrrinitial_componentsrr\rrr determine_path_prefixs z,OpenAPISchemaGenerator.determine_path_prefixcCs|p |j|||S)aCheck if a given endpoint should be included in the resulting schema. :param str path: request path :param str method: http request method :param view: instantiated view callback :param bool public: if True, all endpoints are included regardless of access through `request` :returns: true if the view should be excluded :rtype: bool )rhas_view_permissions)r:rrprrrrr rKs z.OpenAPISchemaGenerator.should_include_endpointcCs tj|dS)zConstruct the Swagger Paths object. :param OrderedDict[str,openapi.PathItem] paths: mapping of paths to :class:`.PathItem` objects :returns: the :class:`.Paths` object :rtype: openapi.Paths r&)rPaths)r:r&rrr get_paths_objects z'OpenAPISchemaGenerator.get_paths_objectc Cs|s tjiddfS|t|pd}d|vsJdt}t|D]H\}\}} i} | D]!\} } ||| | |s=q0| | ||| ||} | durQ| | | <q0| rn|t |d}| dsed|}| ||| ||<q&|||fS)a+Generate the Swagger Paths for the API from the given endpoints. :param dict endpoints: endpoints as returned by get_endpoints :param ReferenceResolver components: resolver/container for Swagger References :param Request request: the request made against the schema view; can be None :param bool public: if True, all endpoints are included regardless of access through `request` :returns: the :class:`.Paths` object and the longest common path prefix, as a 2-tuple :rtype: tuple[openapi.Paths,str] rrFrz,base path cannot be templated in swagger 2.0Nr)rrrr.keysrrnrrK get_operationrr startswith get_path_itemr)r:rrr9rr\r&rrr operationsrpr operation path_suffixrrr rs*    z OpenAPISchemaGenerator.get_pathsc Cs||t|d||}|||}tj} t|d| } |d| } | dur(dS| |||||||} | |} | dur=dSd| vrMt| j t|j krM| ` d| vr]t| j t|j kr]| ` | S)aGet an :class:`.Operation` for the given API endpoint (path, method). This method delegates to :meth:`~.inspectors.ViewInspector.get_operation` of a :class:`~.inspectors.ViewInspector` determined according to settings and :func:`@swagger_auto_schema <.swagger_auto_schema>` overrides. :param view: the view associated with this endpoint :param str path: the path component of the operation URL :param str prefix: common path prefix among all endpoints :param str method: the http method of the operation :param openapi.ReferenceResolver components: referenceable components :param Request request: the request made against the schema view; can be None :rtype: openapi.Operation NrJ auto_schemarr) rr get_overridesrDEFAULT_AUTO_SCHEMA_CLASSrLrrr`rr) r:rrr\rprr9operation_keysrview_inspector_clsview_inspectorrrrr rs     z$OpenAPISchemaGenerator.get_operationcCs |||}tjdd|i|S)aGet a :class:`.PathItem` object that describes the parameters and operations related to a single path in the API. :param str path: the path :param type view_cls: the view that was bound to this path in urlpatterns :param dict[str,openapi.Operation] operations: operations defined on this path, keyed by lowercase HTTP method :rtype: openapi.PathItem parametersNr)get_path_parametersrPathItem)r:rrrpath_parametersrrr rs z$OpenAPISchemaGenerator.get_path_itemcCsF|}t|d|}t||d}t|di}||vr||}t|S)a"Get overrides specified for a given operation. :param view: the view associated with the operation :param str method: HTTP method :return: a dictionary containing any overrides set by :func:`@swagger_auto_schema <.swagger_auto_schema>` :rtype: dict r4Nr)rrLcopydeepcopy)r:rrpr4 action_methodrrrr r"s    z$OpenAPISchemaGenerator.get_overridesc Csg}t|}t|D]^}t||\}}t|pdtji}t|dd|kr8|dtjkr8t|d|dd|d<|rDt|ddrD|j } n|rRt|ddrRt ||} nd} tj d |t | d tj d |} || q |S) aReturn a list of Parameter instances corresponding to any templated path variables. :param str path: templated request path :param type view_cls: the view class associated with the path :return: path parameters :rtype: list[openapi.Parameter] rm lookup_fieldNlookup_value_regexrb help_textF primary_keyT)rdrrequiredin_r)r uritemplate variablesrrr TYPE_STRINGrLrrr ParameterrIN_PATHrg) r:rrrquerysetvariabler model_fieldattrsrfieldrrr r3s,  z*OpenAPISchemaGenerator.get_path_parameters)rFNNN)NFr6)r~rr__doc__rrrrrr8propertyrrrrrrrrrrKrrrrrrrrrr rs8 $   3"  %& r)3rloggingrs urllib.parseparser collectionsrrr django.urlsrrrest_frameworkrrest_framework.schemas.openapir!rest_framework.schemas.generatorsr_EndpointEnumeratorr r rest_framework.schemas.utilsr r rest_framework.settingsr rFr app_settingsrerrorsrinspectors.fieldrrrrrutilsrrr getLoggerr~rAcompilerwr-r5rrrrr s4