virtual_oulu/database.py at master · DF-thangld/virtual_oulu · 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
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
import config
import sqlite3

def create_connection():
    conn = sqlite3.connect(config.DATABASE_FILE)
    conn.row_factory = sqlite3.Row
    return conn

def close_connection(connection, commit=False):
    if commit:
        connection.commit()
    connection.close()

def run_command(sql_script, parameter=None, connection=None, commit=True):
    if parameter is None:
        parameter = ()
    conn = None
    if connection is None:
        conn = create_connection()
    else:
        conn = connection
    cur = conn.cursor()

    cur.execute(sql_script, parameter)

    if commit:
        conn.commit()

    if connection is None:
        close_connection(connection)

def commit(connection):
    connection.commit()

def roll_back(connection):
    pass

def query_one(sql_script, parameter=None, connection=None):
    conn = None
    if parameter is None:
        parameter = ()
    if connection is None:
        conn = create_connection()
    else:
        conn = connection
    cur = conn.cursor()

    cur.execute(sql_script, parameter)
    row = cur.fetchone()

    if connection is None:
        close_connection(connection)

    return row

def query_all(sql_script, parameter=None, connection=None):
    conn = None
    if parameter is None:
        parameter = ()
    if connection is None:
        conn = create_connection()
    else:
        conn = connection
    cur = conn.cursor()

    cur.execute(sql_script, parameter)
    row = cur.fetchall()

    if connection is None:
        close_connection(connection)

    return row