mirror of
https://github.com/PyMySQL/mysqlclient.git
synced 2025-08-15 11:10:58 +08:00
26 lines
581 B
Python
26 lines
581 B
Python
"""Configure database connection for tests."""
|
|
|
|
from os import environ, path
|
|
|
|
tests_path = path.dirname(__file__)
|
|
conf_file = environ.get('TESTDB', 'default.cnf')
|
|
conf_path = path.join(tests_path, conf_file)
|
|
connect_kwargs = dict(
|
|
read_default_file = conf_path,
|
|
read_default_group = "MySQLdb-tests",
|
|
)
|
|
|
|
def connection_kwargs(kwargs):
|
|
db_kwargs = connect_kwargs.copy()
|
|
db_kwargs.update(kwargs)
|
|
return db_kwargs
|
|
|
|
def connection_factory(**kwargs):
|
|
import MySQLdb
|
|
db_kwargs = connection_kwargs(kwargs)
|
|
db = MySQLdb.connect(**db_kwargs)
|
|
return db
|
|
|
|
|
|
|