o
    cZh;                     @   s   d Z ddlZddlZddlmZmZ ddlmZ ddlm	Z
 ddlmZmZ dd ZG d	d
 d
eZG d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dS )z+
Provides various authentication policies.
    N)authenticateget_user_model)CsrfViewMiddleware)gettext_lazy)HTTP_HEADER_ENCODING
exceptionsc                 C   s&   | j dd}t|tr|t}|S )z
    Return request's 'Authorization:' header, as a bytestring.

    Hide some test client ickyness where the header can be unicode.
    ZHTTP_AUTHORIZATION    )METAget
isinstancestrencoder   )requestauth r   T/var/www/html/lang_env/lib/python3.10/site-packages/rest_framework/authentication.pyget_authorization_header   s   

r   c                   @   s   e Zd Zdd ZdS )	CSRFCheckc                 C   s   |S Nr   )selfr   reasonr   r   r   _reject   s   zCSRFCheck._rejectN)__name__
__module____qualname__r   r   r   r   r   r      s    r   c                   @       e Zd ZdZdd Zdd ZdS )BaseAuthenticationzF
    All authentication classes should extend BaseAuthentication.
    c                 C   s   t d)zS
        Authenticate the request and return a two-tuple of (user, token).
        z#.authenticate() must be overridden.)NotImplementedErrorr   r   r   r   r   r   &   s   zBaseAuthentication.authenticatec                 C   s   dS )z
        Return a string to be used as the value of the `WWW-Authenticate`
        header in a `401 Unauthenticated` response, or `None` if the
        authentication scheme should return `403 Permission Denied` responses.
        Nr   r   r   r   r   authenticate_header,   s   z&BaseAuthentication.authenticate_headerN)r   r   r   __doc__r   r   r   r   r   r   r   !   s    r   c                   @   s.   e Zd ZdZdZdd Zd
ddZdd	 ZdS )BasicAuthenticationz>
    HTTP Basic authentication against username/password.
    apic              
   C   s   t | }|r|d  dkrdS t|dkr!td}t|t|dkr0td}t|z'zt|d 	d}W n t
yP   t|d 	d	}Y nw |d
}W n tt
tjfyk   td}t|w |d |d }}| |||S )z
        Returns a `User` if a correct username and password have been supplied
        using HTTP Basic authentication.  Otherwise returns `None`.
        r   s   basicN   z.Invalid basic header. No credentials provided.   zCInvalid basic header. Credentials string should not contain spaces.zutf-8zlatin-1:z?Invalid basic header. Credentials not correctly base64 encoded.)r   splitlowerlen_r   AuthenticationFailedbase64	b64decodedecodeUnicodeDecodeError	partition	TypeErrorbinasciiErrorauthenticate_credentials)r   r   r   msgZauth_decodedZ
auth_partsuseridpasswordr   r   r   r   ;   s,   


z BasicAuthentication.authenticateNc                 C   sT   t  j|d|i}tdd|i|}|du rttd|js&ttd|dfS )z
        Authenticate the userid and password against username and password
        with optional request for context.
        r6   r   NzInvalid username/password.User inactive or deleted.r   )r   ZUSERNAME_FIELDr   r   r*   r)   	is_active)r   r5   r6   r   credentialsuserr   r   r   r3   Y   s   z,BasicAuthentication.authenticate_credentialsc                 C   s
   d| j  S )NzBasic realm="%s")www_authenticate_realmr   r   r   r   r   l   s   
z'BasicAuthentication.authenticate_headerr   )r   r   r   r    r;   r   r3   r   r   r   r   r   r!   5   s    
r!   c                   @   r   )SessionAuthenticationz<
    Use Django's session framework for authentication.
    c                 C   s.   t |jdd}|r|jsdS | | |dfS )z{
        Returns a `User` if the request session currently has a logged in user.
        Otherwise returns `None`.
        r:   N)getattr_requestr8   enforce_csrfr   r   r:   r   r   r   r   u   s
   

z"SessionAuthentication.authenticatec                 C   s@   dd }t |}|| ||ddi }|rtd| dS )zK
        Enforce CSRF validation for session based authentication.
        c                 S   s   d S r   r   )r   r   r   r   dummy_get_response   s   z>SessionAuthentication.enforce_csrf.<locals>.dummy_get_responseNr   zCSRF Failed: %s)r   process_requestZprocess_viewr   ZPermissionDenied)r   r   rA   checkr   r   r   r   r?      s   
z"SessionAuthentication.enforce_csrfN)r   r   r   r    r   r?   r   r   r   r   r<   p   s    r<   c                   @   s:   e Zd ZdZdZdZdd Z	 dd Zdd	 Zd
d Z	dS )TokenAuthenticationa  
    Simple token based authentication.

    Clients should authenticate by passing the token key in the "Authorization"
    HTTP header, prepended with the string "Token ".  For example:

        Authorization: Token 401f7ac837da42b97f613d789819ff93537bee6a
    TokenNc                 C   s    | j d ur| j S ddlm} |S )Nr   )rE   )modelZrest_framework.authtoken.modelsrE   )r   rE   r   r   r   	get_model   s   
zTokenAuthentication.get_modelc                 C   s   t | }|r|d  | j  krd S t|dkr&td}t|t|dkr5td}t|z|d 	 }W n t
yM   td}t|w | |S )Nr   r#   z.Invalid token header. No credentials provided.r$   z=Invalid token header. Token string should not contain spaces.zIInvalid token header. Token string should not contain invalid characters.)r   r&   r'   keywordr   r(   r)   r   r*   r-   UnicodeErrorr3   )r   r   r   r4   tokenr   r   r   r      s    



z TokenAuthentication.authenticatec                 C   s`   |   }z|jdj|d}W n |jy   ttdw |jj	s+ttd|j|fS )Nr:   )keyzInvalid token.r7   )
rG   objectsZselect_relatedr
   ZDoesNotExistr   r*   r)   r:   r8   )r   rK   rF   rJ   r   r   r   r3      s   
z,TokenAuthentication.authenticate_credentialsc                 C   s   | j S r   )rH   r   r   r   r   r      s   z'TokenAuthentication.authenticate_header)
r   r   r   r    rH   rF   rG   r   r3   r   r   r   r   r   rD      s    	rD   c                   @   s   e Zd ZdZdZdd ZdS )RemoteUserAuthenticationa  
    REMOTE_USER authentication.

    To use this, set up your web server to perform authentication, which will
    set the REMOTE_USER environment variable. You will need to have
    'django.contrib.auth.backends.RemoteUserBackend in your
    AUTHENTICATION_BACKENDS setting
    ZREMOTE_USERc                 C   s0   t ||j| jd}|r|jr|d fS d S d S )N)r   Zremote_user)r   r	   r
   headerr8   r@   r   r   r   r      s   
z%RemoteUserAuthentication.authenticateN)r   r   r   r    rN   r   r   r   r   r   rM      s    rM   )r    r+   r1   Zdjango.contrib.authr   r   Zdjango.middleware.csrfr   Zdjango.utils.translationr   r)   Zrest_frameworkr   r   r   r   r   r!   r<   rD   rM   r   r   r   r   <module>   s    ;'?