File: //home/arjun/projects/buyercall/buyercall/lib/__pycache__/db_seed.cpython-310.pyc
o
�we� � @ s� d dl Z d dlmZ d dlmZ d dlmZ d dlmZ zd dl m
Z
e
j fZ e
j Z W n' e
y= e �d� ed� Y n eyR d dlm
Z
e
j fZ e
j Z Y nw d d
� Zd d � Zd
d� Zdd� ZdS )� N)�User)�Report)� Partnership)�db)�settingsz:Ensure __init__.py and settings.py both exist in instance/� c C s2 t �t�dur dS dtddd�} t di | ���� S )z>
Create an admin account.
:return: User instance
N�sysadmin�password�SysAdmin)�role�emailr �company� )r �find_by_identity�SEED_ADMIN_EMAIL�save)�paramsr r �7/home/arjun/projects/buyercall/buyercall/lib/db_seed.py�create_sysadmin s �r c C s8 t j�t jdk��� du rt dd�} tj�| � dS dS )z
Create reports.
zList of InterestN)�name)r �query�filterr �firstr �session�add)�reportr r r �create_reports* s
�r c C s� t j�t jdk��� } | du r t � } d| _d| _tj�| � | durKt j�t j
dkt j � d���� du rMdt
dd| d�}t d i |��}tj�|� dS dS dS )
z!
Create partnership.
� BuyercallN�accountzBuyercall PartnershipF�partnerr )r r r r
� partnershipr )r r r r r �default_billing_typer r r r r
�is_deactivated�is_�SEED_PARTNER_EMAIL)r r r r r r �create_partnership3 s ,��r% c C s t � t� t� tj�� dS )zW
Entry point to seed the database with whatever we see fit.
:return: None
N)r r r% r r �commitr r r r �
seed_databaseK s
r' )�logging� buyercall.blueprints.user.modelsr �#buyercall.blueprints.reports.modelsr �'buyercall.blueprints.partnership.modelsr �buyercall.extensionsr �instancer r r$ � ImportError�error�exit�AttributeError�configr r r% r' r r r r �<module> s*
�