node04 3 дней назад
Родитель
Сommit
0957586780
3 измененных файлов с 25 добавлено и 4 удалено
  1. 18 1
      config.py
  2. 4 2
      data_loader.py
  3. 3 1
      uo_atlas_import.py

+ 18 - 1
config.py

@@ -21,7 +21,7 @@ mongo_table_uo = "import_flights_uo_info_tab"
 
 
 # UO原始城市对
-uo_city_pairs = [
+uo_city_pairs_old = [
     "BJSBKK", "BJSCNX", "BJSDAD", "BJSHAN", "BJSHKG",
     "BJSKUL", "BJSMNL", "BJSOSA", "BJSPEN", "BJSSDJ",
     "BJSTPE", "BJSTYO", "BKIHKG", "BKKBJS", "BKKHKG",
@@ -48,3 +48,20 @@ uo_city_pairs = [
     "TYOHKG", "TYONGB", "TYOTPE", "YIWBKK", "YIWHKG",
 ]
 
+# UO近期有数据的城市对
+uo_city_pairs_new =[
+    "BJSHKG", "BKIHKG", "BKKHKG", "BKKNGB", "CJUHKG",
+    "CJUNGB", "CNXHKG", "CRKHKG", "CZXHAN", "CZXHKG",
+    "DADHKG", "FUKHKG", "HANHKG", "HANMNL", "HIJHKG",
+    "HKGBJS", "HKGBKI", "HKGBKK", "HKGCJU", "HKGCNX",
+    "HKGCRK", "HKGCZX", "HKGDAD", "HKGFUK", "HKGHAN",
+    "HKGHIJ", "HKGHKT", "HKGKHH", "HKGKUL", "HKGKWE",
+    "HKGMNL", "HKGNGB", "HKGNGO", "HKGOKA", "HKGOSA",
+    "HKGPEN", "HKGPQC", "HKGPUS", "HKGRMQ", "HKGSDJ",
+    "HKGSEL", "HKGSYX", "HKGTAE", "HKGTAK", "HKGTPE",
+    "HKGTYO", "HKTHKG", "KHHHKG", "KULHKG", "KWEHKG",
+    "KWEPEN", "MNLHKG", "NGBBKK", "NGBHKG", "NGOHKG",
+    "OKAHKG", "OSAHKG", "PENHKG", "PQCHKG", "PUSHKG",
+    "RMQHKG", "SDJHKG", "SELHKG", "SYXHKG", "TAEHKG",
+    "TAKHKG", "TPEBKK", "TPEHKG", "TYOHKG", "YIWHKG",
+]

+ 4 - 2
data_loader.py

@@ -11,7 +11,7 @@ import matplotlib.pyplot as plt
 from matplotlib import font_manager
 import matplotlib.dates as mdates
 from uo_atlas_import import mongo_con_parse
-from config import mongo_config, mongo_table_uo, uo_city_pairs
+from config import mongo_config, mongo_table_uo, uo_city_pairs_old, uo_city_pairs_new
 
 font_path = "./simhei.ttf"
 font_prop = font_manager.FontProperties(fname=font_path)
@@ -547,7 +547,9 @@ if __name__ == "__main__":
     os.makedirs(output_dir, exist_ok=True)
 
     from_date_begin = "2026-03-17"
-    from_date_end = "2026-04-01"
+    from_date_end = "2026-03-26"
+
+    uo_city_pairs = uo_city_pairs_new.copy()
 
     uo_city_pair_list = [f"{pair[:3]}-{pair[3:]}" for pair in uo_city_pairs]
 

+ 3 - 1
uo_atlas_import.py

@@ -5,7 +5,7 @@ from datetime import datetime
 import pymongo
 from pymongo.errors import PyMongoError, ServerSelectionTimeoutError, BulkWriteError
 # import pandas as pd
-from config import atlas_config, mongo_config, atlas_table, mongo_table_uo, uo_city_pairs
+from config import atlas_config, mongo_config, atlas_table, mongo_table_uo, uo_city_pairs_old, uo_city_pairs_new
 
 
 def import_flight_range_status(atlas_db, mongo_db, city_pair, create_at_begin_stamp, create_at_end_stamp, 
@@ -215,6 +215,8 @@ def main_import_process(create_at_begin, create_at_end):
     print(f"create_at_begin: {create_at_begin}, timestamp: {create_at_begin_stamp}")
     print(f"create_at_end: {create_at_end}, timestamp: {create_at_end_stamp}")
 
+    uo_city_pairs = uo_city_pairs_old.copy()
+
     for idx, city_pair in enumerate(uo_city_pairs):
         atlas_client, atlas_db = mongo_con_parse(atlas_config)
         mongo_client, mongo_db = mongo_con_parse(mongo_config)