37 lines
1.2 KiB
Python
37 lines
1.2 KiB
Python
"""
|
|
models/base.py — declarative base & the Messages model.
|
|
"""
|
|
|
|
from sqlalchemy import create_engine, Integer, String, Text
|
|
from sqlalchemy.orm import DeclarativeBase, Mapped, mapped_column, sessionmaker
|
|
from typing import Optional
|
|
|
|
|
|
class Base(DeclarativeBase):
|
|
pass
|
|
|
|
|
|
class Messages(Base):
|
|
__tablename__ = "messages"
|
|
|
|
id: Mapped[int] = mapped_column(Integer, primary_key=True)
|
|
channel: Mapped[str] = mapped_column(String(32), nullable=False, default="general")
|
|
ip: Mapped[str] = mapped_column(String(64), nullable=False)
|
|
content: Mapped[str] = mapped_column(Text, nullable=False)
|
|
gif_url: Mapped[Optional[str]] = mapped_column(Text, nullable=True, default=None)
|
|
timestamp: Mapped[int] = mapped_column(Integer, nullable=False)
|
|
|
|
|
|
def make_engine(db_cfg):
|
|
"""Create a SQLAlchemy engine from DatabaseConfig."""
|
|
if db_cfg.engine == "mariadb":
|
|
url = (
|
|
f"mysql+pymysql://{db_cfg.mariadb_user}:{db_cfg.mariadb_password}"
|
|
f"@{db_cfg.mariadb_host}:{db_cfg.mariadb_port}/{db_cfg.mariadb_database}"
|
|
)
|
|
else:
|
|
url = f"sqlite:///{db_cfg.sqlite_path}"
|
|
engine = create_engine(url, echo=False)
|
|
Base.metadata.create_all(engine)
|
|
return engine
|