-
Notifications
You must be signed in to change notification settings - Fork 0
/
DB_connector.py
37 lines (28 loc) · 1.25 KB
/
DB_connector.py
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 mysql.connector
from mysql.connector import Error
class DBConnector(object):
__connection = None
@classmethod
def get_connection_object(cls, host, user, pwd, db_name):
if cls.__connection is None:
try:
cls.__connection = mysql.connector.connect(host=host, user=user, password=pwd)
print("Connected to Server: Admin")
cursor = cls.__connection.cursor()
result = None
cursor.execute("SHOW DATABASES")
result = cursor.fetchall()
if ("{0}".format(db_name),) not in result:
cursor.execute("CREATE DATABASE {0}".format(db_name))
cls.__connection.commit()
print("Created Database: ", db_name)
cls.__connection = mysql.connector.connect(host=host, user=user, password=pwd, database=db_name)
print("Connected to Database: ", db_name)
except Error as e:
print("Error: ", e)
return cls.__connection
def __init__(self):
if DBConnector.__connection is not None:
raise Exception("Only Single Instance is Permitted!!!")
else:
DBConnector.__connection = self