from sqlalchemy.orm import Mapped, mapped_column, relationship
from sqlalchemy import ForeignKey, create_engine
from sqlalchemy.ext.declarative import declarative_base


Base = declarative_base()
engine = create_engine("postgresql://postgres:password@localhost:5432/trading")


class Users(Base):
    __tablename__ = "users"

    id: Mapped[int] = mapped_column(primary_key=True)

    name: Mapped[str]
    age: Mapped[int]
    email: Mapped[str]

    role_id: Mapped[int] = mapped_column(ForeignKey("roles.id"))
    roles: Mapped["Roles"] = relationship("role", backref="users")


class Roles(Base):
    __tablename__ = "roles"

    id: Mapped[int] = mapped_column(primary_key=True)

    name: Mapped[str]
    permission: Mapped[str]

Base.metadata.create_all(engine)