1
0
mirror of https://gitlab.com/MoonTestUse1/AdministrationItDepartmens.git synced 2025-08-14 00:25:46 +02:00

Fix database

This commit is contained in:
MoonTestUse1
2025-01-07 06:38:45 +06:00
parent 92d6b9c753
commit ee0217cfbb
2 changed files with 26 additions and 49 deletions

View File

@@ -1,37 +1,33 @@
"""Application configuration"""
"""Configuration module"""
import os
from typing import Any
from pydantic_settings import BaseSettings
from functools import lru_cache
class Settings(BaseSettings):
"""Application settings"""
# База данных
POSTGRES_USER: str = "postgres"
POSTGRES_PASSWORD: str = "postgres"
POSTGRES_HOST: str = os.getenv("POSTGRES_HOST", "postgres")
POSTGRES_PORT: str = "5432"
POSTGRES_DB: str = "app"
POSTGRES_TEST_DB: str = "test_app"
PROJECT_NAME: str = "Employee Request System"
API_V1_STR: str = "/api"
# Database
POSTGRES_USER: str = os.getenv("POSTGRES_USER", "postgres")
POSTGRES_PASSWORD: str = os.getenv("POSTGRES_PASSWORD", "postgres")
POSTGRES_HOST: str = os.getenv("POSTGRES_HOST", "localhost")
POSTGRES_PORT: str = os.getenv("POSTGRES_PORT", "5432")
POSTGRES_DB: str = os.getenv("POSTGRES_DB", "app")
# JWT
SECRET_KEY: str = "your-secret-key"
SECRET_KEY: str = os.getenv("SECRET_KEY", "your-secret-key-super-secret")
ALGORITHM: str = "HS256"
ACCESS_TOKEN_EXPIRE_MINUTES: int = 30
# Режим тестирования
TESTING: bool = bool(os.getenv("TESTING", ""))
# Redis
REDIS_HOST: str = "redis"
REDIS_PORT: int = 6379
REDIS_DB: int = 0
REDIS_TEST_DB: int = 1
REDIS_HOST: str = os.getenv("REDIS_HOST", "localhost")
REDIS_PORT: int = int(os.getenv("REDIS_PORT", "6379"))
REDIS_DB: int = int(os.getenv("REDIS_DB", "0"))
# Telegram
TELEGRAM_BOT_TOKEN: str = ""
TELEGRAM_CHAT_ID: str = ""
TELEGRAM_BOT_TOKEN: str = os.getenv("TELEGRAM_BOT_TOKEN", "")
TELEGRAM_CHAT_ID: str = os.getenv("TELEGRAM_CHAT_ID", "")
model_config = {
"case_sensitive": True,
"env_file": ".env",
@@ -40,29 +36,9 @@ class Settings(BaseSettings):
def get_database_url(self) -> str:
"""Get database URL"""
# Получаем URL из переменной окружения, если она есть
if os.getenv("DATABASE_URL"):
return os.getenv("DATABASE_URL")
# Иначе формируем URL на основе настроек
if self.TESTING:
host = os.getenv("POSTGRES_HOST", "postgres")
db = self.POSTGRES_TEST_DB
else:
host = self.POSTGRES_HOST
db = self.POSTGRES_DB
return f"postgresql://{self.POSTGRES_USER}:{self.POSTGRES_PASSWORD}@{host}:{self.POSTGRES_PORT}/{db}"
return os.getenv(
"DATABASE_URL",
f"postgresql://{self.POSTGRES_USER}:{self.POSTGRES_PASSWORD}@{self.POSTGRES_HOST}:{self.POSTGRES_PORT}/{self.POSTGRES_DB}"
)
def get_redis_url(self) -> str:
"""Get Redis URL"""
db = self.REDIS_TEST_DB if self.TESTING else self.REDIS_DB
host = "localhost" if self.TESTING else self.REDIS_HOST
return f"redis://{host}:{self.REDIS_PORT}/{db}"
@lru_cache()
def get_settings() -> Settings:
"""Get cached settings"""
return Settings()
settings = get_settings()
settings = Settings()

View File

@@ -5,6 +5,7 @@ from pydantic_settings import BaseSettings
class TestSettings(BaseSettings):
"""Test settings"""
PROJECT_NAME: str = "Employee Request System Test"
API_V1_STR: str = "/api"
# Database
POSTGRES_USER: str = "postgres"