feat: 优化数据库初始化函数命名

jobs/article_earning_ranking.py

@@ -11,9 +11,12 @@
     ArticleEarningRankingRecordDocument,
     ArticleField,
     EarningField,
-    init_db,
     insert_many,
 )
+from models.article_earning_ranking_record import (
+    init_db as init_article_earning_ranking_record_db,
+)
+from models.jianshu_user import init_db as init_jianshu_user_db
 from models.jianshu_user import insert_or_update_one
 from utils.async_retry import async_retry
 from utils.config_generators import (
@@ -81,7 +84,8 @@ async def process_item(
     )
 )
 async def flow_func() -> State:
-    await init_db()
+    await init_article_earning_ranking_record_db()
+    await init_jianshu_user_db()
 
     target_date = datetime.now().date() - timedelta(days=1)
 

jobs/assets_ranking.py

@@ -11,9 +11,12 @@
 from models.assets_ranking_record import (
     AmountField,
     AssetsRankingRecordDocument,
-    init_db,
     insert_many,
 )
+from models.assets_ranking_record import (
+    init_db as init_assets_ranking_record_db,
+)
+from models.jianshu_user import init_db as init_jianshu_user_db
 from models.jianshu_user import insert_or_update_one
 from utils.async_retry import async_retry
 from utils.config_generators import (
@@ -82,7 +85,8 @@ async def process_item(
     )
 )
 async def flow_func() -> State:
-    await init_db()
+    await init_assets_ranking_record_db()
+    await init_jianshu_user_db()
 
     target_date = datetime.now().date()
 

jobs/daily_update_ranking.py

@@ -10,9 +10,12 @@
 
 from models.daily_update_ranking_record import (
     DailyUpdateRankingRecordDocument,
-    init_db,
     insert_many,
 )
+from models.daily_update_ranking_record import (
+    init_db as init_daily_update_ranking_record_db,
+)
+from models.jianshu_user import init_db as init_jianshu_user_db
 from models.jianshu_user import insert_or_update_one
 from utils.config_generators import (
     generate_deployment_config,
@@ -43,7 +46,8 @@ async def process_item(
     )
 )
 async def flow_func() -> State:
-    await init_db()
+    await init_daily_update_ranking_record_db()
+    await init_jianshu_user_db()
 
     current_date = datetime.now().date()
 

jobs/jianshu_lottery.py

@@ -7,9 +7,12 @@
 from models.jianshu_lottery_win_record import (
     JianshuLotteryWinRecordDocument,
     get_latest_record_id,
-    init_db,
     insert_many,
 )
+from models.jianshu_lottery_win_record import (
+    init_db as init_jianshu_lottery_win_record_db,
+)
+from models.jianshu_user import init_db as init_jianshu_user_db
 from models.jianshu_user import insert_or_update_one
 from utils.config_generators import (
     generate_deployment_config,
@@ -42,7 +45,8 @@ async def process_item(
     )
 )
 async def flow_func() -> State:
-    await init_db()
+    await init_jianshu_lottery_win_record_db()
+    await init_jianshu_user_db()
 
     logger = get_run_logger()
 

jobs/lp_recommended_articles.py

@@ -5,13 +5,16 @@
 from prefect import flow, get_run_logger
 from prefect.states import Completed, State
 
+from models.jianshu_user import init_db as init_jianshu_user_db
 from models.jianshu_user import insert_or_update_one
 from models.lp_recommend_article_record import (
     LPRecommendedArticleRecord,
-    init_db,
     insert_many,
     is_record_stored,
 )
+from models.lp_recommend_article_record import (
+    init_db as init_lp_recommend_article_record_db,
+)
 from utils.config_generators import (
     generate_deployment_config,
     generate_flow_config,
@@ -61,7 +64,8 @@ async def process_item(
     )
 )
 async def flow_func() -> State:
-    await init_db()
+    await init_lp_recommend_article_record_db()
+    await init_jianshu_user_db()
 
     logger = get_run_logger()