INNER CODE UNIT · Python
mysql_connection
bitpoke/mysql-operator · hack/development/stress_test.py:66
def mysql_connection(dsn):
dsn = urlparse(dsn)
db_name = dsn.path.strip('/')
if len(db_name)==0:
raise RuntimeError('Database name not corectly specifyied: {}'.format(db_name))
try:
db = mysql.connector.connect(
host=dsn.hostname,
port=dsn.port,
user=dsn.username,
passwd=dsn.password,
database=db_name,
)
except mysql.connector.errors.ProgrammingError as e:
if 'Unknown database' in e.msg:
db = mysql.connector.connect(
host=dsn.hostname,