Compare commits
1 Commits
14ce7e6e3f
...
a1b3f51b61
| Author | SHA1 | Date | |
|---|---|---|---|
| a1b3f51b61 |
@@ -124,17 +124,7 @@ class Database:
|
||||
|
||||
def init_tables(self):
|
||||
"""初始化数据库表"""
|
||||
cursor = self.conn.cursor()
|
||||
|
||||
# 用户表
|
||||
cursor.execute("""
|
||||
CREATE TABLE IF NOT EXISTS users (
|
||||
user_id INTEGER PRIMARY KEY,
|
||||
username TEXT,
|
||||
created_at INTEGER NOT NULL,
|
||||
last_active INTEGER NOT NULL
|
||||
)
|
||||
""")
|
||||
pass
|
||||
|
||||
def close(self):
|
||||
"""关闭数据库连接"""
|
||||
|
||||
@@ -16,7 +16,6 @@ config = ProjectConfig()
|
||||
|
||||
class DatabaseModel(BaseModel):
|
||||
table_name: str = Field(default="main_table")
|
||||
column_names: List[str] = Field(default=[])
|
||||
column_defs: Dict[str, str] = Field(default={})
|
||||
|
||||
class PluginInterface(ABC):
|
||||
@@ -47,8 +46,8 @@ class PluginInterface(ABC):
|
||||
db_model = self.register_db_model()
|
||||
if db_model:
|
||||
db.define_table(db_model.table_name)
|
||||
for field in db_model.column_names:
|
||||
db.define_column(db_model.table_name, field, db_model.column_defs[field])
|
||||
for name, field_def in db_model.column_defs.items():
|
||||
db.define_column(db_model.table_name, name, field_def)
|
||||
|
||||
return router
|
||||
|
||||
|
||||
Reference in New Issue
Block a user