File: //home/arjun/projects/env/lib/python3.10/site-packages/__pycache__/flask_bouncer.cpython-310.pyc
o
��!^� � @ s� d dl mZ d dlmZmZmZmZ d dlm Z d dl
m Z d dl m
Z
d dlT e dd� �Zd d
� Zd d � ZeZG d
d� de�Zdd� Zdd� ZG dd� de�ZdS )� ��wraps)�request�g� current_app�_app_ctx_stack)�
LocalProxy)� Forbidden)�Ability)�*c C s
t jd S )N�bouncer)r �
extensions� r r �F/home/arjun/projects/env/lib/python3.10/site-packages/flask_bouncer.py�<lambda>
s
r c C sN dt _t�� }t|�}t�� |_tj|_|� | |�r%d�
|| |�}t |��d S )NTz#{0} does not have {1} access to {2}) r � _authorized�_bouncer�get_current_userr
�get_authorization_method�authorization_method�
alias_actions�aliased_actions�cannot�formatr )�action�subject� current_user�ability�msgr r r �ensure s
�r c C s4 dt _t�� }t|�}t�� |_tj|_|� | |�S �NT)
r r r r r
r r r r �can)r r r r r r r r! s
r! c @ s e Zd Zdd� Zdd� ZdS )� Conditionc C s || _ || _d S �N�r r )�selfr r r r r �__init__* s
zCondition.__init__c C s t | j| j� d S r# )r r r �r% r r r �test. s zCondition.testN)�__name__�
__module__� __qualname__r& r( r r r r r" ( s r" c s � �fdd�}|S )Nc s" d� _ t� ��� �fdd��}|S )NTc s t � ���� �| i |��S r# )r" r( ��args�kwargs)r �fr r r �decorated_function6 s z7requires.<locals>.decorator.<locals>.decorated_function)�_explict_rule_setr �r/ r0 r$ �r/ r � decorator3 s zrequires.<locals>.decoratorr )r r r4 r r$ r �requires2 s r5 c s t � �� fdd��}|S )Nc s dt _� | i |��S r )r r r, r3 r r r0 ? s z.skip_authorization.<locals>.decorated_functionr r2 r r3 r �skip_authorization> s r6 c @ s� e Zd ZdZg d�Zd#dd�Zd#dd�Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zedd� �Zejdd� �Zdd � Zd d � Zd d � Zd!d"� ZdS )$�BouncerzYThis class is used to control the Abilities Integration to one or more Flask applications)�get�put�patch�post�delete�indexNc K sV d | _ | �� | _d | _t� | _t� | _| j| _d | _ |d ur)| j
|fi |�� d S d S r# ) � authorization_method_callback�default_alias_actions�_alias_actions�_authorization_method�list�flask_classy_classes�
explict_rules�default_user_loaderr �app�init_app�r% rF r. r r r r& L s
�zBouncer.__init__c C s8 |dur|S | j dur| j S tj}|dur|j S td��)zBHelper method that implements the logic to look up an application.NzZApplication not registered on Bouncer instance and no application bound to current context)rF �stack�top� RuntimeError)r% �
reference_app�ctxr r r �get_appZ s
zBouncer.get_appc K s> || _ | �� | j �| j� |�dd�r | j �| j� dS dS )zv Initializes the Flask-Bouncer extension for the specified application.
:param app: The application.
�ensure_authorizationFN)rF �_init_extension�before_request�check_implicit_rulesr8 �
after_request�check_authorizationrH r r r rG l s �zBouncer.init_appc C s& t | jd�s t� | j_| | jjd< d S )Nr
r )�hasattrrF �dictr
r' r r r rP z s
zBouncer._init_extensionc C s t td�st�tjs t�|S )zBchecks that an authorization call has been made during the requestr )rU r r r )r% �responser r r rT � s
zBouncer.check_authorizationc sT | � � sdS | �� r dS tj�d�\� }� fdd�| jD �d }t||j��� dS )z� if you are using flask classy are using the standard index,new,put,post, etc ... type routes, we will
automatically check the permissions for you
N�:c s g | ] }|j � kr|�qS r �r) ��.0� classy_class��
class_namer r �
<listcomp>� s z0Bouncer.check_implicit_rules.<locals>.<listcomp>r ) �"request_is_managed_by_flask_classy� method_is_explictly_overwrittenr �endpoint�splitrC r" �__target_model__r( )r% r �clazzr r] r rR � s z Bouncer.check_implicit_rulesc C s t jtj }t|d�o|jdu S )Nr1 T)r �view_functionsr rb rU r1 )r% � view_funcr r r ra � s z'Bouncer.method_is_explictly_overwrittenc sN t jd u rdS dt jvrdS t j�d�\� }t� fdd�| jD ��o&|| jv S )NFrX c 3 s � | ]}� |j kV qd S r# rY rZ r] r r � <genexpr>� s � z=Bouncer.request_is_managed_by_flask_classy.<locals>.<genexpr>)r rb rc �anyrC �special_methods)r% r r r] r r` � s
�z*Bouncer.request_is_managed_by_flask_classyc C s( t td�rtjS t td�rtjS td��)Nr �userz#Excepting current_user on flask's g)rU r r rk � Exceptionr' r r r rE � s
zBouncer.default_user_loaderc C s
|| _ |S )zP
Use this method decorator to overwrite the default user loader
)r �r% �valuer r r � user_loader� s zBouncer.user_loaderc C s | j S r# �r@ r' r r r r � s zBouncer.alias_actionsc C s
|| _ dS )z$if you want to override your actionsNrp rm r r r r � s
c C s t tttgttttgtt t
giS r# ) �READ�INDEX�SHOW�GET�CREATE�NEW�PUT�POST�UPDATE�EDIT�PATCHr' r r r r? � s
�z Bouncer.default_alias_actionsc G s | j �|� d S r# )rC �extend)r% �
classy_routesr r r �monitor� s zBouncer.monitorc C s || _ | j S )z:
the callback for defining user abilities
)rA rm r r r r � s z Bouncer.authorization_methodc C s | j d ur| j S td��)Nz'Expected authorication method to be set)rA rl r' r r r r � s
z Bouncer.get_authorization_methodr# )r) r* r+ �__doc__rj r& rN rG rP rT rR ra r` rE ro �propertyr �setterr? r~ r r r r r r r7 F s*
r7 N)� functoolsr �flaskr r r r rI �werkzeug.localr �werkzeug.exceptionsr r r
�bouncer.constantsr r r! �bounce�objectr" r5 r6 r7 r r r r �<module> s