pythonでDB選択コマンドを実装する

ソース

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

#!/usr/bin/env python

# pip install colorama
import colorama
from colorama import Fore, Back, Style

import os

if __name__ == "__main__":
msg_list_1 = [
'1:MySQL',
'2:PostgreSQL',
'99:exit',
Fore.YELLOW + '接続するホストのNo.を入力してください。' + Style.RESET_ALL,
]
print("\n".join(msg_list_1))
selected_num_1 = int(input('>> '))

if selected_num_1 == 1:
msg_list_2= [
Fore.BLUE + 'MySQLへ接続します。' + Style.RESET_ALL,
'1:MySQL_DB_101',
'2:MySQL_DB_102',
'99:exit',
Fore.YELLOW + '選択するDBのNo.を入力してください。' + Style.RESET_ALL,
]
print("\n".join(msg_list_2))
selected_num_2 = int(input('>> '))
if selected_num_2 == 1:
os.system('mysql -u user_name -h 127.0.0.1 -D mysql_db_1 --password=*****')
elif selected_num_2 == 2:
os.system('mysql -u user_name -h 127.0.0.1 -D mysql_db_2 --password=*****')
elif selected_num_2 == 99:
print("exit")

elif selected_num_1 == 2:
msg_list_2= [
Fore.BLUE + 'PostgreSQLへ接続します。' + Style.RESET_ALL,
'1:Postgre_DB_101',
'2:Postgre_DB_102',
'99:exit',
Fore.YELLOW + '選択するDBのNo.を入力してください。' + Style.RESET_ALL,
]
print("\n".join(msg_list_2))
selected_num_2 = int(input('>> '))
if selected_num_2 == 1:
os.system('psql -h localhost -U user_name -d postgre_db_1')
elif selected_num_2 == 2:
os.system('psql -h localhost -U user_name -d postgre_db_2')
elif selected_num_2 == 99:
print("exit")

elif selected_num_1 == 99:
print("exit")

実行

1
$ python db_select.py