from datetime import date, datetime
from decimal import Decimal

from pony.orm import (  # type: ignore[import-untyped]
    Database,
    LongStr,
    Optional,
    PrimaryKey,
    Required,
    Set,
    set_sql_debug,
)

db = Database

db = Database()


class Cases(db.Entity):  # type: ignore[name-defined]
    case_id = PrimaryKey(int, auto=True)
    docket_id = Required(str)
    petitioners = Required(str)
    respondents = Optional(str)
    date = Required(date)
    active = Required(bool)
    entries = Set("Entries")
    lhs_id = Set("Cases", column="lhs")
    rhs_id = Set("Cases", column="rhs")


class Entries(db.Entity):  # type: ignore[name-defined]
    entry_id = PrimaryKey(int, auto=True)
    case_id = Required(Cases)
    date = Required(date)
    text = Required(LongStr)
    documents = Set("Documents")


class Documents(db.Entity):  # type: ignore[name-defined]
    document_id = PrimaryKey(int, auto=True)
    entry_id = Required(Entries)
    name = Required(str)
    url = Required(LongStr)


class History(db.Entity):  # type: ignore[name-defined]
    history_id = PrimaryKey(int, auto=True)
    year = Required(str, max_len=3)
    edocket = Required(bool)
    number = Required(int)


set_sql_debug(True)
db.bind(
    provider="mysql",
    user="scotus",
    host="ceph5",
    database="scotus",
    password="lechOtvirf8Om/",
)
db.generate_mapping(create_tables=True)
db.disconnect()