Browse Source

added function to check if database already exists

bugfixes
harshadyeola 10 years ago
parent
commit
564477c336
  1. 15
      ee/core/mysql.py

15
ee/core/mysql.py

@ -1,5 +1,6 @@
"""EasyEngine MySQL core classes.""" """EasyEngine MySQL core classes."""
import pymysql import pymysql
from pymysql import connections, DatabaseError
import configparser import configparser
from os.path import expanduser from os.path import expanduser
import sys import sys
@ -102,3 +103,17 @@ class EEMysql():
except Exception as e: except Exception as e:
Log.error(self, "Error: process exited with status %s" Log.error(self, "Error: process exited with status %s"
% e) % e)
def check_db_exists(self, db_name):
try:
connection = connections.Connection(db=db_name,
read_default_file='~/.my.cnf')
if connection:
return True
except DatabaseError as e:
if e.args[1] == '#42000Unknown database \'{0}\''.format(db_name):
return False
else:
raise RuntimeError
except Exception as e:
Log.error(self, "Runtime Exception occured")

Loading…
Cancel
Save