Fix 'execute' so that it can work with sqlalchemy params

asyncpgsa/__init__.py

@@ -2,6 +2,6 @@
 from .pgsingleton import PG
 from .connection import compile_query
 
-__version__ = '0.14.1'
+__version__ = '0.14.2'
 pg = PG()
 

asyncpgsa/connection.py

@@ -101,12 +101,13 @@ class SAConnection(connection.Connection):
 
     def _execute(self, query, args, limit, timeout, return_status=False):
         query, compiled_args = compile_query(query, dialect=self._dialect)
-        args = tuple(compiled_args) + args
+        args = compiled_args or args
         return super()._execute(query, args, limit, timeout,
                                 return_status=return_status)
 
     async def execute(self, script, *args, **kwargs) -> str:
-        # script, params = compile_query(script, dialect=self._dialect)
+        script, params = compile_query(script, dialect=self._dialect)
+        args = params or args
         result = await super().execute(script, *args, **kwargs)
         return RecordGenerator(result)
 

tests/test_pgsingleton.py

@@ -134,6 +134,12 @@ async def test_sql_with_arguments():
     assert bool(result)
 
 
+async def test_execute_with_sa_arguments():
+    script = sa.select('*').select_from(sa.text('sqrt(:num) as a'))
+    script = script.params(num=16)
+    result = await pg.execute(script)
+    assert bool(result)
+
 async def test_transaction():
     async with pg.transaction() as conn:
         for row in await conn.fetch(query):