Files
2026-07-13 12:35:09 +08:00

44 lines
1.3 KiB
Python

# -*- coding: utf-8 -*-
"""
@Time : 2023/2/9/009 17:15
@Author : 李子
@Url : https://github.com/kslz
"""
import sqlite3
from utils.log import creatlogger
class MyDB():
def __init__(self, path="db/data.db"):
self.con = sqlite3.connect(path)
self.cur = self.con.cursor()
self.sqlitelogger = creatlogger("sqlitelogger")
self.sqlitelogger.debug("数据库连接已启动")
def init_database(self):
self.create_workspace_tbl()
pass
def create_workspace_tbl(self):
self.cur.execute('''CREATE TABLE "workspace_tbl" (
"workspace_name" TEXT,
"workspace_create_time" TEXT,
"workspace_last_use_time" TEXT,
"workspace_info" TEXT
);''')
self.sqlitelogger.debug("新建工作区表")
self.con.commit()
def create_dataset_tbl(self):
self.cur.execute('''CREATE TABLE "dataset_tbl" (
"dataset_id" INTEGER NOT NULL,
"dataset_name" TEXT,
"dataset_create_time" TEXT,
"dataset_last_use_time" TEXT,
"dataset_info" TEXT,
PRIMARY KEY ("dataset_id")
);''')
self.sqlitelogger.debug("新建数据集表")
self.con.commit()