MCP-Python

� Q��gT��v
�dd
lm
Z

ddlZddlZddlmZ
e�
ejdd�Zejdd�Zejdd �Z ejd d�Z ejdd �Zd�Ze
d�
Z e j�defd��Ze j�dedefd��Zy)�)
�FastMCPN)
�load_dotenv�DB_HOST�db�DB_PORT�5432�DB_NAME� mydatabase�DB_USER�postgres�DB_PASSWORD�passwordc�V�t
jd
tttt ��S)z!Return a new database connection.� localhost)�host�port�database�userr)�psycopg2�connectrr rr ���-/Users/rick/Projects/mcp-python/app/server.py�get_db_connectionrs$����� � �� ���rz Demo Donna�returnc�
� t
�}|j�5}
|
jd
�

|
j�}ddd�
|j �
t�
S#1s
w
Y

�$xYw
#t$r}d|��cYd}~Sd}~w
wxYw
)�~ Executes a SQL query and returns the result as a string. Example: query_db("SELECT * FROM my_table;") z*SELECT * FROM information_schema.schemata;N�Error: �r�cursor�execute�fetchall�close�str� Exception)�conn�cur�result�
es r� get_db_schemar*sq��� �"�� �[�[�]�c��K�K�D�E��\�\�^�F�� � � ���6�{�� �]�� �
����}���
��3�A-�"A!�"A-�!A*�&A-�- B�6B�:
B�B�queryc
�
� t
�}
|
j�5}|j|�

|j�}d
d
d
�
|
j �
t�
S#1s
w
Y

�$xYw
#t$r}d|��cYd
}~Sd
}~w
wxYw
)rNrr)r,r&r'r(r)s r�query_dbr./sp��� �"�� �[�[�]�c��K�K����\�\�^�F�� � � ���6�{�� �]�� �
����}���
�r+)�mcp.server.fastmcprr�os�dotenvr�getenvrrr rr r�mcp�toolr$r*r.rrr�<module>r5
s��


�&�� ��
� ��"�)�)�I�t� $�� �"�)�)�I�v� &�� �"�)�)�I�|� ,�� �"�)�)�I�z� *���b�i�i� �z�2��
��l���
����
�s�
��
�"����
�C�
�C�
��
r