mirror of
https://github.com/grillazz/fastapi-sqlalchemy-asyncpg.git
synced 2025-08-26 16:40:40 +03:00
99 lines
3.7 KiB
Python
99 lines
3.7 KiB
Python
import os
|
|
|
|
from pydantic import PostgresDsn, RedisDsn, computed_field
|
|
from pydantic_core import MultiHostUrl
|
|
from pydantic_settings import BaseSettings, SettingsConfigDict
|
|
|
|
|
|
class Settings(BaseSettings):
|
|
model_config = SettingsConfigDict(
|
|
env_file=".env", env_ignore_empty=True, extra="ignore"
|
|
)
|
|
jwt_algorithm: str = os.getenv("JWT_ALGORITHM")
|
|
jwt_expire: int = os.getenv("JWT_EXPIRE")
|
|
|
|
REDIS_HOST: str
|
|
REDIS_PORT: int
|
|
REDIS_DB: str
|
|
|
|
JWT_ALGORITHM: str
|
|
JWT_EXPIRE: int
|
|
|
|
POSTGRES_USER: str
|
|
POSTGRES_PASSWORD: str
|
|
POSTGRES_HOST: str
|
|
POSTGRES_DB: str
|
|
|
|
@computed_field
|
|
@property
|
|
def redis_url(self) -> RedisDsn:
|
|
"""
|
|
This is a computed field that generates a RedisDsn URL for redis-py.
|
|
|
|
The URL is built using the MultiHostUrl.build method, which takes the following parameters:
|
|
- scheme: The scheme of the URL. In this case, it is "redis".
|
|
- host: The host of the Redis database, retrieved from the REDIS_HOST environment variable.
|
|
- port: The port of the Redis database, retrieved from the REDIS_PORT environment variable.
|
|
- path: The path of the Redis database, retrieved from the REDIS_DB environment variable.
|
|
|
|
Returns:
|
|
RedisDsn: The constructed RedisDsn URL for redis-py.
|
|
"""
|
|
return MultiHostUrl.build(
|
|
scheme="redis",
|
|
host=self.REDIS_HOST,
|
|
port=self.REDIS_PORT,
|
|
path=self.REDIS_DB,
|
|
)
|
|
|
|
@computed_field
|
|
@property
|
|
def asyncpg_url(self) -> PostgresDsn:
|
|
"""
|
|
This is a computed field that generates a PostgresDsn URL for asyncpg.
|
|
|
|
The URL is built using the MultiHostUrl.build method, which takes the following parameters:
|
|
- scheme: The scheme of the URL. In this case, it is "postgresql+asyncpg".
|
|
- username: The username for the Postgres database, retrieved from the POSTGRES_USER environment variable.
|
|
- password: The password for the Postgres database, retrieved from the POSTGRES_PASSWORD environment variable.
|
|
- host: The host of the Postgres database, retrieved from the POSTGRES_HOST environment variable.
|
|
- path: The path of the Postgres database, retrieved from the POSTGRES_DB environment variable.
|
|
|
|
Returns:
|
|
PostgresDsn: The constructed PostgresDsn URL for asyncpg.
|
|
"""
|
|
return MultiHostUrl.build(
|
|
scheme="postgresql+asyncpg",
|
|
username=self.POSTGRES_USER,
|
|
password=self.POSTGRES_PASSWORD,
|
|
host=self.POSTGRES_HOST,
|
|
path=self.POSTGRES_DB,
|
|
)
|
|
|
|
@computed_field
|
|
@property
|
|
def postgres_url(self) -> PostgresDsn:
|
|
"""
|
|
This is a computed field that generates a PostgresDsn URL
|
|
|
|
The URL is built using the MultiHostUrl.build method, which takes the following parameters:
|
|
- scheme: The scheme of the URL. In this case, it is "postgres".
|
|
- username: The username for the Postgres database, retrieved from the POSTGRES_USER environment variable.
|
|
- password: The password for the Postgres database, retrieved from the POSTGRES_PASSWORD environment variable.
|
|
- host: The host of the Postgres database, retrieved from the POSTGRES_HOST environment variable.
|
|
- path: The path of the Postgres database, retrieved from the POSTGRES_DB environment variable.
|
|
|
|
Returns:
|
|
PostgresDsn: The constructed PostgresDsn URL.
|
|
"""
|
|
return MultiHostUrl.build(
|
|
scheme="postgres",
|
|
username=self.POSTGRES_USER,
|
|
password=self.POSTGRES_PASSWORD,
|
|
host=self.POSTGRES_HOST,
|
|
path=self.POSTGRES_DB,
|
|
)
|
|
|
|
|
|
settings = Settings()
|