akari-bot/modules/wiki/orm.py

from sqlalchemy import Column, String, Text, TIMESTAMP, text
from sqlalchemy.dialects.mysql import LONGTEXT
from sqlalchemy.ext.declarative import declarative_base

from database.orm import DBSession

Base = declarative_base()
table_prefix = 'module_wiki_'
db = DBSession()
session = db.session
engine = db.engine


class WikiTargetSetInfo(Base):
    __tablename__ = table_prefix + 'TargetSetInfo'
    targetId = Column(String(512), primary_key=True)
    link = Column(LONGTEXT if session.bind.dialect.name == 'mysql' else Text)
    iws = Column(LONGTEXT if session.bind.dialect.name == 'mysql' else Text)
    headers = Column(LONGTEXT if session.bind.dialect.name == 'mysql' else Text)
    prefix = Column(LONGTEXT if session.bind.dialect.name == 'mysql' else Text)


class WikiInfo(Base):
    __tablename__ = table_prefix + 'WikiInfo'
    apiLink = Column(String(512), primary_key=True)
    siteInfo = Column(LONGTEXT if session.bind.dialect.name == 'mysql' else Text)
    timestamp = Column(TIMESTAMP, default=text('CURRENT_TIMESTAMP'))


class WikiAllowList(Base):
    __tablename__ = table_prefix + 'WikiAllowList'
    apiLink = Column(String(512), primary_key=True)
    operator = Column(String(512))


class WikiBlockList(Base):
    __tablename__ = table_prefix + 'WikiBlockList'
    apiLink = Column(String(512), primary_key=True)
    operator = Column(String(512))


Base.metadata.create_all(bind=engine, checkfirst=True)