fix: MySQL et al. super calls (#23971)

This commit is contained in:
John Bodley 2023-05-08 14:12:39 -07:00 committed by GitHub
parent 724fd82919
commit 2af76fc41f
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 8 additions and 6 deletions

View File

@ -205,9 +205,13 @@ class MySQLEngineSpec(BaseEngineSpec, BasicParametersMixin):
catalog: Optional[str] = None,
schema: Optional[str] = None,
) -> Tuple[URL, Dict[str, Any]]:
uri, new_connect_args = super(
MySQLEngineSpec, MySQLEngineSpec
).adjust_engine_params(uri, connect_args, catalog, schema)
uri, new_connect_args = super().adjust_engine_params(
uri,
connect_args,
catalog,
schema,
)
if schema:
uri = uri.set(database=parse.quote(schema, safe=""))

View File

@ -261,9 +261,7 @@ class OcientEngineSpec(BaseEngineSpec):
cls, cursor: Any, limit: Optional[int] = None
) -> List[Tuple[Any, ...]]:
try:
rows: List[Tuple[Any, ...]] = super(OcientEngineSpec, cls).fetch_data(
cursor, limit
)
rows: List[Tuple[Any, ...]] = super().fetch_data(cursor, limit)
except Exception as exception:
with OcientEngineSpec.query_id_mapping_lock:
del OcientEngineSpec.query_id_mapping[