python_sqlclient/main.py at master · tangingw/python_sqlclient · 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
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
import json
import os
import re
import sys
import waitress
from interface.view import initial_interface
from interface.view import database_interface
from tools.init import init_db


def db_main(sql_lib_definition_dict: dict):

    database_type, database_nickname = None, None

    print("\nWelcome to the Universal SQL Client:\n")

    if len(sys.argv) == 2:

        database_type = sql_lib_definition_dict[sys.argv[1]]
        database_nickname = initial_interface()

    elif len(sys.argv) > 2:

        database_type = sql_lib_definition_dict[sys.argv[1]]
        database_nickname = sys.argv[2]

    database_interface(database_type, database_nickname)


def main():

    sql_lib_definition_dict = None

    with open("config/db_definition.json", "r") as definition_obj:

        sql_lib_definition_dict = json.loads(
            definition_obj.read()
        )

    if len(sys.argv) >= 2 and len(sys.argv) <= 3:

        if re.search(r"^(?i)(init|del)$", sys.argv[1]):

            init_db()

        else:

            db_main(sql_lib_definition_dict)

        #Clear the environment variable after the webapp
        del os.environ["DB_NAME"]
        del os.environ["DB_TYPE"]

    elif len(sys.argv) > 3 and len(sys.argv) <= 4:

        os.environ["DB_TYPE"] = sql_lib_definition_dict[sys.argv[2]]
        os.environ["DB_NAME"] = sys.argv[3]

        if sys.argv[1] == "webapp":

            from app.webapp import app

            app.run(host="127.0.0.1", port=5000)

        elif sys.argv[1] == "falcon":

            from app.webapp_falcon import app

            waitress.serve(app, host='127.0.0.1', port=8041, url_scheme='https')

        del os.environ["DB_TYPE"]
        del os.environ["DB_NAME"]

    else:

        print(
        """
        Help:

            python init.py init                                 -- to create a DB metadata
            python init.py del <db_nickname>                    -- to delete a DB metadata

            e.g. python init.py del yahoo_1223

            python main.py <sql_engine>                         -- to enter the client prompt
            python main.py <sql_engine> <db_nickname>           -- to connect a DB's client prompt

            e.g.
            python main.py mysql yahoo_1223

            Webapp
            python main.py webapp <sql_engine> <db_nickname>    -- to run webapp

        """
        )


if __name__ == "__main__":

    main()