This commit is contained in:
Zyy554
2025-11-30 14:50:27 +08:00
parent e4c411bbe2
commit 52e2f4dcb1
41 changed files with 5767 additions and 27 deletions

12
app.py
View File

@@ -12,13 +12,17 @@ from simulation_model import SimulationModel
matplotlib.rcParams["font.family"] = ["Microsoft YaHei", "SimHei", "sans-serif"]
matplotlib.rcParams["axes.unicode_minus"] = False
# set year
year = json.load(open('year.json', 'r', encoding='utf-8'))['year']
filename = f"{year}"
# Load default parameters from data/model_params.json
with open("data/model_params.json", "r", encoding="utf-8") as f:
with open(f"data/{filename}/model_params.json", "r", encoding="utf-8") as f:
_cfg = json.load(f)
def _load_factory_name_map():
try:
with open("data/factory_mapping.json", "r", encoding="utf-8") as f:
with open(f"data/{filename}/factory_mapping.json", "r", encoding="utf-8") as f:
data = json.load(f)
except FileNotFoundError:
return {}
@@ -31,9 +35,9 @@ FACTORY_NAME_MAP = _load_factory_name_map()
def _load_product_catalog():
try:
try:
df = pd.read_csv("data/product.csv", encoding="utf-8")
df = pd.read_csv(f"data/{filename}/product.csv", encoding="utf-8")
except UnicodeDecodeError:
df = pd.read_csv("data/product.csv", encoding="gbk")
df = pd.read_csv(f"data/{filename}/product.csv", encoding="gbk")
except FileNotFoundError:
return pd.DataFrame()
df.columns = [c.strip() for c in df.columns]