python-sqlite3-parent-class/DbClass.py at main · Miko0187/python-sqlite3-parent-class · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
import sqlite3

class Database:

    def __init__(self, path: str) -> None:
        self.conn = sqlite3.connect(path)
        self.cursor = self.conn.cursor()

    def execute(self, code) -> None:
        self.cursor.execute(code)

        self.conn.commit()

    def execute_args(self, code, args) -> None:
        self.cursor.execute(code, args)

        self.conn.commit()

    def select_all(self, code) -> list:
        self.cursor.execute(code)

        return self.cursor.fetchall()

    def select_all_args(self, code, args) -> list:
        self.cursor.execute(code, args)

        return self.cursor.fetchall()

    def select_one(self, code):
        self.cursor.execute(code)

        return self.cursor.fetchone()

    def select_one_args(self, code, args):
        self.cursor.execute(code, args)

        return self.cursor.fetchone()