change the api base route by user defined (#17)

lxy1992 · Mar 3, 2024 · a2d8d86 · a2d8d86
1 parent 34a2c27
commit a2d8d86
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/commands.py b/commands.py
@@ -20,6 +20,7 @@
 db_config = config['database']
 bot_config = config['telegram_bot']
 TOKEN = bot_config['token']
+API = config["fund_api"]
 
 DATABASE_URL = db_config['url']
 # SSL参数
@@ -31,7 +32,7 @@
 
 
 class FundApi:
-    BASE_URL = 'https://api.doctorxiong.club/v1/fund'  # 请替换为实际的API基础URL
+    BASE_URL = API["base_url"]
 
     @staticmethod
     def search_funds(keyword):
@@ -193,6 +194,8 @@ async def get_daily_report(user_id, need_diagram=False):
                        FundDetail.day_growth).where(FundDetail.code == fund_code)
             )
             fund_detail = fund_detail.first()
+            if not fund_detail:
+                continue
             data = FundApi().get_real_time_fund([fund_code])
 
             # 计算估计的涨跌金额

diff --git a/config.yml b/config.yml
@@ -5,4 +5,4 @@ telegram_bot:
     token: ""
 
 fund_api:
-    base_url: "https://api.doctorxiong.club/v1/fund"
+    base_url: ""