classes.py

import dataclasses
import pathlib
import typing
 
 
@dataclasses.dataclass(frozen=True)
class ConnectionInfo:
    """Mysql connection info class.
 
    :param port: database port
    :param hostname: database hostname
    :param user: database user
    :param password: database password
    :param dbname: database name
    """
 
    port: int = 3306
    hostname: str = 'localhost'
    user: str | None = None
    password: str | None = None
    dbname: str | None = None
 
    def replace(self, **kwargs) -> 'ConnectionInfo':
        """Returns new instance with attributes updated."""
        return dataclasses.replace(self, **kwargs)
 
 
@dataclasses.dataclass(frozen=True)
class ServiceSettings:
    port: int
 
    def get_conninfo(self) -> ConnectionInfo:
        return ConnectionInfo(port=self.port, user='root')
 
 
@dataclasses.dataclass(frozen=True)
class DatabaseConfig:
    dbname: str
    migrations: list[pathlib.Path]
    create: bool = True
    keep_tables: typing.Sequence[str] = ()
    truncate_non_empty: bool = False
 
 
DatabasesDict = dict[str, DatabaseConfig]