�
���hM � � � d dl Z d dlZd dlZd dlmZ d dlmZ d� Zdd�Ze dk( r% e� dZ
ee
� \ Z Z e
ee
e e � � yy) � N)�Error)� load_dotenvc
� � d}d} t j j t j d� t j d� t j d� t j d� t j dd� �� }|j � r�|j
� }|j | � | j � j � j d � rT|j � }||j f|�|j � |�"|j � r|j � S S S |j � |j df|�|j � |�"|j � r|j � S S S |�|j � |�"|j � r|j � yyy# t $ rN}t# d
|� �� Y d}~|�|j � |�"|j � r|j � y y y d}~ww xY w# |�|j � |�"|j � r|j � w w w xY w) a
Connects to a MySQL database and executes a query.
Parameters:
- query (str): SQL query to execute
Returns:
- For SELECT queries: List of tuples containing the results
- For other queries: Number of rows affected
- None if an error occurs
N�DB_HOST�DB_NAME�DB_USER� DB_PASSWORD�DB_PORTi� )�host�database�user�password�port�SELECTz!Error while connecting to MySQL: )NN)�mysql� connector�connect�os�getenv� is_connected�cursor�execute�strip�upper�
startswith�fetchall� description�close�commit�rowcountr �print)�query�
connectionr �result�es �,/mnt/c/Users/abbas/Documents/MCPServer/db.py�execute_mysql_queryr' s% � � �J�
�F�# ��_�_�,�,����9�%��Y�Y�y�)����9�%��Y�Y�}�-����9�d�+� -�
�
� � "� "� $� �&�&�(�F�
�N�N�5� !� �{�{�}�"�"�$�/�/��9� � � �*�� �v� 1� 1�1� � � �L�L�N� � !�j�&=�&=�&?� � � � � '@� !� �!�!�#� ����,� � � �L�L�N� � !�j�&=�&=�&?� � � � � '@� !�1 %�, � � �L�L�N� � !�j�&=�&=�&?� � � � � '@� !�� � �
�1�!��5�6�� � � �L�L�N� � !�j�&=�&=�&?� � � � � '@� !���� � � �L�L�N� � !�j�&=�&=�&?� � � � � '@� !�s0 �C?G �< G � H �H� H" �H � H" �"8Ic � � | j � j � j � d }| |t j j � j � dd�}|dk( rn|s d|d< g |d< |S |D �cg c] }|d �� }}g } |D ]& }
| j t t ||
� � � �( |j | t |� d d
�� |S |d v r|j ||� d �d
�� |S |j |� d�dd�� |S c c}w # t $ r0 t t |d � � D �cg c] }d|� ��� nc c}w }}Y ��w xY w)aa
Formats MySQL query results for optimal LLM comprehension.
Args:
query (str): The SQL query that was executed
results (list): The query results (for SELECT)
rowcount (int): Number of rows affected (for INSERT/UPDATE/DELETE)
Returns:
dict: A structured response with metadata and formatted data
r �success)r" �type� timestamp�statusr z3Query executed successfully but returned no results�message�results�column_z5Each result is a dictionary with column names as keys)r. �count�format)�INSERT�UPDATE�DELETEz! operation completed successfully)�
rows_affectedr- z operation executed successfullyz(No results to return for this query type)r- �note)
r r �split�datetime�now� isoformat�AttributeError�range�len�append�dict�zip�update) r" r. r r �
query_type�response�desc�columns�i�formatted_results�rows r& �format_results_for_llmrI ? s� � � ��� �$�$�&�,�,�.�q�1�J��� �&�&�*�*�,�6�6�8�� �H� �X� ��"W�H�Y� �"$�H�Y� ��O� F�+6�7�4�t�A�w�7�G�7� ��� >�C� � $� $�T�#�g�s�*;�%<� =� >� ���(���\�M�
� �$ �O�
�5� 5���� %�$� �%F�G�
� � �O� ���$� �%E�F�>�
� �
�O��9 8�� � F�.3�C���
�O�.D�E�����}�E��E�G�E� F�s0 �5D �9 D
�D �
D �"E�1 D>�=E�E�__main__z�INSERT INTO cars (brand, model, color, `condition`, mileage, price, owner_email)
VALUES
('Toyota', 'Corola', 'Silver', 'used', 45000, 18500.00, 'john.dee@example.com'),)N)r8 r �mysql.connectorr r �dotenvr r' rI �__name__r" r. r r! � � r& �<module>rP s_ �� � � � !� �2 �n9�z �z���M�
X�E� /�u�5��G�[� �
��w� �
<�=� rO