U
    Æ½„bk  ã                   @   sF  d 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	 ddl
mZ ddlmZmZ d+d	d
„Zdd„ Zdd„ ZG dd„ deƒZG dd„ deƒZG dd„ deƒZG dd„ deƒZG dd„ deƒZG dd„ deƒZG dd„ deƒZG dd„ deƒZG dd „ d eƒZG d!d"„ d"eƒZG d#d$„ d$eƒZG d%d&„ d&eƒZd'd(„ Z d)d*„ Z!dS ),z·
Handled exceptions raised by REST framework.

In addition Django's built in 403 and 404 exceptions are handled.
(`django.http.Http404` and `django.core.exceptions.PermissionDenied`)
é    N)ÚJsonResponse)Ú	force_str)Úgettext_lazy)Úngettext©Ústatus)Ú
ReturnDictÚ
ReturnListc                    s–   t | ttfƒr<‡ fdd„| D ƒ}t | tƒr8t|| jd�S |S t | tƒrx‡ fdd„|  ¡ D ƒ}t | tƒrtt|| jd�S |S t| ƒ}t	| dˆ ƒ}t
||ƒS )zw
    Descend into a nested data structure, forcing any
    lazy translation strings or strings into `ErrorDetail`.
    c                    s   g | ]}t |ˆ ƒ‘qS © ©Ú_get_error_details©Ú.0Úitem©Údefault_coder
   ú=/tmp/pip-unpacked-wheel-iohwuy7z/rest_framework/exceptions.pyÚ
<listcomp>   s    z&_get_error_details.<locals>.<listcomp>)Ú
serializerc                    s   i | ]\}}|t |ˆ ƒ“qS r
   r   ©r   ÚkeyÚvaluer   r
   r   Ú
<dictcomp>   s   ÿ z&_get_error_details.<locals>.<dictcomp>Úcode)Ú
isinstanceÚlistÚtupler	   r   ÚdictÚitemsr   r   ÚgetattrÚErrorDetail)Údatar   ÚretÚtextr   r
   r   r   r      s"    
ÿ


þ
r   c                 C   s:   t | tƒrdd„ | D ƒS t | tƒr4dd„ |  ¡ D ƒS | jS )Nc                 S   s   g | ]}t |ƒ‘qS r
   ©Ú
_get_codesr   r
   r
   r   r   .   s     z_get_codes.<locals>.<listcomp>c                 S   s   i | ]\}}|t |ƒ“qS r
   r$   r   r
   r
   r   r   0   s      z_get_codes.<locals>.<dictcomp>©r   r   r   r   r   ©Údetailr
   r
   r   r%   ,   s
    

r%   c                 C   s@   t | tƒrdd„ | D ƒS t | tƒr4dd„ |  ¡ D ƒS | | jdœS )Nc                 S   s   g | ]}t |ƒ‘qS r
   ©Ú_get_full_detailsr   r
   r
   r   r   6   s     z%_get_full_details.<locals>.<listcomp>c                 S   s   i | ]\}}|t |ƒ“qS r
   r)   r   r
   r
   r   r   8   s      z%_get_full_details.<locals>.<dictcomp>)Úmessager   r&   r'   r
   r
   r   r*   4   s    

þr*   c                       sJ   e Zd ZdZdZd‡ fdd„	Z‡ fdd„Zdd„ Zd	d
„ Zdd„ Z	‡  Z
S )r    zA
    A string-like object that can additionally have a code.
    Nc                    s   t ƒ  | |¡}||_|S ©N)ÚsuperÚ__new__r   )ÚclsÚstringr   Úself©Ú	__class__r
   r   r.   E   s    zErrorDetail.__new__c                    sH   t ƒ  |¡}|tkrtS z|o(| j|jkW S  tk
rB   | Y S X d S r,   )r-   Ú__eq__ÚNotImplementedr   ÚAttributeError)r1   ÚotherÚrr2   r
   r   r4   J   s    zErrorDetail.__eq__c                 C   s   |   |¡ S r,   )r4   )r1   r7   r
   r
   r   Ú__ne__S   s    zErrorDetail.__ne__c                 C   s   dt | ƒ| jf S )NzErrorDetail(string=%r, code=%r))Ústrr   ©r1   r
   r
   r   Ú__repr__V   s    þzErrorDetail.__repr__c                 C   s   t t| ƒƒS r,   )Úhashr:   r;   r
   r
   r   Ú__hash__\   s    zErrorDetail.__hash__)N)Ú__name__Ú
__module__Ú__qualname__Ú__doc__r   r.   r4   r9   r<   r>   Ú__classcell__r
   r
   r2   r   r    ?   s   	r    c                   @   sD   e Zd ZdZejZedƒZdZ	ddd„Z
dd„ Zd	d
„ Zdd„ ZdS )ÚAPIExceptionz‚
    Base class for REST framework exceptions.
    Subclasses should provide `.status_code` and `.default_detail` properties.
    zA server error occurred.ÚerrorNc                 C   s,   |d kr| j }|d kr| j}t||ƒ| _d S r,   )Údefault_detailr   r   r(   ©r1   r(   r   r
   r
   r   Ú__init__i   s
    zAPIException.__init__c                 C   s
   t | jƒS r,   )r:   r(   r;   r
   r
   r   Ú__str__q   s    zAPIException.__str__c                 C   s
   t | jƒS )ze
        Return only the code part of the error details.

        Eg. {"name": ["required"]}
        )r%   r(   r;   r
   r
   r   Ú	get_codest   s    zAPIException.get_codesc                 C   s
   t | jƒS )z 
        Return both the message & code parts of the error details.

        Eg. {"name": [{"message": "This field is required.", "code": "required"}]}
        )r*   r(   r;   r
   r
   r   Úget_full_details|   s    zAPIException.get_full_details)NN)r?   r@   rA   rB   r   ÚHTTP_500_INTERNAL_SERVER_ERRORÚstatus_codeÚ_rF   r   rH   rI   rJ   rK   r
   r
   r
   r   rD   `   s   
rD   c                   @   s(   e Zd ZejZedƒZdZddd„Z	dS )ÚValidationErrorzInvalid input.ÚinvalidNc                 C   sZ   |d kr| j }|d kr| j}t|tƒr0t|ƒ}nt|tƒsJt|tƒsJ|g}t||ƒ| _d S r,   )rF   r   r   r   r   r   r   r(   rG   r
   r
   r   rH   ‘   s    

zValidationError.__init__)NN)
r?   r@   rA   r   ÚHTTP_400_BAD_REQUESTrM   rN   rF   r   rH   r
   r
   r
   r   rO   Œ   s   rO   c                   @   s   e Zd ZejZedƒZdZdS )Ú
ParseErrorzMalformed request.Zparse_errorN)	r?   r@   rA   r   rQ   rM   rN   rF   r   r
   r
   r
   r   rR   ¡   s   rR   c                   @   s   e Zd ZejZedƒZdZdS )ÚAuthenticationFailedz%Incorrect authentication credentials.Zauthentication_failedN©	r?   r@   rA   r   ZHTTP_401_UNAUTHORIZEDrM   rN   rF   r   r
   r
   r
   r   rS   §   s   rS   c                   @   s   e Zd ZejZedƒZdZdS )ÚNotAuthenticatedz-Authentication credentials were not provided.Znot_authenticatedNrT   r
   r
   r
   r   rU   ­   s   rU   c                   @   s   e Zd ZejZedƒZdZdS )ÚPermissionDeniedz2You do not have permission to perform this action.Zpermission_deniedN)	r?   r@   rA   r   ZHTTP_403_FORBIDDENrM   rN   rF   r   r
   r
   r
   r   rV   ³   s   rV   c                   @   s   e Zd ZejZedƒZdZdS )ÚNotFoundz
Not found.Ú	not_foundN)	r?   r@   rA   r   ZHTTP_404_NOT_FOUNDrM   rN   rF   r   r
   r
   r
   r   rW   ¹   s   rW   c                       s0   e Zd ZejZedƒZdZd‡ fdd„	Z	‡  Z
S )ÚMethodNotAllowedzMethod "{method}" not allowed.Úmethod_not_allowedNc                    s,   |d krt | jƒj|d�}tƒ  ||¡ d S )N)Úmethod©r   rF   Úformatr-   rH   )r1   r[   r(   r   r2   r
   r   rH   Ä   s    zMethodNotAllowed.__init__)NN)r?   r@   rA   r   ZHTTP_405_METHOD_NOT_ALLOWEDrM   rN   rF   r   rH   rC   r
   r
   r2   r   rY   ¿   s   rY   c                       s0   e Zd ZejZedƒZdZd‡ fdd„	Z	‡  Z
S )ÚNotAcceptablez,Could not satisfy the request Accept header.Únot_acceptableNc                    s   || _ tƒ  ||¡ d S r,   )Úavailable_renderersr-   rH   )r1   r(   r   r`   r2   r
   r   rH   Ï   s    zNotAcceptable.__init__)NNN)r?   r@   rA   r   ZHTTP_406_NOT_ACCEPTABLErM   rN   rF   r   rH   rC   r
   r
   r2   r   r^   Ê   s   r^   c                       s0   e Zd ZejZedƒZdZd‡ fdd„	Z	‡  Z
S )ÚUnsupportedMediaTypez1Unsupported media type "{media_type}" in request.Úunsupported_media_typeNc                    s,   |d krt | jƒj|d�}tƒ  ||¡ d S )N)Ú
media_typer\   )r1   rc   r(   r   r2   r
   r   rH   Ù   s    zUnsupportedMediaType.__init__)NN)r?   r@   rA   r   ZHTTP_415_UNSUPPORTED_MEDIA_TYPErM   rN   rF   r   rH   rC   r
   r
   r2   r   ra   Ô   s   ra   c                       s@   e Zd ZejZedƒZedƒZedƒZ	dZ
d‡ fdd„	Z‡  ZS )	Ú	ThrottledzRequest was throttled.z$Expected available in {wait} second.z%Expected available in {wait} seconds.Z	throttledNc              	      sj   |d krt | jƒ}|d k	rRt |¡}d |t t| jj|d�| jj|d�|ƒƒf¡}|| _	t
ƒ  ||¡ d S )Nú )Úwait)r   rF   ÚmathÚceilÚjoinr   Úextra_detail_singularr]   Úextra_detail_pluralrf   r-   rH   )r1   rf   r(   r   r2   r
   r   rH   æ   s    

þþzThrottled.__init__)NNN)r?   r@   rA   r   ZHTTP_429_TOO_MANY_REQUESTSrM   rN   rF   rj   rk   r   rH   rC   r
   r
   r2   r   rd   ß   s   rd   c                 O   s   ddi}t |tjd�S )z$
    Generic 500 error handler.
    rE   zServer Error (500)r   )r   r   rL   )ÚrequestÚargsÚkwargsr!   r
   r
   r   Úserver_errorô   s     ÿro   c                 O   s   ddi}t |tjd�S )z$
    Generic 400 error handler.
    rE   zBad Request (400)r   )r   r   rQ   )rl   Ú	exceptionrm   rn   r!   r
   r
   r   Úbad_requestþ   s     ÿrq   )N)"rB   rg   Zdjango.httpr   Zdjango.utils.encodingr   Zdjango.utils.translationr   rN   r   Zrest_frameworkr   Z'rest_framework.utils.serializer_helpersr   r	   r   r%   r*   r:   r    Ú	ExceptionrD   rO   rR   rS   rU   rV   rW   rY   r^   ra   rd   ro   rq   r
   r
   r
   r   Ú<module>   s0   
!,

