U
    Ha•  ã                   @   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}|s"td | jj¡ƒ‚t|ƒS )zK
        Override this method to override the login_url attribute.
        zr{0} is missing the login_url attribute. Define {0}.login_url, settings.LOGIN_URL, or override {0}.get_login_url().)Ú	login_urlr   Z	LOGIN_URLr   ÚformatÚ	__class__Ú__name__Ústr)Úselfr   © r   ú>/tmp/pip-unpacked-wheel-3jxiddxt/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&   D   s    zLoginRequiredMixin.dispatch)r   r    r!   r"   r&   Ú__classcell__r   r   r*   r   r#   B   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   s<   | j dkrtd | jj¡ƒ‚t| j tƒr2| j f}n| j }|S )zv
        Override this method to override the permission_required attribute.
        Must return an iterable.
        Nz|{0} is missing the permission_required attribute. Define {0}.permission_required, or override {0}.get_permission_required().)Úpermission_requiredr   r   r   r   Ú
isinstancer   ©r   Zpermsr   r   r   Úget_permission_requiredN   s    
ÿÿ
z/PermissionRequiredMixin.get_permission_requiredc                 C   s   |   ¡ }| jj |¡S )zT
        Override this method to customize the way permissions are checked.
        )r0   r   r   Z	has_permsr/   r   r   r   Úhas_permission^   s    z&PermissionRequiredMixin.has_permissionc                    s$   |   ¡ s|  ¡ S tƒ j|f|ž|ŽS r$   )r1   r   r%   r&   r'   r*   r   r   r&   e   s    z PermissionRequiredMixin.dispatch)	r   r    r!   r"   r-   r0   r1   r&   r+   r   r   r*   r   r,   J   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.)ÚNotImplementedErrorr   r   r   r   r   r   r   Ú	test_funcq   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_funcv   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   r2   k   s   r2   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,   r2   r   r   r   r   Ú<module>   s   8!