lan
35 lines
1.1 KiB
Python
35 lines
1.1 KiB
Python
import datetime
|
|
from sqlalchemy import Boolean, Column, Integer, String, DateTime
|
|
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 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(String(500))
|
|
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)
|