models.py

import enum

from sqlalchemy import Column, Enum, ForeignKey, Integer, String, Text
from sqlalchemy.orm import relationship

from database import Base


class Gender(enum.Enum):
    MALE = "male"
    FEMALE = "female"


class User(Base):
    __tablename__ = "users"
    id = Column(Integer, primary_key=True)
    first_name = Column(String(64), nullable=False)
    last_name = Column(String(64), nullable=True)
    username = Column(String(128), nullable=False, unique=True)
    email = Column(String(120), nullable=True, unique=True)
    gender = Column(Enum(Gender, name="gender_types"), nullable=True)
    bio = Column(String(300), nullable=True)
    password = Column(String(255), nullable=False)

    notes = relationship("Note", back_populates="user", cascade="all, delete-orphan")

    def __repr__(self) -> str:
        return f"<User {self.id} -> {self.username}>"


class Note(Base):
    __tablename__ = "notes"
    id = Column(Integer, primary_key=True)
    content = Column(Text, nullable=False)
    owner_id = Column(
        Integer, ForeignKey("users.id", ondelete="CASCADE"), nullable=False, index=True
    )
    user = relationship("User", back_populates="notes")

    def __repr__(self) -> str:
        return f"<Note {self.id} owner_id={self.owner_id}>"