Fix the types

camptocamp · Feb 1, 2023 · e495732 · e495732
1 parent 132b88a
commit e495732
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 3 deletions.
diff --git a/c2cwsgiutils/sql_profiler/_impl.py b/c2cwsgiutils/sql_profiler/_impl.py
@@ -45,7 +45,12 @@ def profile(
                     LOG.info("parameters: %s", repr(parameters))
                     with conn.engine.begin() as c:
                         output = "\n  ".join(
-                            [row[0] for row in c.execute("EXPLAIN ANALYZE " + statement, parameters)]
+                            [
+                                row[0]
+                                for row in c.execute(
+                                    sqlalchemy.text(f"EXPLAIN ANALYZE {statement}"), parameters
+                                )
+                            ]
                         )
                     LOG.info(output)
                 except Exception:  # nosec  # pylint: disable=broad-except

diff --git a/c2cwsgiutils/sqlalchemylogger/handlers.py b/c2cwsgiutils/sqlalchemylogger/handlers.py
@@ -98,9 +98,11 @@ def create_db(self) -> None:
         if not isinstance(self.Log.__table_args__, type(None)) and self.Log.__table_args__.get(
             "schema", None
         ):
-            if not self.engine.dialect.has_schema(self.engine, self.Log.__table_args__["schema"]):
+            if not self.engine.dialect.has_schema(
+                self.engine.raw_connection, self.Log.__table_args__["schema"]
+            ):
                 with self.engine.begin() as connection:
-                    connection.execute(sqlalchemy.schema.CreateSchema(self.Log.__table_args__["schema"]))
+                    connection.execute(sqlalchemy.schema.CreateSchema(self.Log.__table_args__["schema"]))  # type: ignore
         Base.metadata.create_all(self.engine)
 
     def emit(self, record: Any) -> None: