BlessedRemzy/nostalgia_persuasive_system_backend
0
1�
2 fi� � �� � d dl Z d dlZd dlmZmZmZ d dlmZ d dlm Z e e3� � j j Z e dz Z
ee
� � e j dd� � Zd� Zd ed4eeef fd �ZdS ) � N)�Optional�Dict�Any)� load_dotenv)�Pathz.env� DATABASE_URLz3postgresql://postgres:postgres@localhost:5432/myappc �4 � t j t � � S )N)�psycopg2�connectr � � �YC:\Users\USER\Documents\Code\nostalgic-persuasive-model-final\fastapi-backend\db_utils.py�get_db_connectionr s � � � �L� )� )�)r
�user_id�returnc �P � t � � }|� � � } |� d| f� � |� � � }|rR|d }|d }|�t |� � nd|r|ndd�|� � � |� � � S n)# t $ r }t d|� �� � Y d}~nd}~ww xY w|� � � |� � � n-# |� � � |� � � w xY wddd�S ) a 5 Fetch the latest stress and emotion context for a user from daily_habit_logs.6 7 Args:8 user_id: The user's ID9 10 Returns:11 Dict with 'stress_score' (float) and 'emotion' (str)12 Defaults to neutral if no log found.13 z�14 SELECT stress_level, emotion 15 FROM daily_habit_logs 16 WHERE user_id = %s 17 ORDER BY date DESC, created_at DESC 18 LIMIT 1;19 r � Ng �?�neutral)� stress_score�emotionz Error fetching latest context: )r �cursor�execute�fetchone�float�close� Exception�print)r �connr �row�stressr �es r �fetch_latest_contextr" sY � � � � �D�
�[�[�]�]�F�� ��� � �Z�
� � � �o�o� � �� � ���V�F��!�f�G�17�1C��f�
�
�
��&-�<�7�7�9�� � � � ���� �20�21� � � � � �� � 5� 5� 5�
�3��3�3�4�4�4�4�4�4�4�4�����5���� � � ���� �22�23� � � � �� � � ���� �24�25� � � � ���� � � � � s0 �AB$ �#C6 �$26C27�.C� C6 �C28�29C6 �6*D )�osr30 �typingr r r �dotenvr �pathlibr �__file__�parent� PROJECT_ROOT�ENV_FILE�getenvr r �strr" r r
r �<module>r- s� �� � � � � ���� &� &� &� &� &� &� &� &� &� &� � � � � � � � � � � � � �t�H�~�~�$�+� � �&� �� � �H� � � ��r�y� �)^�_�_� �*� *� *�-�#� -�$�s�C�x�.� -� -� -� -� -� -r