import datetime
from sqlalchemy import Boolean, Column, Integer, String, DateTime, JSON, Text
from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy.ext.asyncio import create_async_engine
from sqlalchemy.ext.asyncio.session import AsyncSession
import settings

engine = create_async_engine(settings.DATABASE_URL)

Base = declarative_base()


async def init_models():
    async with engine.begin() as conn:
        await conn.run_sync(Base.metadata.create_all)


async def get_session():
    async with AsyncSession(engine, expire_on_commit=False) as s:
        yield s


class Values(Base):
    __tablename__ = 'values'
    id = Column(Integer, primary_key=True, index=True)
    key = Column(String, unique=True)
    value = Column(JSON)


class Codes(Base):
    __tablename__ = "codes"
    id = Column(Integer, primary_key=True, index=True)
    code = Column(String(10), unique=True, index=True)
    key = Column(String(30), unique=True, index=True)
    name = Column(String(500))
    size = Column(Integer)
    type = Column(String(20))
    text = Column(Text)
    used = Column(Boolean, default=False)
    count = Column(Integer, default=-1)
    use_time = Column(DateTime, default=datetime.datetime.now, onupdate=datetime.datetime.now)
    exp_time = Column(DateTime, nullable=True)