Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
10 changes: 5 additions & 5 deletions verdict-backend/alembic/env.py
Original file line number Diff line number Diff line change
Expand Up @@ -7,7 +7,7 @@

import app.models # noqa: F401 # register models for autogenerate
from alembic import context
from app.config import get_settings
from app.database_config import get_database_settings

# this is the Alembic Config object, which provides
# access to the values within the .ini file in use.
Expand Down Expand Up @@ -38,8 +38,8 @@ def run_migrations_offline() -> None:
script output.

"""
settings = get_settings()
url = settings.database_url
db_settings = get_database_settings()
url = db_settings.database_url
context.configure(
url=url,
target_metadata=target_metadata,
Expand All @@ -58,9 +58,9 @@ def run_migrations_online() -> None:
and associate a connection with the context.

"""
settings = get_settings()
db_settings = get_database_settings()
connectable = engine_from_config(
{"sqlalchemy.url": settings.database_url},
{"sqlalchemy.url": db_settings.database_url},
prefix="sqlalchemy.",
poolclass=pool.NullPool,
)
Expand Down
27 changes: 3 additions & 24 deletions verdict-backend/app/config.py
Original file line number Diff line number Diff line change
@@ -1,35 +1,14 @@
# app/config.py
from functools import lru_cache

from pydantic import Field, SecretStr
from pydantic_settings import BaseSettings, SettingsConfigDict
from pydantic import Field

from app.database_config import DatabaseSettings

class Settings(BaseSettings):
model_config = SettingsConfigDict(
env_file=".env",
env_file_encoding="utf-8",
case_sensitive=False,
extra="allow",
)

database_host: str = Field(default=...)
database_port: int = Field(default=...)
database_name: str = Field(default=...)
database_user: str = Field(default=...)
database_password: SecretStr = Field(default=...)

class Settings(DatabaseSettings):
asset_inventory_url: str = Field(default=...)
cmdb_url: str = Field(default=...)

@property
def database_url(self) -> str:
return (
f"postgresql+psycopg2://"
f"{self.database_user}:{self.database_password.get_secret_value()}"
f"@{self.database_host}:{self.database_port}/{self.database_name}"
)

debug: bool = False
log_level: str = "info"

Expand Down
32 changes: 32 additions & 0 deletions verdict-backend/app/database_config.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,32 @@
from functools import lru_cache

from pydantic import Field, SecretStr
from pydantic_settings import BaseSettings, SettingsConfigDict


class DatabaseSettings(BaseSettings):
model_config = SettingsConfigDict(
env_file=".env",
env_file_encoding="utf-8",
case_sensitive=False,
extra="allow",
)

database_host: str = Field(default=...)
database_port: int = Field(default=...)
database_name: str = Field(default=...)
database_user: str = Field(default=...)
database_password: SecretStr = Field(default=...)

@property
def database_url(self) -> str:
return (
f"postgresql+psycopg2://"
f"{self.database_user}:{self.database_password.get_secret_value()}"
f"@{self.database_host}:{self.database_port}/{self.database_name}"
)


@lru_cache
def get_database_settings() -> DatabaseSettings:
return DatabaseSettings()
Loading