update：后端-修改yaml配置后的参数

rpyc_scheduler/config.py

@@ -1,13 +1,20 @@
-from pydantic_settings import BaseSettings
-from pydantic import MySQLDsn, RedisDsn
+import yaml
+import os
+from pathlib import Path
 
+# 获取环境变量，默认为 development
+env = os.getenv('ENV', 'development')
+config_path = Path(__file__).parent.parent / 'config' / f'{env}.yaml'
 
-class RpcConfig(BaseSettings):
-    # apscheduler指定job store和excutors
-    apscheduler_job_store: MySQLDsn = 'mysql+pymysql://root:[email protected]/devops'
-    redis: dict = {'host': '192.168.137.129', 'password': 'seraphim', 'port': 6379, 'health_check_interval': 30}
-    rpc_port: int = 18861
+# 初始化配置变量
+rpc_config = {}
 
+# 读取配置文件
+def load_config():
+    global rpc_config
+    with open(config_path, 'r', encoding='utf-8') as f:
+        config = yaml.safe_load(f)
+        rpc_config.update(config['scheduler'])
 
-
-rpc_config = RpcConfig()
+# 加载配置
+load_config()

rpyc_scheduler/models.py

@@ -2,11 +2,11 @@
 from datetime import datetime
 from sqlmodel import SQLModel, Field, Column, Integer, create_engine, Session
 from sqlalchemy.dialects import mysql
-from config import rpc_config
+from .config import rpc_config
 from sqlmodel import SQLModel
 from typing import Union
 
-engine = create_engine(str(rpc_config.apscheduler_job_store), pool_size=5, max_overflow=10, pool_timeout=30,
+engine = create_engine(str(rpc_config['apscheduler_job_store']), pool_size=5, max_overflow=10, pool_timeout=30,
                        pool_pre_ping=True)
 
 

rpyc_scheduler/tasks.py

@@ -9,13 +9,13 @@
 from datetime import datetime
 from loguru import logger
 from sqlmodel import text
-from utils import Channel, hosts_to_inventory
-from config import rpc_config
-from models import engine, InventoryHost
+from .utils import Channel, hosts_to_inventory
+from .config import rpc_config
+from .models import engine, InventoryHost
 
 
 def local_executor(job_id, host, command):
-    with Channel(rpc_config.redis, job_id=f"{job_id}:{host}") as channel:
+    with Channel(rpc_config['redis'], job_id=f"{job_id}:{host}") as channel:
         start_time = datetime.now()
         channel.send({'msg': '开始执行任务：'})
         channel.send({'msg': f"执行命令：{command}"})
@@ -91,7 +91,7 @@ def ansible_task(job_id: str, targets: List[int], ansible_args: Dict[str, Any],
             (project_dir / playbook_name).write_text(playbook_content)
             ansible_args['playbook'] = playbook_name
     # 执行任务，日志通过event_handler写入redis，达到实时写入的效果
-    with Channel(rpc_config.redis, job_id=job_id) as channel:
+    with Channel(rpc_config['redis'], job_id=job_id) as channel:
         channel.send({'msg': '开始执行任务'})
         ansible_msg = "执行主机：" + ','.join(ansible_inventory['all']['hosts'].keys()) + '\r\n'
         if ansible_args['module']: