U
    e,                     @   s   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lmZ G dd dZG d	d
 d
eZG dd deZG dd deZdS )    )urlparse)settings)REDIRECT_FIELD_NAME)redirect_to_login)ImproperlyConfiguredPermissionDenied)resolve_urlc                   @   s@   e Zd ZdZdZdZdZeZdd Z	dd Z
d	d
 Zdd ZdS )AccessMixinz^
    Abstract CBV mixin that gives access mixins the same customizable
    functionality.
    N Fc                 C   s>   | j p
tj}|s6t| jj d| jj d| jj dt|S )zK
        Override this method to override the login_url attribute.
        z, is missing the login_url attribute. Define z,.login_url, settings.LOGIN_URL, or override z.get_login_url().)	login_urlr   Z	LOGIN_URLr   	__class____name__str)selfr    r   >/tmp/pip-unpacked-wheel-lctamlir/django/contrib/auth/mixins.pyget_login_url   s     zAccessMixin.get_login_urlc                 C   s   | j S )z[
        Override this method to override the permission_denied_message attribute.
        )permission_denied_messager   r   r   r   get_permission_denied_message"   s    z)AccessMixin.get_permission_denied_messagec                 C   s   | j S )zU
        Override this method to override the redirect_field_name attribute.
        )redirect_field_namer   r   r   r   get_redirect_field_name(   s    z#AccessMixin.get_redirect_field_namec                 C   s   | j s| jjjrt|  | j }t|  }t	|d d \}}t	|d d \}}|rf||kr||rr||kr|| j
 }t|||  S )N   )raise_exceptionrequestuseris_authenticatedr   r   Zbuild_absolute_urir   r   r   Zget_full_pathr   r   )r   pathZresolved_login_urlZlogin_schemeZlogin_netlocZcurrent_schemeZcurrent_netlocr   r   r   handle_no_permission.   s"    

z AccessMixin.handle_no_permission)r   
__module____qualname____doc__r   r   r   r   r   r   r   r   r   r   r   r   r   r	   
   s   r	   c                       s    e Zd ZdZ fddZ  ZS )LoginRequiredMixinz.Verify that the current user is authenticated.c                    s$   |j js|  S t j|f||S N)r   r   r   superdispatchr   r   argskwargsr   r   r   r%   F   s    zLoginRequiredMixin.dispatch)r   r   r    r!   r%   __classcell__r   r   r)   r   r"   C   s   r"   c                       s4   e Zd ZdZdZdd Zdd Z fddZ  ZS )	PermissionRequiredMixinz;Verify that the current user has all specified permissions.Nc                 C   sP   | j dkr0t| jj d| jj d| jj dt| j trF| j f}n| j }|S )zv
        Override this method to override the permission_required attribute.
        Must return an iterable.
        Nz6 is missing the permission_required attribute. Define z".permission_required, or override z.get_permission_required().)permission_requiredr   r   r   
isinstancer   r   Zpermsr   r   r   get_permission_requiredQ   s    
 
z/PermissionRequiredMixin.get_permission_requiredc                 C   s   |   }| jj|S )zT
        Override this method to customize the way permissions are checked.
        )r/   r   r   Z	has_permsr.   r   r   r   has_permissionc   s    z&PermissionRequiredMixin.has_permissionc                    s$   |   s|  S t j|f||S r#   )r0   r   r$   r%   r&   r)   r   r   r%   j   s    z PermissionRequiredMixin.dispatch)	r   r   r    r!   r,   r/   r0   r%   r*   r   r   r)   r   r+   L   s
   r+   c                       s0   e Zd ZdZdd Zdd Z fddZ  ZS )UserPassesTestMixinz]
    Deny a request with a permission error if the test_func() method returns
    False.
    c                 C   s   t d| jjd S )Nz;{} is missing the implementation of the test_func() method.)NotImplementedErrorformatr   r   r   r   r   r   	test_funcv   s
    zUserPassesTestMixin.test_funcc                 C   s   | j S )zK
        Override this method to use a different test_func method.
        )r4   r   r   r   r   get_test_func}   s    z!UserPassesTestMixin.get_test_funcc                    s*   |    }|s|  S t j|f||S r#   )r5   r   r$   r%   )r   r   r'   r(   Zuser_test_resultr)   r   r   r%      s    
zUserPassesTestMixin.dispatch)r   r   r    r!   r4   r5   r%   r*   r   r   r)   r   r1   p   s   r1   N)urllib.parser   Zdjango.confr   Zdjango.contrib.authr   Zdjango.contrib.auth.viewsr   Zdjango.core.exceptionsr   r   Zdjango.shortcutsr   r	   r"   r+   r1   r   r   r   r   <module>   s   9	$