80 lines
No EOL
2.1 KiB
Python
80 lines
No EOL
2.1 KiB
Python
import sqlite3
|
|
|
|
from utils import asset
|
|
|
|
class VNDB:
|
|
|
|
def __init__(self, db_name=asset('vn_database.db')):
|
|
self.__db_name = db_name
|
|
connection = sqlite3.connect(self.__db_name)
|
|
cursor = connection.cursor()
|
|
|
|
cursor.execute('''
|
|
CREATE TABLE IF NOT EXISTS marks (
|
|
id INTEGER NOT NULL PRIMARY KEY AUTOINCREMENT,
|
|
tag TEXT NOT NULL,
|
|
value TEXT NOT NULL
|
|
);
|
|
''')
|
|
|
|
cursor.execute('''
|
|
CREATE TABLE IF NOT EXISTS authors (
|
|
id INTEGER NOT NULL PRIMARY KEY AUTOINCREMENT,
|
|
title TEXT NOT NULL,
|
|
description TEXT NOT NULL,
|
|
thumbnail TEXT NOT NULL
|
|
);
|
|
''')
|
|
|
|
cursor.execute('''
|
|
CREATE TABLE IF NOT EXISTS novels (
|
|
id INTEGER NOT NULL PRIMARY KEY AUTOINCREMENT,
|
|
title TEXT NOT NULL,
|
|
author INTEGER NOT NULL
|
|
);
|
|
''')
|
|
|
|
cursor.execute('''
|
|
CREATE TABLE IF NOT EXISTS posts_log (
|
|
|
|
novel INTEGER NOT NULL,
|
|
|
|
channel INTEGER NOT NULL,
|
|
|
|
marks JSON NOT NULL,
|
|
|
|
title TEXT NOT NULL,
|
|
|
|
description TEXT NOT NULL,
|
|
|
|
author INTEGER NOT NULL,
|
|
|
|
files_on_disk JSON NOT NULL,
|
|
post_info JSON,
|
|
post_at INTEGER NOT NULL,
|
|
created_at INTEGER NOT NULL
|
|
);
|
|
''')
|
|
|
|
connection.commit()
|
|
connection.close()
|
|
|
|
def __execute(self, sql: str, params: tuple):
|
|
connection = sqlite3.connect(self.__db_name)
|
|
connection.cursor().execute(sql, params)
|
|
connection.commit()
|
|
connection.close()
|
|
|
|
def insert_mark(self, type: str, value: str):
|
|
self.__execute("INSERT INTO marks (tag, value) "
|
|
"VALUES (?, ?)", (type, value))
|
|
|
|
def insert_novel(self, title: str, author_id: int):
|
|
self.__execute("INSERT INTO novels (title, author) "
|
|
"VALUES (?, ?)", (title, author_id))
|
|
|
|
def insert_author(self, title: str, desc: str, thumb: str):
|
|
self.__execute("INSERT INTO authors (title, description, thumbnail) "
|
|
"VALUES (?, ?, ?)", (title, desc, thumb))
|
|
|
|
|