feat: 支持配置 JKit 简书数据源 Endpoint

Author: FHU-yezi

backend/api/v1/articles.py

@@ -5,6 +5,7 @@
 from typing import Annotated
 
 from jkit.article import Article
+from jkit.config import CONFIG as JKIT_CONFIG
 from jkit.constants import ARTICLE_SLUG_REGEX
 from jkit.exceptions import ResourceUnavailableError
 from litestar import Response, Router, get
@@ -27,6 +28,9 @@
 )
 from utils.config import CONFIG
 
+if CONFIG.jianshu_endpoint:
+    JKIT_CONFIG.datasources.jianshu.endpoint = CONFIG.jianshu_endpoint
+
 splitter = WordSplitter(
     access_key_id=CONFIG.word_split_access_key.access_key_id,
     access_key_secret=CONFIG.word_split_access_key.access_key_secret,

backend/api/v1/users.py

@@ -3,6 +3,7 @@
 from datetime import datetime
 from typing import Annotated, Literal
 
+from jkit.config import CONFIG as JKIT_CONFIG
 from jkit.constants import USER_SLUG_REGEX
 from jkit.exceptions import ResourceUnavailableError
 from jkit.identifier_check import is_user_slug
@@ -25,6 +26,10 @@
 )
 from models.jianshu.lottery_win_record import LotteryWinRecord
 from models.jianshu.user import User as DbUser
+from utils.config import CONFIG
+
+if CONFIG.jianshu_endpoint:
+    JKIT_CONFIG.datasources.jianshu.endpoint = CONFIG.jianshu_endpoint
 
 MembershipTextType = Literal[
     "铜牌", "银牌", "金牌", "白金", "（旧版）普通会员", "（旧版）尊享会员"

backend/config.example.toml

@@ -1,3 +1,5 @@
+jianshu_endpoint = ""
+
 [jtools_postgres]
     host = "localhost"
     port = 5432

backend/main.py

@@ -49,6 +49,12 @@ class CustomOpenAPIController(OpenAPIController):
 )
 
 if __name__ == "__main__":
+    # TODO: 集中管理 JKit 简书数据源 Endpoint 配置
+    if CONFIG.jianshu_endpoint:
+        logger.debug("已成功配置 JKit 简书数据源 Endpoint")
+    else:
+        logger.warn("未配置 JKit 简书数据源 Endpoint，可能出现限流错误")
+
     logger.info("启动 API 服务")
     uvicorn_run(
         app="main:app",

backend/utils/config.py

@@ -8,6 +8,7 @@
 
 
 class _Config(ConfigBase, frozen=True):
+    jianshu_endpoint: str
     jtools_postgres: PostgresBlock
     jianshu_postgres: PostgresBlock
     jpep_postgres: PostgresBlock