import sqlite3
import json


class NovelCiwei:
    def __init__(self, db_path: str):
        self._db = sqlite3.connect(db_path, check_same_thread=False)

    def get_book_in_shelf(self, book_id: int):
        cur = self._db.execute(
            'SELECT book_info FROM shelf_book_info WHERE book_id = ?;',
            [str(book_id)])
        for i in cur:
            return json.loads(i)

    def get_chapter(self, chapter_id: int):
        cur = self._db.execute(
            'SELECT * FROM catalog1 WHERE chapter_id = ?;', [str(chapter_id)])
        cur.row_factory = sqlite3.Row
        return cur.fetchone()