kiwi-vpn/api/kiwi_vpn_api/db/connection.py

"""
Database connection management.
"""

from sqlmodel import Session, SQLModel, create_engine


class Connection:
    """
    Namespace for the database connection
    """

    engine = None

    @classmethod
    def connect(cls, connection_url: str) -> None:
        """
        Connect ORM to a database engine.
        """

        cls.engine = create_engine(connection_url)
        SQLModel.metadata.create_all(cls.engine)

    @classmethod
    @property
    def session(cls) -> Session:
        """
        Create an ORM session using a context manager.
        """

        if cls.engine is None:
            raise ValueError("Not connected to database, can't create session")

        return Session(cls.engine)