Files
Auto_Bangumi/src/module/database/connector.py
EstrellaXD 5c71b9d35d - Rewrite config class
- Change multiprocessing to thread
- black all code
2023-05-06 23:47:44 +08:00

40 lines
1.1 KiB
Python

import os
import sqlite3
from module.conf import DATA_PATH
class DataConnector:
def __init__(self):
if not os.path.isfile(DATA_PATH):
os.makedirs(os.path.dirname(DATA_PATH), exist_ok=True)
self._conn = sqlite3.connect(DATA_PATH)
self._cursor = self._conn.cursor()
self._cursor.execute(
"""
CREATE TABLE IF NOT EXISTS bangumi (
id INTEGER PRIMARY KEY,
official_title TEXT NOT NULL,
title_raw TEXT NOT NULL,
season INTEGER NOT NULL,
season_raw TEXT NOT NULL,
subtitle TEXT,
group_name TEXT,
source TEXT,
dpi TEXT,
eps_collect INTEGER NOT NULL,
offset INTEGER NOT NULL,
filter TEXT NOT NULL,
rss TEXT NOT NULL
);
"""
)
self._conn.commit()
def __enter__(self):
return self
def __exit__(self, exc_type, exc_val, exc_tb):
self._conn.close()