B
    CVvg•  ã               @   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-install-o3oxmrkh/Django/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_name)r   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   Ú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.
        )r,   r   r   Z	has_perms)r   r+   r   r   r   Úhas_permission^   s    z&PermissionRequiredMixin.has_permissionc                s$   |   ¡ s|  ¡ S tƒ j|f|ž|ŽS )N)r-   r   r#   r$   )r   r   r%   r&   )r   r   r   r$   e   s    z PermissionRequiredMixin.dispatch)	r   r   r    r!   r)   r,   r-   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.
        )r0   )r   r   r   r   Úget_test_funcv   s    z!UserPassesTestMixin.get_test_funcc                s*   |   ¡ ƒ }|s|  ¡ S tƒ j|f|ž|ŽS )N)r1   r   r#   r$   )r   r   r%   r&   Zuser_test_result)r   r   r   r$   |   s    
zUserPassesTestMixin.dispatch)r   r   r    r!   r0   r1   r$   r'   r   r   )r   r   r.   k   s   r.   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(   r.   r   r   r   r   Ú<module>   s   8!