database.cpython-311.pyc•2.2 kB
�
� �h: � �x � d dl mZmZmZ d dlmZmZmZmZm Z d dl
Z
dZ ee d�� � Z d� Z
d� Zd ed
efd �ZdS ) � )�SQLModel�
create_engine�Session)�User� TrackedQuery�Run�RunItem� AggregateNz sqlite:///./visibility.sqliteF)�echoc �N � t j � t � � dS )zCreate database and tablesN)r �metadata�
create_all�engine� � �F/Users/kulraj/agent09/mcp-starter/mcp-bearer-token/../data/database.py�create_db_and_tablesr s � � �� � ��(�(�(�(�(r c # �l K � t t � � 5 } | V � ddd� � dS # 1 swxY w Y dS )zGet database sessionN)r r )�sessions r � get_sessionr
s| � � � � �� � � �G��
�
�
�� � � � � � � � � � � ���� � � � � � s �)�-�-�phone�returnc � � t t � � 5 }|� t � � � t j | k � � � � � }|sNt | �� � }|� |� � |� � � |� |� � |cddd� � S # 1 swxY w Y dS )z!Ensure user exists, create if not)r N)
r r �queryr �filterr �first�add�commit�refresh)r r �users r �ensure_user_existsr! s� � � �� � � �G��}�}�T�"�"�)�)�$�*��*=�>�>�D�D�F�F��� "��e�$�$�$�D� �K�K�� � � � �N�N� � � � �O�O�D� !� !� !��� � � � � � � � � � � ���� � � � � � s �B C�C�C)�sqlmodelr r r �modelsr r r r r
�os� DATABASE_URLr r r �strr! r r r �<module>r' s� �� 5� 5� 5� 5� 5� 5� 5� 5� 5� 5� >� >� >� >� >� >� >� >� >� >� >� >� >� >� � � � � /� � � �|�%� 0� 0� 0��)� )� )�� � �
�c� �d� � � � � � r