File: //home/arjun/projects/buyercall/buyercall/blueprints/agents/__pycache__/serializers.cpython-310.pyc
o
we� � @ s� d dl mZ d dlmZmZ d dlmZmZ d dlm Z d dl
m Z d dl m
Z
G dd� de j�ZG d d
� d
e j�ZG d d � d e j�ZG d
d� de j�ZdS )� )�fields)�Agent�Team)� Partnership�PartnershipAccount)�User)�ma)�generate_presigned_file_urlc @ s, e Zd Ze�d�ZG dd� d�Zdd� ZdS )�AgentWithoutTeamSchema�get_agent_rolec @ � e Zd ZeZg d�ZdS )zAgentWithoutTeamSchema.Meta)�sid� firstname�lastname� phonenumber�email�roleN��__name__�
__module__� __qualname__r �modelr � r r �I/home/arjun/projects/buyercall/buyercall/blueprints/agents/serializers.py�Meta
� r c C s |j jS �N��userr ��self�objr r r r s z%AgentWithoutTeamSchema.get_agent_roleN)r r r r �Methodr r r r r r r r
s
r
c @ s\ e Zd Ze�d�Ze�d�Zejdd� dd�Ze j
dd�Z G d d
� d
�Z d d � Z
d
d� ZdS )�
TeamSchema�get_partnership_account_id�get_partnership_idc C s
t dd�S �NT)�many)r
r r r r �<lambda> s
zTeamSchema.<lambda>T)� dump_only�name)� attributec @ r )zTeamSchema.Meta)�id� is_activer* N)r r r r r �excluder r r r r r r c C � |r t j|jd�S d S �N)r, )r �get_sid_from_id�partnership_idr r r r r% � z TeamSchema.get_partnership_idc C r/ r0 )r r1 �partnership_account_idr r r r r$ " r3 z%TeamSchema.get_partnership_account_idN)r r r r r" r4 r2 �Nested�membersr �String�teamNamer r% r$ r r r r r# s
r# c @ s e Zd ZG dd� d�ZdS )� AgentSchemac @ r )zAgentSchema.Meta)r, � extension�is_group�agentsN)r r r r r r. r r r r r ' r r N)r r r r r r r r r9 & s r9 c @ sn e Zd Ze�d�Ze�d�Zejdd�Z e�d�Z
e�d�Z G dd� d�Z d d
� Z
d d � Zd
d� Zdd� ZdS )�AgentMiniSchemar � get_teamsF)� dump_default�
get_avatar� get_user_idc @ r )zAgentMiniSchema.Meta)
r
r r r r r �teams�isTeam�
userAvatar�userIdNr r r r r r 4 r r c C s |j r|j jS dS )N� r r r r r r 8 s z AgentMiniSchema.get_agent_rolec C s |j r tdd��|j �S g S r& )rB r# �dumpr r r r r> ; s zAgentMiniSchema.get_teamsc
C sX t �|j�}|r*z|j�d�\}}t||�W S ty) } zW Y d }~dS d }~ww dS )Nz::rF )r � get_by_id�user_id� user_avatar�splitr � Exception)r r! r � bucket_name�key�er r r r@ > s
��zAgentMiniSchema.get_avatarc C s t �|j�S r )r r1 rI r r r r rA H s zAgentMiniSchema.get_user_idN)r r r r r" r rB r �BooleanrC rD rE r r r> r@ rA r r r r r= , s
r= N)� marshmallowr �"buyercall.blueprints.agents.modelsr r �'buyercall.blueprints.partnership.modelsr r � buyercall.blueprints.user.modelsr �buyercall.extensionsr �buyercall.lib.util_boto3_s3r �SQLAlchemyAutoSchemar
r# r9 r= r r r r �<module> s