0209review: firm risk component + 2way interation on n_max_trial and cap_limit_level

analysis_sum_result.py

@@ -1,4 +1,5 @@
 from orm import engine
+from sqlalchemy import text
 import pandas as pd
 import networkx as nx
 import json
@@ -12,8 +13,8 @@ BomNodes = pd.read_csv('BomNodes.csv', index_col=0)
 
 with open('SQL_analysis_risk.sql', 'r') as f:
     str_sql = f.read()
-result = pd.read_sql(sql=str_sql,
-                     con=engine)
+with engine.connect() as conn:
+    result = pd.read_sql(sql=text(str_sql), con=conn)
 result.to_csv('analysis\\count.csv',
               index=False,
               encoding='utf-8-sig')
@@ -23,10 +24,11 @@ print(result)
 plt.rcParams['font.sans-serif'] = 'SimHei'
 
 exp_id = 1
-G_bom_str = pd.read_sql(
-    sql=f'select g_bom from iiabmdb.without_exp_experiment '
-        f'where id = {exp_id};',
-    con=engine)['g_bom'].tolist()[0]
+with engine.connect() as conn:
+    G_bom_str = pd.read_sql(
+        sql=text(f'select g_bom from iiabmdb.without_exp_experiment '
+                 f'where id = {exp_id};'),
+        con=conn)['g_bom'].tolist()[0]
 G_bom = nx.adjacency_graph(json.loads(G_bom_str))
 pos = nx.nx_agraph.graphviz_layout(G_bom, prog="twopi", args="")
 node_labels = nx.get_node_attributes(G_bom, 'Name')
@@ -44,9 +46,10 @@ plt.close()
 plt.rcParams['font.sans-serif'] = 'SimHei'
 
 sample_id = 1
-G_firm_str = pd.read_sql(
-    sql=f'select g_firm from iiabmdb.without_exp_sample where id = {exp_id};',
-    con=engine)['g_firm'].tolist()[0]
+with engine.connect() as conn:
+    G_firm_str = pd.read_sql(
+        sql=text(f'select g_firm from iiabmdb.without_exp_sample where id = {exp_id};'),
+        con=conn)['g_firm'].tolist()[0]
 G_firm = nx.adjacency_graph(json.loads(G_firm_str))
 pos = nx.nx_agraph.graphviz_layout(G_firm, prog="twopi", args="")
 node_label = nx.get_node_attributes(G_firm, 'Name')