163 lines
4.4 KiB
Python
163 lines
4.4 KiB
Python
import os
|
|
import sys
|
|
import sqlite3
|
|
|
|
import UtilPack as util
|
|
|
|
class MgrCalibreDB:
|
|
def __init__(self, path):
|
|
self.path = path
|
|
|
|
def __enter__(self):
|
|
pass
|
|
|
|
def __exit__(self, ex_type, ex_value, traceback):
|
|
self.conn.close()
|
|
|
|
def init(self):
|
|
# 데이터베이스 연결 (파일이 없으면 새로 생성됨)
|
|
self.conn = sqlite3.connect(self.path)
|
|
self.conn.create_function("title_sort", 1, self._title_sort )
|
|
self.cursor = self.conn.cursor()
|
|
|
|
def _title_sort(self, value):
|
|
return 0
|
|
|
|
def GetTableAll(self, table):
|
|
if None == self.cursor:
|
|
return None
|
|
|
|
if None == table or True == util.IsEmptyStr(table):
|
|
return None
|
|
|
|
strSQL = f"SELECT * FROM \"{table}\";"
|
|
self.cursor.execute(strSQL)
|
|
listRet = self.cursor.fetchall()
|
|
|
|
return listRet
|
|
|
|
def GetTableSchm(self, table):
|
|
if None == self.cursor:
|
|
return None
|
|
|
|
if None == table or True == util.IsEmptyStr(table):
|
|
return None
|
|
|
|
strSQL = f"PRAGMA table_info (\"{table}\");"
|
|
self.cursor.execute(strSQL)
|
|
listRet = self.cursor.fetchall()
|
|
|
|
return listRet
|
|
|
|
def GetBookListforUI_ArcList(self):
|
|
if None == self.cursor:
|
|
return None
|
|
|
|
try:
|
|
strSQL = f"Select title, path from books;"
|
|
self.cursor.execute(strSQL)
|
|
listRet = self.cursor.fetchall()
|
|
except sqlite3.Error as e:
|
|
listRet = []
|
|
print(f"SQLite Erro occurred: {e}")
|
|
|
|
return listRet
|
|
|
|
|
|
def UpdateTableValue(self,Table, nID, ColName, Value):
|
|
if None == self.cursor:
|
|
return
|
|
|
|
if 0 >= nID or True == util.IsEmptyStr(Table) or True == util.IsEmptyStr(ColName):
|
|
return
|
|
|
|
strSQL = f"UPDATE {Table} SET \'{ColName}\' = \'{Value}\' WHERE id = {nID};"
|
|
self.cursor.execute(strSQL)
|
|
self.conn.commit()
|
|
|
|
def InsertAuthor(self, strAuthor):
|
|
if None == self.cursor:
|
|
return
|
|
|
|
if True == util.IsEmptyStr(strAuthor):
|
|
return
|
|
|
|
strTableName = "authors"
|
|
strSQL = f"SELECT MAX(id) FROM \'{strTableName}\';"
|
|
self.cursor.execute(strSQL)
|
|
nMaxID = self.cursor.fetchone()[0]
|
|
nMaxID += 1
|
|
|
|
strAuthorSort = strAuthor.replace(" ", ",")
|
|
strSQL = f"INSERT OR IGNORE INTO \'{strTableName}\' VALUES ({nMaxID}, \'{strAuthor}\',\'{strAuthorSort}\', \'\');"
|
|
self.cursor.execute(strSQL)
|
|
self.conn.commit()
|
|
|
|
def InsertTag(self, tag):
|
|
if None == self.cursor:
|
|
return
|
|
|
|
if True == util.IsEmptyStr(tag):
|
|
return
|
|
|
|
strTableName = "tags"
|
|
strSQL = f"SELECT MAX(id) FROM \'{strTableName}\';"
|
|
self.cursor.execute(strSQL)
|
|
nMaxID = self.cursor.fetchone()[0]
|
|
nMaxID += 1.
|
|
|
|
strSQL = f"INSERT OR IGNORE INTO \'{strTableName}\' VALUES ({nMaxID}, \'{strAuthor}\', \'\');"
|
|
self.cursor.execute(strSQL)
|
|
self.conn.commit()
|
|
|
|
def main():
|
|
db = MgrCalibreDB("/Users/minarinari/캘리버 서재/metadata.db")
|
|
db.init()
|
|
|
|
#db.UpdateTableValue("books", 3, "title", "삭막")
|
|
db.InsertAuthor("Oyster")
|
|
db.InsertAuthor("Rustle")
|
|
db.InsertAuthor("ShindoL")
|
|
|
|
listValues = db.GetTableAll("authors")
|
|
print(listValues)
|
|
|
|
# For Main Loop
|
|
if __name__ == '__main__':
|
|
main()
|
|
|
|
|
|
# 테이블 생성
|
|
#cursor.execute('''CREATE TABLE IF NOT EXISTS users
|
|
# (id INTEGER PRIMARY KEY AUTOINCREMENT,
|
|
# name TEXT NOT NULL,
|
|
# age INTEGER,
|
|
# email TEXT)''')
|
|
|
|
# 데이터 삽입
|
|
#cursor.execute("INSERT INTO users (name, age, email) VALUES (?, ?, ?)",
|
|
# ("Alice", 30, "alice@example.com"))
|
|
#conn.commit()
|
|
|
|
# 데이터 조회
|
|
# cursor.execute("SELECT * FROM books")
|
|
# rows = cursor.fetchall()
|
|
# for nRow in range(0, len(rows)):
|
|
# row = rows[nRow]
|
|
# for nCol in range(0,len(row)):
|
|
# item = row[nCol]
|
|
|
|
# print(f"{nRow}:{nCol} -> {item}")
|
|
|
|
# print("\r\n")
|
|
|
|
# 데이터 업데이트
|
|
#cursor.execute("UPDATE users SET age = ? WHERE name = ?", (31, "Alice"))
|
|
#conn.commit()
|
|
|
|
# 데이터 삭제
|
|
#cursor.execute("DELETE FROM users WHERE name = ?", ("Alice",))
|
|
#conn.commit()
|
|
|
|
# 연결 닫기
|
|
#conn.close() |