import string

from typing import NewType
from typing import Annotated
from collections.abc import Iterable
from pydantic import StringConstraints
from pydantic import AfterValidator
from sqlmodel import SQLModel
from sqlmodel import Field
from sqlmodel import create_engine

def _validate_charset(s: str, charset: Iterable[str]) -> str:
    illegal_characters = set(s).difference(set(charset))
    assert (
            illegal_characters == {}
    ), f'string "{s}" contains illegal characters: {illegal_characters}'
    return s

PrintableString = Annotated[str, StringConstraints(max_length=16), AfterValidator(lambda s: _validate_charset(s, string.printable))]

# Suppose I also want a type called SpecialString to ensure during type checking that
# plain old PrintableStrings don't get confused for a SpecialString
SpecialString = NewType("SpecialString", PrintableString)

class MyModel(SQLModel, table=True):
    id: int | None = Field(default=None, primary_key=True)
    special_data: SpecialString

sqlite_file_name = "database.db"
sqlite_url = f"sqlite:///{sqlite_file_name}"

if __name__ == "__main__":
    engine = create_engine(sqlite_url, echo=True)
    SQLModel.metadata.create_all(engine) # error! TypeError: issubclass() arg 1 must be a class

# Suppose I also want a type called SpecialString to ensure during type checking that
# plain old PrintableStrings don't get confused for a SpecialString
if TYPE_CHECKING:
    SpecialString = NewType("SpecialString", PrintableString)
else:
    SpecialString = PrintableString