|
@@ -15,6 +15,7 @@ import pandas as pd
|
|
|
|
|
|
from commom import f_read_file
|
|
|
from config import BaseConfig
|
|
|
+from strategy_parse import StrategyParse
|
|
|
from .manager import engine
|
|
|
|
|
|
DATA_SUB_DIR = "data"
|
|
@@ -61,12 +62,14 @@ def _get_upload_data(data) -> pd.ExcelFile:
|
|
|
excel = pd.ExcelFile(file_path)
|
|
|
return excel
|
|
|
|
|
|
-def _get_upload_data_path(data) ->str:
|
|
|
+
|
|
|
+def _get_upload_data_path(data) -> str:
|
|
|
base_dir = _get_base_dir(data)
|
|
|
save_path = os.path.join(base_dir, DATA_SUB_DIR)
|
|
|
file_path = _get_prefix_file(save_path, UPLOAD_DATA_PREFIX)
|
|
|
return file_path
|
|
|
|
|
|
+
|
|
|
def _get_node_func_dict(data) -> dict:
|
|
|
node_func_dict_path = _get_save_path(data, BaseConfig.node_map_name)
|
|
|
if not os.path.exists(node_func_dict_path):
|
|
@@ -178,11 +181,10 @@ def f_project_load(data):
|
|
|
}
|
|
|
|
|
|
return {
|
|
|
- engine.get_elem_by_id("data_upload"): gr.update(visible=False),
|
|
|
- engine.get_elem_by_id("code_view"): gr.update(visible=False),
|
|
|
- engine.get_elem_by_id("download_code"): gr.update(visible=False),
|
|
|
- }
|
|
|
-
|
|
|
+ engine.get_elem_by_id("data_upload"): gr.update(visible=False),
|
|
|
+ engine.get_elem_by_id("code_view"): gr.update(visible=False),
|
|
|
+ engine.get_elem_by_id("download_code"): gr.update(visible=False),
|
|
|
+ }
|
|
|
|
|
|
|
|
|
def f_code_generate(data, progress=gr.Progress(track_tqdm=True)):
|
|
@@ -200,7 +202,7 @@ def f_code_generate(data, progress=gr.Progress(track_tqdm=True)):
|
|
|
yield _reset_component_state()
|
|
|
|
|
|
# 清空储存目录
|
|
|
- # _clean_base_dir(data)
|
|
|
+ _clean_base_dir(data)
|
|
|
|
|
|
yield _reset_component_state()
|
|
|
|
|
@@ -211,8 +213,8 @@ def f_code_generate(data, progress=gr.Progress(track_tqdm=True)):
|
|
|
try:
|
|
|
excel = _get_upload_data(data)
|
|
|
|
|
|
- # strategy_parse = StrategyParse(**all_param)
|
|
|
- # strategy_parse.f_parse_strategy(excel, progress)
|
|
|
+ strategy_parse = StrategyParse(**all_param)
|
|
|
+ strategy_parse.f_parse_strategy(excel, progress)
|
|
|
|
|
|
excel.close()
|
|
|
except Exception as msg:
|