def create_database(self):
""" Create database """
db_config = Dbtools.read_db_config()
db_name = db_config['database']
conx = MySQLConnection(user=db_config['user'],
password=db_config['password'])
print('Connecting to MySQL...')
cursor = conx.cursor()
try:
cursor.execute("DROP DATABASE IF EXISTS {}".format(db_name))
cursor.execute("CREATE DATABASE {} \
DEFAULT CHARACTER SET 'utf8'".format(db_name))
print('Creating database {}...'.format(db_name))
conx.database = db_name
except Error as error:
print(error)
exit(1)
finally:
self.cursor = cursor
self.conx = conx
def connect_db():
conn = MySQLConnection(host=db_config['host'],
user=db_config['user'],
password=db_config['password'])
try:
conn.database = db_config['database']
yield conn.cursor(dictionary=True)
finally:
try:
conn.commit()
finally:
conn.close()
def connect():
""" Connect to MySQL database """
try:
conn = MySQLConnection(
host=database_host,
# database=database_name,
user=database_user,
password=database_password)
if conn.is_connected() and conn.database == None:
if database_name != conn.database:
conn.cursor().execute("DROP DATABASE IF EXISTS " +
database_name)
conn.cursor().execute("CREATE DATABASE " + database_name)
conn.database = database_name
print('База данных ' + database_name + ' созданна, Ура!')
except Error as e:
print(e)
finally:
conn.commit()
conn.close()
