？

lgc-NB2Dev · Jan 13, 2024 · a3acbf4 · a3acbf4
1 parent 4af3be8
commit a3acbf4
Show file tree

Hide file tree

Showing 2 changed files with 26 additions and 10 deletions.
diff --git a/scripts/base/const.py b/scripts/base/const.py
@@ -11,3 +11,10 @@
 DATA_PATH = ROOT_PATH / "data"
 TRAIN_FILE = DATA_PATH / "train.json"
 ALIAS_FILE = DATA_PATH / "alias.json"
+
+
+def const_ready():
+    if not TMP_PATH.exists():
+        TMP_PATH.mkdir(parents=True, exist_ok=True)
+    if not DATA_PATH.exists():
+        DATA_PATH.mkdir(parents=True, exist_ok=True)
diff --git a/scripts/pre_deploy/update_trains.py b/scripts/pre_deploy/update_trains.py
@@ -12,22 +12,31 @@ def convert_trains():
     }
     for train in train_data_lines:
         train_datas = train.split("\t")
-        train_ids = train_datas[6].split("#")
-        train_no = []
-        for train_id in train_ids:
-            train_id_split = train_id.split("/")
-            train_no.append(train_id_split[0])
-            if len(train_id_split) > 1:
-                train_no.append(
-                    train_id_split[0][: -(len(train_id_split[1]))] + train_id_split[1],
-                )
+        # train_ids = train_datas[6].split("#")
+        # train_no = []
+        # for train_id in train_ids:
+        #     train_id_split = train_id.split("/")
+        #     train_no.append(train_id_split[0])
+        #     if len(train_id_split) > 1:
+        #         train_no.append(
+        #             train_id_split[0][: -(len(train_id_split[1]))] + train_id_split[1],
+        #         )
+        # train_data.append(
+        #     {
+        #         "running_start": train_datas[0],
+        #         "running_end": train_datas[1],
+        #         "emu_note": train_datas[4],
+        #         "administrator": administrator_data[train_datas[5].lower()],
+        #         "train_no": train_no,
+        #     },
+        # )
         train_data.append(
             {
                 "running_start": train_datas[0],
                 "running_end": train_datas[1],
                 "emu_note": train_datas[4],
                 "administrator": administrator_data[train_datas[5].lower()],
-                "train_no": train_no,
+                "train_no": train_datas[6],
             },
         )
     TRAIN_FILE.write_text(json.dumps(train_data, ensure_ascii=False), "UTF-8")