File: //home/arjun/projects/buyercall/buyercall/lib/__pycache__/util_rest.cpython-310.pyc
o
we� � @ s� d dl mZ d dlmZmZmZmZ d dlm Z d dl
m Z m Z e � Z
e dd� �Ze dd� �Ze dd� �Ze d d� �Ze d
d� �Zg d �Zd d
� Zdd� Zdd� Zdd� Zi dddfdd�Zdd� Zdd � ZG d d � d �Zd S ) � ��wraps)�
make_response�request�jsonify� current_app)� current_user)�Local�
LocalProxyc C � t jS �N)�_local�partnership_account� r r �9/home/arjun/projects/buyercall/buyercall/lib/util_rest.py�<lambda>
� r c C r r )r
�is_partnership_accountr r r r r r c C r r )r
�is_partnershipr r r r r r c C r r )r
� partnershipr r r r r r c C r r )r
� api_token_idr r r r r r )�DELETE�POST�PUTc s dd� � � �fdd�}|S )Nc S s$ t | d d�}d�|| �|jd< |S )Nz
� �<Bearer realm="buyercall", error="{}", error_description="{}"�WWW-Authenticate)r �format�headers)�text�code�responser r r �
make_error s
�z rest_method.<locals>.make_errorc
s� ddl m}m}m} tj�dd��d�}t|�dks |d dkr%� dd �S |d
� � }|�
|�}|rR|� |�}|t _
|r@|t _n|�|�} | d urM| t _n
� d d �S � d d �S �| i |��S )
Nr ��PartnershipAccount�ApiToken� Partnership� Authorize� � � �Bearerz Invalid authorization method�invalid_request� zInvalid credentials�
invalid_token)�'buyercall.blueprints.partnership.modelsr$ r% r& r r �get�split�len�strip� check_token� get_partnership_account_by_tokenr
r r �get_partnership_by_tokenr )
�args�kwargsr$ r% r& �header�tokenr r r �r" �wrappedr r �wrapper s. �
��z rest_method.<locals>.wrapperr )r< r= r r; r � rest_method s r> c C s, ddi}t |�}d|_d�|| �|jd< |S )N�messagezAuthentication error.r r r )r � status_coder r )r r r? �respr r r � authenticate@ s �rB c C s ddi} t | �}d|_|S )Nr? zSError processing request. POST, PUT and DELETE requests are disabled in DEBUG mode.r )r r@ )r? rA r r r �unsupported_in_debug_modeL s rC c � t � �� fdd��}|S )Nc s ddl m}m}m} dt_dt_tj� dd��
d�}tj� dd�� � }t j
� dd�du r8tjtv r8|d kr8t� S t|�d
ksD|d d krItd d
�S |d �� }|�|�}|d ur{|�|�} |t_| d urk| t_dt_n|�|�}
|
d urz|
t_dt_ntdd�S � | i |��S )Nr r# Fr' r( r) �
Documentation�DEBUG�truer* r+ z Invalid authorization method.r, r- TzInvalid credentials.r. )r/ r$ r% r& r
r r r r r0 r1 �lower�app�config�method� blocked_debug_request_methodsrC r2 rB r3 r4 r5 r r r6 r ) r7 r8 r$ r% r& r9 �
api_doc_debugr: r � retrieved_partnership_account�retrieved_partnership��fr r � decoratedU s4
�
z requires_auth.<locals>.decoratedr )rQ rR r rP r �
requires_authT s #rS �� r( Tc C s t | |||d��S )N��datar@ r? �success)r rU r r r � api_jsonify| s �rX c rD )Nc s" t js
tdddd�S � | i |��S )Nr � UnauthorizedF�r@ r? rW �r �is_authenticatedrX )r7 r8 rP r r �decorated_function� s z)auth_required.<locals>.decorated_functionr )rQ r] r rP r �
auth_required� s r^ c rD )z
is there a valid user logged in?
:param *func: The function which the decorator being used
:return: Function
c s. t jsd}d}d}t|||d�S � | i |��S )Ni� Fz3You do not have permission to access this resource.rZ r[ )r7 r8 r@ rW r? ��funcr r r] � s z.api_login_required.<locals>.decorated_functionr )r` r] r r_ r �api_login_required� s ra c @ s e Zd Zddd�ZdS )�BlueprintMixinFNc K s� z3|r |dv r | j �tj�� d��|�� d�dd|i|��W S | j �tj�� | j d �dd|i|��W S | j dd|i|�� Y S )N)�in�outr0 rc �many�defaultr )� serializersr0 r rK rH )�selfre �keyr8 r r r �get_serializer_class� s 0*z#BlueprintMixin.get_serializer_class)FN)�__name__�
__module__� __qualname__rj r r r r rb � s rb N) � functoolsr �flaskr r r r rI � flask_loginr �werkzeug.localr r
r
�rest_partnership_account�rest_is_partnership_account�rest_is_partnership�rest_partnership�rest_api_token_idrL r> rB rC rS rX r^ ra rb r r r r �<module> s& * (