97 lines
3.3 KiB
Python
97 lines
3.3 KiB
Python
from sqlalchemy import Column, Integer, String, Text, TIMESTAMP, Boolean, text
|
|
from sqlalchemy.dialects.mysql import LONGTEXT
|
|
|
|
from database.orm import Session, DB_LINK
|
|
from database.orm_base import Base
|
|
from core.utils.i18n import default_locale
|
|
|
|
is_mysql = DB_LINK.startswith('mysql')
|
|
|
|
|
|
class SenderInfo(Base):
|
|
"""发送者信息"""
|
|
__tablename__ = "SenderInfo"
|
|
id = Column(String(512), primary_key=True)
|
|
isInBlockList = Column(Boolean, default=False)
|
|
isInAllowList = Column(Boolean, default=False)
|
|
isSuperUser = Column(Boolean, default=False)
|
|
warns = Column(Integer, default='0')
|
|
disable_typing = Column(Boolean, default=False)
|
|
|
|
|
|
class TargetInfo(Base):
|
|
__tablename__ = "TargetInfo"
|
|
targetId = Column(String(512), primary_key=True)
|
|
enabledModules = Column(LONGTEXT if is_mysql else Text, default='[]')
|
|
options = Column(LONGTEXT if is_mysql else Text, default='{}')
|
|
custom_admins = Column(LONGTEXT if is_mysql else Text, default='[]')
|
|
muted = Column(Boolean, default=False)
|
|
locale = Column(String(512), default=default_locale)
|
|
petal = Column(Integer, default=0)
|
|
|
|
|
|
class StoredData(Base):
|
|
"""数据存储"""
|
|
__tablename__ = "StoredData"
|
|
name = Column(String(512), primary_key=True)
|
|
value = Column(LONGTEXT if is_mysql else Text)
|
|
|
|
|
|
class CommandTriggerTime(Base):
|
|
"""命令触发时间"""
|
|
__tablename__ = "CommandTriggerTime"
|
|
targetId = Column(String(512), primary_key=True)
|
|
commandName = Column(String(512))
|
|
timestamp = Column(TIMESTAMP, default=text('CURRENT_TIMESTAMP'))
|
|
|
|
|
|
class GroupAllowList(Base):
|
|
__tablename__ = "GroupAllowList"
|
|
targetId = Column(String(512), primary_key=True)
|
|
|
|
|
|
class AnalyticsData(Base):
|
|
"""统计信息"""
|
|
__tablename__ = "Analytics"
|
|
id = Column(Integer, primary_key=True)
|
|
moduleName = Column(String(512))
|
|
moduleType = Column(String(512))
|
|
targetId = Column(String(512))
|
|
senderId = Column(String(512))
|
|
command = Column(LONGTEXT if is_mysql else Text)
|
|
timestamp = Column(TIMESTAMP, default=text('CURRENT_TIMESTAMP'))
|
|
|
|
__table_args__ = {'mysql_charset': 'utf8mb4'}
|
|
|
|
|
|
class DBVersion(Base):
|
|
__tablename__ = "DBVersion"
|
|
value = Column(String(512), primary_key=True)
|
|
|
|
|
|
class UnfriendlyActionsTable(Base):
|
|
__tablename__ = "unfriendly_action"
|
|
id = Column(Integer, primary_key=True)
|
|
targetId = Column(String(512))
|
|
senderId = Column(String(512))
|
|
action = Column(String(512))
|
|
detail = Column(String(512))
|
|
timestamp = Column(TIMESTAMP, default=text('CURRENT_TIMESTAMP'))
|
|
|
|
|
|
class JobQueueTable(Base):
|
|
__tablename__ = "job_queues"
|
|
taskid = Column(String(512), primary_key=True)
|
|
targetClient = Column(String(512))
|
|
hasDone = Column(Boolean, default=False)
|
|
action = Column(String(512))
|
|
args = Column(LONGTEXT if is_mysql else Text, default='{}')
|
|
returnVal = Column(LONGTEXT if is_mysql else Text, default='{}')
|
|
timestamp = Column(TIMESTAMP, default=text('CURRENT_TIMESTAMP'))
|
|
|
|
__table_args__ = {'mysql_charset': 'utf8mb4'}
|
|
|
|
|
|
Session.create()
|
|
__all__ = ["SenderInfo", "TargetInfo", "CommandTriggerTime", "GroupAllowList",
|
|
"StoredData", "DBVersion", "AnalyticsData", "UnfriendlyActionsTable", "JobQueueTable"]
|