o
    yNh                     @   s   d dl Z d dlZd dlZd dlmZ d dlmZ e  e j	de
e j	dde j	de j	de j	d	d
Zdd Zdd Zdd ZdS )    N)Decimal)load_dotenv
MYSQL_HOST
MYSQL_PORTi  
MYSQL_USERMYSQL_PASSWORDMYSQL_DATABASE)hostportuserpassworddatabasec                   C   s   t jjdi tS )N )mysql	connectorconnectcfgr   r   r   /var/www/html/ai_worker/db.pyconn   s    r   c                 C   sN   t  }| }d}||| f | }W d    |S 1 s w   Y  |S )NzSELECT f.id, f.label, f.unit, f.input_type, v.value_raw
               FROM business_case_values v
               JOIN business_case_factors f ON f.id=v.factor_id
               WHERE v.id_survey_item=%s)r   cursorexecutefetchall)	answer_id
connectionr   qrowsr   r   r   fetch_case_values   s   

r   c           	      C   sT   t  }| }|d| t||||||f W d    d S 1 s#w   Y  d S )NzREPLACE INTO business_case_results
                     (answer_id,kpi_json,narrative,pptx_url, chart_url, html_table, html_narrative)
                     VALUES (%s,%s,%s,%s, %s, %s, %s))r   r   r   jsondumps)	r   kpi_json	narrativepptx_url	chart_url
html_tablehtml_narrativer   r   r   r   r   insert_result   s   "r%   )osmysql.connectorr   r   decimalr   dotenvr   environgetintr   r   r   r%   r   r   r   r   <module>   s    



