# %%
# ZED Python REPL

from sqlalchemy import Column, Date, Integer, String, create_engine
from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy.orm import sessionmaker

# Engine
engine = create_engine("sqlite:///gotodon.db")

# Base
Base = declarative_base()


# Models
class Notifikasi(Base):
    __tablename__ = "notifikasi"

    id = Column(Integer, primary_key=True)
    inreplyto = Column(String)
    post_id = Column(String, unique=True)
    created_at = Column(Date)
    handler = Column(String)
    display_name = Column(String)
    type = Column(String)
    status = Column(String)
    url = Column(String)
    remark = Column(String)

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


# Buat tabel
Base.metadata.create_all(engine)

# Session
Session = sessionmaker(bind=engine)