TypeError: cursor() missing 1 required positional argument: 'query'

[ankur09011]

• asyncpg version: 0.24.0
• Python version: 3.8
• Platform: Ubuntu
• Do you use pgbouncer?: No
• Did you install asyncpg with pip?: Yes
• If you built asyncpg locally, which version of Cython did you use?:
• Can the issue be reproduced under both asyncio and
  uvloop?:

Issue when using Pandas read_sql function.

async def get_conn():
        return await asyncpg.connect(DATABASE_URL)

conn = asyncio.run(asyncpg.connect(DATABASE_URL))

df = pd.read_sql(sql=query, con=conn)

Traceback:

File "/home/cozmo/cpd/application/try/app/api/utils/query_utils.py", line 100, in <module>
    df = pd.read_sql(sql=query, con=conn)
  File "/home/cozmo/.virtualenvs/cpd_micro/lib/python3.8/site-packages/pandas/io/sql.py", line 602, in read_sql
    return pandas_sql.read_query(
  File "/home/cozmo/.virtualenvs/cpd_micro/lib/python3.8/site-packages/pandas/io/sql.py", line 2116, in read_query
    cursor = self.execute(*args)
  File "/home/cozmo/.virtualenvs/cpd_micro/lib/python3.8/site-packages/pandas/io/sql.py", line 2054, in execute
    cur = self.con.cursor()
TypeError: cursor() missing 1 required positional argument: 'query' ```

[elprans]

I don't think pandas supports asyncpg as the database driver. You need a DB-API compatible one, like psycopg2.

[tehmufifnman]

I don't think pandas supports asyncpg as the database driver. You need a DB-API compatible one, like psycopg2.

Yup I believe this is true as well - from the Pandas docs they specify:
SQLAlchemy connectable, str, or sqlite3 connection

I don't know how performant this is - but since I only use it once in my code it works well enough. This is how I use dataframes with asyncpg:

    async def fetch_as_dataframe(self, sql, *args):
        async with self._pool.acquire() as connection:
            stmt = await connection.prepare(sql)
            columns = [a.name for a in stmt.get_attributes()]
            data = await stmt.fetch(*args)
            return pd.DataFrame(data, columns=columns)