second
This commit is contained in:
12
app.py
12
app.py
@@ -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]
|
||||
|
||||
Reference in New Issue
Block a user