Spaces:
Running
Running
File size: 2,467 Bytes
d825c91 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 |
from pydantic_settings import BaseSettings
from typing import Optional, Dict, Any
import os
from dotenv import load_dotenv
load_dotenv()
class Settings(BaseSettings):
# Application
APP_NAME: str = "TestAuto"
DEBUG: bool = False
VERSION: str = "1.0.0"
# API
API_V1_STR: str = "/api/v1"
PROJECT_NAME: str = "TestAuto API"
# Security
SECRET_KEY: str = os.getenv("SECRET_KEY", "your-secret-key")
ACCESS_TOKEN_EXPIRE_MINUTES: int = 60 * 24 * 8 # 8 days
# Database
POSTGRES_SERVER: str = os.getenv("POSTGRES_SERVER", "localhost")
POSTGRES_USER: str = os.getenv("POSTGRES_USER", "postgres")
POSTGRES_PASSWORD: str = os.getenv("POSTGRES_PASSWORD", "postgres")
POSTGRES_DB: str = os.getenv("POSTGRES_DB", "testauto")
SQLALCHEMY_DATABASE_URI: Optional[str] = None
# AI Providers
OPENAI_API_KEY: Optional[str] = os.getenv("OPENAI_API_KEY")
OPENROUTER_API_KEY: Optional[str] = os.getenv("OPENROUTER_API_KEY")
LOCAL_LLM_PATH: Optional[str] = os.getenv("LOCAL_LLM_PATH")
# Message Queue
RABBITMQ_HOST: str = os.getenv("RABBITMQ_HOST", "localhost")
RABBITMQ_PORT: int = int(os.getenv("RABBITMQ_PORT", "5672"))
RABBITMQ_USER: str = os.getenv("RABBITMQ_USER", "guest")
RABBITMQ_PASSWORD: str = os.getenv("RABBITMQ_PASSWORD", "guest")
# Test Management Tools
TESTRAIL_API_KEY: Optional[str] = os.getenv("TESTRAIL_API_KEY")
TESTRAIL_URL: Optional[str] = os.getenv("TESTRAIL_URL")
JIRA_API_TOKEN: Optional[str] = os.getenv("JIRA_API_TOKEN")
JIRA_URL: Optional[str] = os.getenv("JIRA_URL")
QTEST_API_TOKEN: Optional[str] = os.getenv("QTEST_API_TOKEN")
QTEST_URL: Optional[str] = os.getenv("QTEST_URL")
# Document Processing
UPLOAD_FOLDER: str = "uploads"
MAX_CONTENT_LENGTH: int = 16 * 1024 * 1024 # 16MB
ALLOWED_EXTENSIONS: set = {"pdf", "docx", "doc", "txt", "png", "jpg", "jpeg"}
# Logging
LOG_LEVEL: str = os.getenv("LOG_LEVEL", "INFO")
LOG_FILE: str = "logs/testauto.log"
# Docker
DOCKER_IMAGE: str = "testauto:latest"
def __init__(self, **kwargs: Any):
super().__init__(**kwargs)
self.SQLALCHEMY_DATABASE_URI = (
f"postgresql://{self.POSTGRES_USER}:{self.POSTGRES_PASSWORD}"
f"@{self.POSTGRES_SERVER}/{self.POSTGRES_DB}"
)
settings = Settings() |