config.py 3.2 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596
  1. from pathlib import Path
  2. from typing import Dict, List
  3. from enum import Enum
  4. from typing import ClassVar
  5. import json
  6. from pydantic_settings import BaseSettings, SettingsConfigDict
  7. class Settings(BaseSettings):
  8. BASE_PATH: ClassVar[Path] = Path(__file__).parent.parent.parent.absolute()
  9. TEST_DATA: str = '环境变量读取测试, 目前使用本地字段'
  10. @property
  11. def TEST_DATA_END(self) -> str:
  12. return f'{self.TEST_DATA}/end'
  13. SERVER_PORT:int = 7755
  14. # --- MinIO 配置 ---
  15. MINIO_ENDPOINT: str = "192.168.77.249:9000"
  16. MINIO_ACCESS_KEY: str = "pZEwCGnpNN05KPnmC2Yh"
  17. MINIO_SECRET_KEY: str = "KfJRuWiv9pVxhIMcFqbkv8hZT9SnNTZ6LPx592D4"
  18. MINIO_SECURE: bool = False # 是否使用 https
  19. MINIO_BUCKET: str = "grading"
  20. MINIO_BASE_PREFIX: str = "score_server_data"
  21. # DATA_HOST_URL 现在直接指向 MinIO 的前缀路径
  22. # (注意: 需要在 MinIO 后台将 grading 存储桶的访问权限配置为 Public 或开启特定读策略)
  23. @property
  24. def DATA_HOST_URL(self) -> str:
  25. return f"http://{self.MINIO_ENDPOINT}/{self.MINIO_BUCKET}/{self.MINIO_BASE_PREFIX}"
  26. # CONFIG_PATH: str = BASE_PATH / 'Config.json'
  27. API_PREFIX: str = "/api" # 通用前缀
  28. SCORE_CONFIG_PATH: Path = BASE_PATH / "app/core/scoring_config.json"
  29. # 分数计算接口url
  30. SCORE_UPDATE_SERVER_URL: str = "http://127.0.0.1:7754"
  31. @property
  32. def SCORE_RECALCULATE_ENDPOINT(self) -> str:
  33. return f"{self.SCORE_UPDATE_SERVER_URL}/api/card_score/score_recalculate"
  34. # 分数计算配置接口
  35. @property
  36. def SCORE_SERVER_CONFIG_URL(self) -> str:
  37. return f"{self.SCORE_UPDATE_SERVER_URL}/api/config/scoring_config"
  38. # --- 数据库配置 ---
  39. DB_NAME: str = 'card_score_gray_database'
  40. DB_CARD_TABLE_NAME: str = 'cards'
  41. DB_IMAGE_TABLE_NAME: str = 'card_images'
  42. DB_GRAY_IMAGE_TABLE_NAME: str = 'card_gray_images' # 灰度图表名
  43. DB_USER_CARD_TABLE_NAME: str = 'user_card_bindings'
  44. RATING_REPORT_HISTORY_TABLE_NAME: str = "rating_report_history"
  45. DB_USER: str = 'root'
  46. DB_PASSWORD: str = '123456'
  47. DB_HOST: str = '127.0.0.1'
  48. @property
  49. def DATABASE_CONFIG(self) -> Dict[str, str]:
  50. return {
  51. 'user': self.DB_USER,
  52. 'password': self.DB_PASSWORD,
  53. 'host': self.DB_HOST,
  54. }
  55. # 连接到指定数据库的配置
  56. @property
  57. def DATABASE_CONFIG_WITH_DB(self) -> Dict[str, str]:
  58. return {
  59. **self.DATABASE_CONFIG,
  60. 'database': self.DB_NAME
  61. }
  62. # 优先级:系统环境变量 > .env 文件
  63. model_config = SettingsConfigDict(env_file=".env",
  64. env_file_encoding='utf-8')
  65. def get_full_url(self, path: str) -> str:
  66. """将相对路径转换为可以直接打开的 MinIO 绝对 URL"""
  67. if not path:
  68. return path
  69. if str(path).startswith("http"):
  70. return path
  71. # 移除开头的斜杠防止双斜杠 (如: /Data/xxx -> Data/xxx)
  72. clean_path = str(path).lstrip("/\\")
  73. return f"{self.DATA_HOST_URL}/{clean_path}"
  74. settings = Settings()
  75. print("测试读取env", settings.TEST_DATA)
  76. print("测试拼接env", settings.TEST_DATA_END)