TOOLBOX_Classic/SQLite.py at master · YUDONGLING/TOOLBOX_Classic · 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
73
74
75
76
77
78
79
80
81
82
83
84
85
def CreateDB(Cfg = None):
    import os
    import sqlite3

    if __name__ == '__main__':
        from  Merge import MergeCfg
    else:
        from .Merge import MergeCfg

    Config = {
        'Folder': '',
        'DB'    : ''
    }
    Config = MergeCfg(Config, Cfg)

    Response = {
        'ErrorCode': 0,
        'ErrorMsg' : '',
        'Folder'   : '',
        'File'     : '',
        'Path'     : '',
        'Size'     : -1
    }

    # MAKE FOLDER
    try:
        Folder = Config['Folder'] if Config['Folder'] != '' else '.'
        os.makedirs(Folder, exist_ok = True)
    except Exception as errorMsg:
        Response['ErrorCode'] = 50001
        Response['ErrorMsg']  = f'Fail to make folder, {str(errorMsg).lower().rstrip(".")}'
        return Response

    # CREATE DB
    try:
        File = Config['DB']
        Path = os.path.join(Folder, File)

        _ = sqlite3.connect(Path)
        _.close()

        Response['Folder'] = Folder
        Response['File']   = File
        Response['Path']   = Path
        Response['Size']   = os.path.getsize(Path)
    except Exception as errorMsg:
        Response['ErrorCode'] = 50002
        Response['ErrorMsg']  = f'Fail to create database, {str(errorMsg).lower().rstrip(".")}'
        return Response

    return Response


def ExecuteDB(Cfg = None):
    import os
    import sqlite3

    if __name__ == '__main__':
        from  Merge import MergeCfg
    else:
        from .Merge import MergeCfg

    Config = {
        'Folder': '',
        'DB'    : '',
        'Query' : '',
        'Param' : ()
    }
    Config = MergeCfg(Config, Cfg)

    Response = {
        'ErrorCode': 0,
        'ErrorMsg' : '',
        'Data'     : []
    }

    try:
        DB = sqlite3.connect(os.path.join(Config['Folder'] if Config['Folder'] != '' else '.', Config['DB']))
        CR = DB.cursor(); CR.execute(Config['Query'], Config['Param'])
        Response['Data'] = CR.fetchall()
        DB.commit(); DB.close()
    except Exception as errorMsg:
        Response['ErrorCode'] = 50000
        Response['ErrorMsg']  = f'Fail to operate database, {str(errorMsg).lower().rstrip(".")}'
    return Response