tests.seed_db

Complexity

Total Complexity

2

Size/Duplication

Total Lines	23
Duplicated Lines	0 %

Importance

Changes

0

from curses import echo
from sqlalchemy import (create_engine, Column, String, Integer, Index, engine)
from sqlalchemy.ext.declarative import declarative_base

Base = declarative_base()


class User(Base):
    __tablename__ = "user"
    username = Column("username", String(20), primary_key=True)
    age = Column("age", Integer)
    __table_args__ = (
        Index("idx_user_username", "username"),
    )


def create_table():
    e: engine.Engine = create_engine(
        "sqlite:///test.db", echo=True, future=True)
    with e.begin() as c:
        Base.metadata.create_all(bind=c)
    return e