o
    K Žiýh  ã                   @   s   d dl Z d dlZd dlmZ d dl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mZ d d
l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' G dd„ de"ƒZ(G dd„ de$ƒZ)dS )é    N)ÚDecimal)Ú
attrgetter)Úurljoin)ÚDecimalValidatorÚEmailValidatorÚMaxLengthValidatorÚMaxValueValidatorÚMinLengthValidatorÚMinValueValidatorÚRegexValidatorÚURLValidator)Úmodels)Ú	force_str)Ú
exceptionsÚ	renderersÚserializers)Ú
inflectionÚuritemplate)Ú_UnvalidatedFieldÚempty)Úapi_settingsé   )ÚBaseSchemaGenerator)ÚViewInspector)Úget_pk_descriptionÚis_list_viewc                   @   s&   e Zd Zdd„ Zdd„ Zd	dd„ZdS )
ÚSchemaGeneratorc                 C   s.   | j pd| jpddœ}| jd ur| j|d< |S )NÚ )ÚtitleÚversionÚdescription)r   r   r    )ÚselfÚinfo© r#   úhC:\Users\psant\Downloads\Anu Raj Backend\school\venv\Lib\site-packages\rest_framework/schemas/openapi.pyÚget_info   s   þ

zSchemaGenerator.get_infoc                 C   s‚   i }|D ]:}|| D ]3}d|| | vrq
|| | d }||v r6t  dj|| d || d |||d�¡ ||dœ||< q
qd S )NÚoperationIdzîYou have a duplicated operationId in your OpenAPI schema: {operation_id}
	Route: {route1}, Method: {method1}
	Route: {route2}, Method: {method2}
	An operationId has to be unique across your schema. Your schema may not work in other tools.ÚrouteÚmethod)Zroute1Zmethod1Zroute2Zmethod2Úoperation_id)r'   r(   )ÚwarningsÚwarnÚformat)r!   ÚpathsÚidsr'   r(   r)   r#   r#   r$   Úcheck_duplicate_operation_id&   s,   

÷ÿþîÿz,SchemaGenerator.check_duplicate_operation_idNFc                 C   s   |   ¡  i }i }|  |rdn|¡\}}|D ]_\}}}	|  |||	¡s"q|	j ||¡}
|	j ||¡}| ¡ D ]}||vr;q4|| || krDq4t d|› d�¡ q4| 	|¡ | 
d¡r^|dd… }t| jpcd|ƒ}| |i ¡ |
|| | ¡ < q|  |¡ d|  ¡ |dœ}t|ƒdkrŽd	|i|d
< |S )z,
        Generate a OpenAPI schema.
        NzSchema component "z-" has been overridden with a different value.ú/r   z3.0.2)Zopenapir"   r-   r   ZschemasÚ
components)Z_initialise_endpointsZ_get_paths_and_endpointsZhas_view_permissionsÚschemaÚget_operationÚget_componentsÚkeysr*   r+   ÚupdateÚ
startswithr   ÚurlÚ
setdefaultÚlowerr/   r%   Úlen)r!   ÚrequestÚpublicZcomponents_schemasr-   Ú_Zview_endpointsÚpathr(   ÚviewÚ	operationr1   Úkr2   r#   r#   r$   Ú
get_schema@   s<   


ýÿzSchemaGenerator.get_schema)NF)Ú__name__Ú
__module__Ú__qualname__r%   r/   rC   r#   r#   r#   r$   r      s    r   c                       s  e Zd Zd>‡ fdd„	Zg Zg Zdddddd	œZd
d„ Zdd„ Zdd„ Z	dd„ Z
dd„ Zdd„ Zdd„ Zdd„ Zdd„ Zdd„ Zdd„ Zd d!„ Zd"d#„ Zd$d%„ Zd&d'„ Zd(d)„ Zd*d+„ Zd,d-„ Zd.d/„ Zd0d1„ Zd2d3„ Zd4d5„ Zd6d7„ Zd8d9„ Zd:d;„ Zd<d=„ Z ‡  Z!S )?Ú
AutoSchemaNc                    s>   |rt dd„ |D ƒƒstdƒ‚|| _|| _|| _tƒ  ¡  dS )a  
        :param operation_id_base: user-defined name in operationId. If empty, it will be deducted from the Model/Serializer/View name.
        :param component_name: user-defined component's name. If empty, it will be deducted from the Serializer's class name.
        c                 s   ó   � | ]}t |tƒV  qd S ©N©Ú
isinstanceÚstr)Ú.0Útagr#   r#   r$   Ú	<genexpr>{   ó   € z&AutoSchema.__init__.<locals>.<genexpr>z'tags must be a list or tuple of string.N)ÚallÚ
ValueErrorÚ_tagsÚoperation_id_baseÚcomponent_nameÚsuperÚ__init__)r!   ÚtagsrT   rU   ©Ú	__class__r#   r$   rW   v   s   zAutoSchema.__init__ÚretrieveÚcreater6   ZpartialUpdateÚdestroy)ÚgetÚpostÚputÚpatchÚdeletec                 C   sœ   i }|   ||¡|d< |  ||¡|d< g }||  ||¡7 }||  ||¡7 }||  ||¡7 }||d< |  ||¡}|r<||d< |  ||¡|d< |  ||¡|d< |S )Nr&   r    Ú
parametersZrequestBodyÚ	responsesrX   )Úget_operation_idÚget_descriptionÚget_path_parametersÚget_pagination_parametersÚget_filter_parametersÚget_request_bodyÚget_responsesÚget_tags)r!   r?   r(   rA   rc   Úrequest_bodyr#   r#   r$   r3   �   s   zAutoSchema.get_operationc                 C   sP   | j dur| j S |jj}t dtj¡}| d|¡}|dkr&td |jj¡ƒ‚|S )z¡
        Compute the component's name from the serializer.
        Raise an exception if the serializer's class name is "Serializer" (case-insensitive).
        NÚ
serializerr   z�"{}" is an invalid class name for schema generation. Serializer's class name should be unique and explicit. e.g. "ItemSerializer")	rU   rZ   rD   ÚreÚcompileÚ
IGNORECASEÚsubÚ	Exceptionr,   )r!   rn   rU   Úpatternr#   r#   r$   Úget_component_name¡   s   

ýzAutoSchema.get_component_namec                 C   sˆ   |  ¡ dkri S |  ||¡}|  ||¡}i }t|tjƒr,|  |¡}|  |¡}| ||¡ t|tjƒrB|  |¡}|  |¡}| ||¡ |S )zN
        Return components with their properties from the serializer.
        rb   N)	r:   Úget_request_serializerÚget_response_serializerrK   r   Ú
Serializerru   Úmap_serializerr9   )r!   r?   r(   Zrequest_serializerZresponse_serializerr1   rU   Úcontentr#   r#   r$   r4   ¸   s   



zAutoSchema.get_componentsc                 C   s.   |  d¡}|d d dd„ |dd … D ƒ¡ S )Nr>   r   r   c                 s   s   � | ]}|  ¡ V  qd S rI   )r   )rM   Úxr#   r#   r$   rO   Õ   s   € z,AutoSchema._to_camel_case.<locals>.<genexpr>r   )ÚsplitÚjoin)r!   Z	snake_strr1   r#   r#   r$   Ú_to_camel_caseÑ   s   
$zAutoSchema._to_camel_casec                 C   sê   t t | jddƒddƒ}| jdur| j}nP|dur|j}nH|  ||¡dur8|  ||¡jj}| d¡r7|dd… }n,| jjj}| d¡rI|dd… }n| d¡rT|dd	… }| | ¡ ¡rd|dt|ƒ … }|d
krst	snJ dƒ‚t	 
|¡}|S )za
        Compute the base part for operation ID from the model, serializer or view name.
        ÚquerysetNÚmodelrx   iöÿÿÿZAPIViewiùÿÿÿZViewéüÿÿÿÚlistz:`inflection` must be installed for OpenAPI schema support.)Úgetattrr@   rT   rD   Úget_serializerrZ   Úendswithr   r;   r   Z	pluralize)r!   r?   r(   Úactionr€   Únamer#   r#   r$   Úget_operation_id_base×   s*   

€



z AutoSchema.get_operation_id_basec                 C   s`   t | jd| ¡ ƒ}t||| jƒrd}n|| jvr|  |¡}n| j| ¡  }|  |||¡}|| S )z^
        Compute an operation ID from the view type and get_operation_id_base method.
        r†   r‚   N)rƒ   r@   r:   r   Úmethod_mappingr~   rˆ   )r!   r?   r(   Úmethod_namer†   r‡   r#   r#   r$   re   ý   s   
zAutoSchema.get_operation_idc           	   	   C   s¾   t sJ dƒ‚tt| jddƒddƒ}g }t  |¡D ]D}d}|durMz|j |¡}W n ty3   d}Y nw |durA|jrAt|jƒ}n|durM|j	rMt
||ƒ}|dd|dd	id
œ}| |¡ q|S )zL
        Return a list of parameters from templated path variables.
        z;`uritemplate` must be installed for OpenAPI schema support.r   Nr€   r   r?   TÚtypeÚstring)r‡   ÚinÚrequiredr    r2   )r   rƒ   r@   Ú	variablesÚ_metaÚ	get_fieldrs   Ú	help_textr   Zprimary_keyr   Úappend)	r!   r?   r(   r€   rc   Úvariabler    Úmodel_fieldZ	parameterr#   r#   r$   rg     s0   ÿ
ÿû	zAutoSchema.get_path_parametersc                 C   s8   |   ||¡sg S g }| jjD ]}||ƒ  | j¡7 }q|S rI   )Úallows_filtersr@   Úfilter_backendsÚget_schema_operation_parameters)r!   r?   r(   rc   Zfilter_backendr#   r#   r$   ri   1  s   z AutoSchema.get_filter_parametersc                 C   s:   t | jddƒdu rdS t| jdƒr| jjdv S | ¡ dv S )zÜ
        Determine whether to include filter Fields in schema.

        Default implementation looks for ModelViewSet or GenericAPIView
        actions/methods that cause filtering on the default implementation.
        r—   NFr†   )r‚   r[   r6   Zpartial_updater]   )r^   r`   ra   rb   )rƒ   r@   Úhasattrr†   r:   ©r!   r?   r(   r#   r#   r$   r–   9  s
   zAutoSchema.allows_filtersc                 C   s0   | j }t|||ƒsg S |  ¡ }|sg S | |¡S rI   )r@   r   Úget_paginatorr˜   )r!   r?   r(   r@   Ú	paginatorr#   r#   r$   rh   F  s   
z$AutoSchema.get_pagination_parametersc                 C   sŒ   t t |j¡ƒ}tdd„ |D ƒƒrd}n&tdd„ |D ƒƒr d}ntdd„ |D ƒƒr,d}ntdd„ |D ƒƒr8d	}nd }d
|i}|rD||d< |S )Nc                 s   rH   rI   )rK   Úbool©rM   Úchoicer#   r#   r$   rO   T  rP   z-AutoSchema.map_choicefield.<locals>.<genexpr>Úbooleanc                 s   rH   rI   )rK   Úintrž   r#   r#   r$   rO   V  rP   Úintegerc                 s   s    � | ]}t |tttfƒV  qd S rI   )rK   r¡   Úfloatr   rž   r#   r#   r$   rO   X  s   € Únumberc                 s   rH   rI   rJ   rž   r#   r#   r$   rO   [  rP   rŒ   Úenumr‹   )r‚   ÚdictÚfromkeysÚchoicesrQ   )r!   Úfieldr¨   r‹   Úmappingr#   r#   r$   Úmap_choicefieldR  s   ý
zAutoSchema.map_choicefieldc                 C   s  t |tjƒrd|  |j¡dœS t |tjƒr |  |¡}d|d< |S t |tjƒr/d|  |j¡dœS t |tj	ƒr[t
|ddƒrB| j|jd�S t
|jdd ƒ}|d ur[|jj}t |tjƒr[dd	iS t |tjƒrid|  |¡dœS t |tjƒrt|  |¡S t |tjƒr�di dœ}t |jtƒs�|  |j¡|d
< |S t |tjƒršdddœS t |tjƒr¥dddœS t |tjƒr°dddœS t |tjƒr»dddœS t |tjƒrÆdddœS t |tjƒrÜddi}|jdkrÚ|j|d< |S t |tjƒ�r%t
|dtjƒrðdddœ}nddi}|j �rt!d|j d d  d ƒ|d< |j"�rt#|j"d ƒd |d< |d  |d< |  $||¡ |S t |tj%ƒ�r8ddi}|  $||¡ |S t |tj&ƒ�redd	i}|  $||¡ t#| 'dd¡ƒd k�s_t#| 'dd¡ƒd k�rcd!|d< |S t |tj(ƒ�rqdd"dœS tj)d#tj*dtj+dtj,di}d| '|j-d¡iS )$NÚarray©r‹   ÚitemsÚobjectr‹   Úpk_fieldF)r©   r€   r¢   r®   rŒ   Údate)r‹   r,   z	date-timeÚemailÚuriÚuuidZbothr,   Úcoerce_to_stringÚdecimalr¤   Ú.r   Ú0Ú1Ú
multipleOfÚ9ÚmaximumÚminimumr   iÿÿÿZint64Úbinaryr    ).rK   r   ZListSerializerry   Úchildrx   ZManyRelatedFieldÚ	map_fieldZchild_relationZPrimaryKeyRelatedFieldrƒ   r°   r   r�   Úpkr   Z	AutoFieldZMultipleChoiceFieldr«   ZChoiceFieldÚ	ListFieldr   Z	DateFieldZDateTimeFieldZ
EmailFieldZURLFieldZ	UUIDFieldZIPAddressFieldÚprotocolZDecimalFieldr   ÚCOERCE_DECIMAL_TO_STRINGÚdecimal_placesr£   Zmax_whole_digitsr¡   Ú_map_min_maxZ
FloatFieldZIntegerFieldr^   Z	FileFieldZBooleanFieldZ	JSONFieldZ	DictFieldZHStoreFieldrZ   )r!   r©   Údatar€   r•   rª   rz   ZFIELD_CLASS_SCHEMA_TYPEr#   r#   r$   rÀ   n  s¾   
þ

þþ
þþþþþþÿ

þÿÿÿ,þüzAutoSchema.map_fieldc                 C   s(   |j r|j |d< |jr|j|d< d S d S )Nr¼   r½   )Z	max_valueZ	min_value)r!   r©   rz   r#   r#   r$   rÆ   þ  s
   
ÿzAutoSchema._map_min_maxc                 C   sæ   g }i }|j  ¡ D ]\}t|tjƒrq	|jr |js | |  |¡¡ |  	|¡}|j
r,d|d< |jr3d|d< |jr:d|d< |jd urN|jtkrNt|jƒsN|j|d< |jrXt|jƒ|d< |  ||¡ |||  |¡< q	d|dœ}|rq||d	< |S )
NTZreadOnlyZ	writeOnlyZnullableÚdefaultr    r¯   )r‹   Ú
propertiesrŽ   )ÚfieldsÚvaluesrK   r   ZHiddenFieldrŽ   Úpartialr“   Úget_field_namerÀ   Z	read_onlyZ
write_onlyZ
allow_nullrÈ   r   Úcallabler’   rL   Úmap_field_validators)r!   rn   rŽ   rÉ   r©   r2   Úresultr#   r#   r$   ry     s4   

þzAutoSchema.map_serializerc                 C   s\  |j D ]¨}t|tƒrd|d< t|tƒrd|d< t|tƒr'|jj dd¡|d< qt|tƒr<d}t|t	j
ƒr6d}|j||< qt|tƒrQd	}t|t	j
ƒrKd
}|j||< qt|tƒr\|j|d< qt|tƒrg|j|d< qt|tƒr«t|dtjƒs«|jr…td|jd d  d ƒ|d< |jr«|j}|jdurš|jdkrš||j8 }t|d ƒd |d< |d  |d< qdS )z&
        map field validators
        r²   r,   r³   z\Zz\zrt   Z	maxLengthZmaxItemsZ	minLengthZminItemsr¼   r½   rµ   r·   r   r¸   r¹   rº   Nr   r»   )Z
validatorsrK   r   r   r   Úregexrt   Úreplacer   r   rÂ   Zlimit_valuer	   r   r
   r   rƒ   r   rÄ   rÅ   r£   Z
max_digitsr¡   )r!   r©   r2   ÚvÚ	attr_nameÚdigitsr#   r#   r$   rÏ   (  sD   








ÿ
€ÞzAutoSchema.map_field_validatorsc                 C   s   |j S )z�
        Override this method if you want to change schema field name.
        For example, convert snake_case field name to camelCase.
        N)Ú
field_name)r!   r©   r#   r#   r$   rÍ   P  s   zAutoSchema.get_field_namec                 C   s   t | jdd ƒ}|r|ƒ S d S )NÚpagination_class)rƒ   r@   )r!   r×   r#   r#   r$   r›   W  s   zAutoSchema.get_paginatorc                 C   s   t ttdƒ| jjƒƒS )NÚ
media_type)r‚   Úmapr   r@   Zparser_classesrš   r#   r#   r$   Úmap_parsers]  s   zAutoSchema.map_parsersc                 C   s0   g }| j jD ]}t|tjƒrq| |j¡ q|S rI   )r@   Zrenderer_classesÚ
issubclassr   ZBrowsableAPIRendererr“   rØ   )r!   r?   r(   Zmedia_typesZrendererr#   r#   r$   Úmap_renderers`  s   zAutoSchema.map_renderersc              
   C   sN   | j }t|dƒs
d S z| ¡ W S  tjy&   t d |jj	||¡¡ Y d S w )Nr„   zt{}.get_serializer() raised an exception during schema generation. Serializer fields will not be generated for {} {}.)
r@   r™   r„   r   ZAPIExceptionr*   r+   r,   rZ   rD   )r!   r?   r(   r@   r#   r#   r$   r„   i  s   

ýûzAutoSchema.get_serializerc                 C   ó   |   ||¡S )zr
        Override this method if your view uses a different serializer for
        handling request body.
        N©r„   rš   r#   r#   r$   rv   x  ó   z!AutoSchema.get_request_serializerc                 C   rÝ   )zu
        Override this method if your view uses a different serializer for
        populating response data.
        NrÞ   rš   r#   r#   r$   rw     rß   z"AutoSchema.get_response_serializerc                 C   s   dd|   |¡› �iS )Nz$refz#/components/schemas/)ru   )r!   rn   r#   r#   r$   Úget_reference†  s   zAutoSchema.get_referencec                    sZ   |dvri S |   ||¡| _|  ||¡}t|tjƒsi ‰ n|  |¡‰ d‡ fdd„| jD ƒiS )N)ÚPUTÚPATCHÚPOSTrz   c                    ó   i | ]}|d ˆ i“qS ©r2   r#   ©rM   Úct©Úitem_schemar#   r$   Ú
<dictcomp>—  ó    ÿÿz/AutoSchema.get_request_body.<locals>.<dictcomp>)rÚ   Úrequest_media_typesrv   rK   r   rx   rà   )r!   r?   r(   rn   r#   rè   r$   rj   ‰  s   
þÿzAutoSchema.get_request_bodyc                    s¬   |dkr
dddiiS |   ||¡| _|  ||¡}t|tjƒs i }n|  |¡}t||| jƒr=d|dœ‰ |  	¡ }|r<| 
ˆ ¡‰ n|‰ |dkrEdnd	}|‡ fd
d„| jD ƒddœiS )NÚDELETEZ204r    r   r¬   r­   rã   Z201Z200c                    rä   rå   r#   ræ   ©Zresponse_schemar#   r$   rê   »  rë   z,AutoSchema.get_responses.<locals>.<dictcomp>)rz   r    )rÜ   Úresponse_media_typesrw   rK   r   rx   rà   r   r@   r›   Zget_paginated_response_schema)r!   r?   r(   rn   ré   rœ   Ústatus_coder#   rî   r$   rk   �  s6   ÿÿ
þ
€
þøÿzAutoSchema.get_responsesc                 C   s:   | j r| j S | d¡r|dd … }| d¡d  dd¡gS )Nr0   r   r   r>   ú-)rS   r7   r|   rÒ   rš   r#   r#   r$   rl   Æ  s
   
zAutoSchema.get_tags)NNN)"rD   rE   rF   rW   rì   rï   r‰   r3   ru   r4   r~   rˆ   re   rg   ri   r–   rh   r«   rÀ   rÆ   ry   rÏ   rÍ   r›   rÚ   rÜ   r„   rv   rw   rà   rj   rk   rl   Ú__classcell__r#   r#   rY   r$   rG   t   sJ    û&$ $(	)rG   )*ro   r*   r¶   r   Úoperatorr   Úurllib.parser   Zdjango.core.validatorsr   r   r   r   r	   r
   r   r   Z	django.dbr   Zdjango.utils.encodingr   Zrest_frameworkr   r   r   Zrest_framework.compatr   r   Zrest_framework.fieldsr   r   Zrest_framework.settingsr   Ú
generatorsr   Z
inspectorsr   Úutilsr   r   r   rG   r#   r#   r#   r$   Ú<module>   s"    (\