| New file |
| | |
|---|
| | | """ |
|---|
| | | 启动脚本 - 处理配置文件和目录初始化 |
|---|
| | | """ |
|---|
| | | |
|---|
| | | import os |
|---|
| | | import sys |
|---|
| | | import json |
|---|
| | | import shutil |
|---|
| | | from pathlib import Path |
|---|
| | | |
|---|
| | | |
|---|
| | | def get_exe_dir(): |
|---|
| | | """获取exe文件所在目录""" |
|---|
| | | if getattr(sys, "frozen", False): |
|---|
| | | # 如果是打包后的exe文件 |
|---|
| | | return os.path.dirname(sys.executable) |
|---|
| | | else: |
|---|
| | | # 如果是开发环境 |
|---|
| | | return os.path.dirname(os.path.abspath(__file__)) |
|---|
| | | |
|---|
| | | |
|---|
| | | def ensure_directories(): |
|---|
| | | """确保必要的目录存在""" |
|---|
| | | exe_dir = get_exe_dir() |
|---|
| | | |
|---|
| | | # 创建日志目录 |
|---|
| | | logs_dir = os.path.join(exe_dir, "logs") |
|---|
| | | if not os.path.exists(logs_dir): |
|---|
| | | os.makedirs(logs_dir) |
|---|
| | | print(f"创建日志目录: {logs_dir}") |
|---|
| | | |
|---|
| | | return exe_dir |
|---|
| | | |
|---|
| | | |
|---|
| | | def ensure_config_file(): |
|---|
| | | """确保配置文件存在""" |
|---|
| | | exe_dir = get_exe_dir() |
|---|
| | | config_file = os.path.join(exe_dir, "config.json") |
|---|
| | | |
|---|
| | | if not os.path.exists(config_file): |
|---|
| | | # 如果config.json不存在，尝试复制生产配置模板 |
|---|
| | | production_config = os.path.join(exe_dir, "config.production.json") |
|---|
| | | example_config = os.path.join(exe_dir, "config.example.json") |
|---|
| | | |
|---|
| | | if os.path.exists(production_config): |
|---|
| | | shutil.copy2(production_config, config_file) |
|---|
| | | print(f"复制生产配置文件: {production_config} -> {config_file}") |
|---|
| | | elif os.path.exists(example_config): |
|---|
| | | shutil.copy2(example_config, config_file) |
|---|
| | | print(f"复制示例配置文件: {example_config} -> {config_file}") |
|---|
| | | else: |
|---|
| | | # 创建默认配置文件 |
|---|
| | | default_config = { |
|---|
| | | "database": { |
|---|
| | | "url": "mysql+pymysql://root:password@localhost:3306/ecloud_dify" |
|---|
| | | }, |
|---|
| | | "redis": {"url": "redis://localhost:6379/0"}, |
|---|
| | | "ecloud": { |
|---|
| | | "base_url": "http://125.122.152.142:9899", |
|---|
| | | "authorization": "your_ecloud_authorization_token", |
|---|
| | | "w_id": "your_ecloud_w_id", |
|---|
| | | }, |
|---|
| | | "dify": { |
|---|
| | | "base_url": "https://api.dify.ai/v1", |
|---|
| | | "api_key": "your_dify_api_key", |
|---|
| | | }, |
|---|
| | | "server": {"host": "0.0.0.0", "port": 7979, "debug": False}, |
|---|
| | | "logging": {"level": "INFO", "file": "logs/app.log"}, |
|---|
| | | "message_processing": { |
|---|
| | | "max_retry_count": 3, |
|---|
| | | "retry_delay": 5, |
|---|
| | | "queue_timeout": 300, |
|---|
| | | }, |
|---|
| | | "customer_service": {"names": ["客服1", "客服2"]}, |
|---|
| | | } |
|---|
| | | |
|---|
| | | with open(config_file, "w", encoding="utf-8") as f: |
|---|
| | | json.dump(default_config, f, indent=2, ensure_ascii=False) |
|---|
| | | print(f"创建默认配置文件: {config_file}") |
|---|
| | | |
|---|
| | | return config_file |
|---|
| | | |
|---|
| | | |
|---|
| | | def setup_environment(): |
|---|
| | | """设置运行环境""" |
|---|
| | | exe_dir = ensure_directories() |
|---|
| | | config_file = ensure_config_file() |
|---|
| | | |
|---|
| | | # 设置工作目录为exe所在目录 |
|---|
| | | os.chdir(exe_dir) |
|---|
| | | print(f"设置工作目录: {exe_dir}") |
|---|
| | | |
|---|
| | | return exe_dir, config_file |
|---|
| | | |
|---|
| | | |
|---|
| | | if __name__ == "__main__": |
|---|
| | | setup_environment() |
|---|
| | | |
|---|
| | | # 导入并启动主应用 |
|---|
| | | try: |
|---|
| | | from main import app |
|---|
| | | import uvicorn |
|---|
| | | from config import settings |
|---|
| | | from loguru import logger |
|---|
| | | |
|---|
| | | # 配置日志 |
|---|
| | | logger.add( |
|---|
| | | settings.log_file, |
|---|
| | | rotation="1 day", |
|---|
| | | retention="7 days", |
|---|
| | | level=settings.log_level, |
|---|
| | | format="{time:YYYY-MM-DD HH:mm:ss} | {level} | {name}:{function}:{line} - {message}", |
|---|
| | | ) |
|---|
| | | |
|---|
| | | logger.info("启动E云管家-DifyAI对接服务") |
|---|
| | | |
|---|
| | | # 启动服务 |
|---|
| | | uvicorn.run( |
|---|
| | | app, |
|---|
| | | host=settings.server_host, |
|---|
| | | port=settings.server_port, |
|---|
| | | log_level=settings.log_level.lower(), |
|---|
| | | ) |
|---|
| | | |
|---|
| | | except Exception as e: |
|---|
| | | print(f"启动失败: {e}") |
|---|
| | | input("按任意键退出...") |
|---|
| | | sys.exit(1) |
|---|
