storage: Ensure pymysql is prefered

copy of Ifc6932701da647e50b7b8180372af1157d168793

Change-Id: I396674398d61599c0bf9e42077d877661a7d7405
This commit is contained in:
gord chung 2017-05-25 19:02:10 +00:00
parent 89a6baf3d6
commit d63184a42a
1 changed files with 12 additions and 1 deletions

View File

@ -23,6 +23,7 @@ from oslo_db.sqlalchemy import utils as oslo_sql_utils
from oslo_log import log
from oslo_utils import timeutils
import sqlalchemy as sa
from sqlalchemy.engine import url as sqlalchemy_url
from panko import storage
from panko.storage import base
@ -133,7 +134,17 @@ class Connection(base.Connection):
# oslo.db doesn't support options defined by Panko
for opt in storage.OPTS:
options.pop(opt.name, None)
self._engine_facade = db_session.EngineFacade(url, **options)
self._engine_facade = db_session.EngineFacade(self.dress_url(url),
**options)
@staticmethod
def dress_url(url):
# If no explicit driver has been set, we default to pymysql
if url.startswith("mysql://"):
url = sqlalchemy_url.make_url(url)
url.drivername = "mysql+pymysql"
return str(url)
return url
def upgrade(self):
engine = self._engine_facade.get_engine()