Move async dialects to pyathena/aio/sqlalchemy/ and add aiosqlalchemy extra

- Move async dialect files from pyathena/sqlalchemy/async_*.py to
  pyathena/aio/sqlalchemy/ to follow existing async code structure
- Add aiosqlalchemy optional dependency (sqlalchemy[asyncio]>=2.0.0)
  which pulls in greenlet required by await_only()

Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>

Author: laughingman7743

pyathena/aio/sqlalchemy/__init__.py

@@ -0,0 +1 @@
+# -*- coding: utf-8 -*-

pyathena/sqlalchemy/async_arrow.py pyathena/aio/sqlalchemy/arrow.pypyathena/sqlalchemy/async_arrow.py renamed to pyathena/aio/sqlalchemy/arrow.py

@@ -1,7 +1,7 @@
 # -*- coding: utf-8 -*-
 from typing import TYPE_CHECKING
 
-from pyathena.sqlalchemy.async_base import AthenaAioDialect
+from pyathena.aio.sqlalchemy.base import AthenaAioDialect
 from pyathena.util import strtobool
 
 if TYPE_CHECKING:
@@ -31,7 +31,7 @@ class AthenaAioArrowDialect(AthenaAioDialect):
 
     See Also:
         :class:`~pyathena.aio.arrow.cursor.AioArrowCursor`: The underlying async cursor.
-        :class:`~pyathena.sqlalchemy.async_base.AthenaAioDialect`: Base async dialect.
+        :class:`~pyathena.aio.sqlalchemy.base.AthenaAioDialect`: Base async dialect.
     """
 
     driver = "aioarrow"

pyathena/sqlalchemy/async_base.py pyathena/aio/sqlalchemy/base.pypyathena/sqlalchemy/async_base.py renamed to pyathena/aio/sqlalchemy/base.py

@@ -1,7 +1,7 @@
 # -*- coding: utf-8 -*-
 from typing import TYPE_CHECKING
 
-from pyathena.sqlalchemy.async_base import AthenaAioDialect
+from pyathena.aio.sqlalchemy.base import AthenaAioDialect
 from pyathena.util import strtobool
 
 if TYPE_CHECKING:
@@ -33,7 +33,7 @@ class AthenaAioPandasDialect(AthenaAioDialect):
 
     See Also:
         :class:`~pyathena.aio.pandas.cursor.AioPandasCursor`: The underlying async cursor.
-        :class:`~pyathena.sqlalchemy.async_base.AthenaAioDialect`: Base async dialect.
+        :class:`~pyathena.aio.sqlalchemy.base.AthenaAioDialect`: Base async dialect.
     """
 
     driver = "aiopandas"

pyathena/sqlalchemy/async_pandas.py pyathena/aio/sqlalchemy/pandas.pypyathena/sqlalchemy/async_pandas.py renamed to pyathena/aio/sqlalchemy/pandas.py

@@ -1,7 +1,7 @@
 # -*- coding: utf-8 -*-
 from typing import TYPE_CHECKING
 
-from pyathena.sqlalchemy.async_base import AthenaAioDialect
+from pyathena.aio.sqlalchemy.base import AthenaAioDialect
 from pyathena.util import strtobool
 
 if TYPE_CHECKING:
@@ -31,7 +31,7 @@ class AthenaAioPolarsDialect(AthenaAioDialect):
 
     See Also:
         :class:`~pyathena.aio.polars.cursor.AioPolarsCursor`: The underlying async cursor.
-        :class:`~pyathena.sqlalchemy.async_base.AthenaAioDialect`: Base async dialect.
+        :class:`~pyathena.aio.sqlalchemy.base.AthenaAioDialect`: Base async dialect.
     """
 
     driver = "aiopolars"

pyathena/sqlalchemy/async_polars.py pyathena/aio/sqlalchemy/polars.pypyathena/sqlalchemy/async_polars.py renamed to pyathena/aio/sqlalchemy/polars.py

@@ -1,7 +1,7 @@
 # -*- coding: utf-8 -*-
 from typing import TYPE_CHECKING
 
-from pyathena.sqlalchemy.async_base import AthenaAioDialect
+from pyathena.aio.sqlalchemy.base import AthenaAioDialect
 
 if TYPE_CHECKING:
     from types import ModuleType
@@ -26,7 +26,7 @@ class AthenaAioRestDialect(AthenaAioDialect):
 
     See Also:
         :class:`~pyathena.aio.cursor.AioCursor`: The underlying async cursor.
-        :class:`~pyathena.sqlalchemy.async_base.AthenaAioDialect`: Base async dialect.
+        :class:`~pyathena.aio.sqlalchemy.base.AthenaAioDialect`: Base async dialect.
     """
 
     driver = "aiorest"

pyathena/sqlalchemy/async_rest.py pyathena/aio/sqlalchemy/rest.pypyathena/sqlalchemy/async_rest.py renamed to pyathena/aio/sqlalchemy/rest.py

@@ -1,7 +1,7 @@
 # -*- coding: utf-8 -*-
 from typing import TYPE_CHECKING
 
-from pyathena.sqlalchemy.async_base import AthenaAioDialect
+from pyathena.aio.sqlalchemy.base import AthenaAioDialect
 
 if TYPE_CHECKING:
     from types import ModuleType
@@ -25,7 +25,7 @@ class AthenaAioS3FSDialect(AthenaAioDialect):
 
     See Also:
         :class:`~pyathena.aio.s3fs.cursor.AioS3FSCursor`: The underlying async cursor.
-        :class:`~pyathena.sqlalchemy.async_base.AthenaAioDialect`: Base async dialect.
+        :class:`~pyathena.aio.sqlalchemy.base.AthenaAioDialect`: Base async dialect.
     """
 
     driver = "aios3fs"

pyathena/sqlalchemy/async_s3fs.py pyathena/aio/sqlalchemy/s3fs.pypyathena/sqlalchemy/async_s3fs.py renamed to pyathena/aio/sqlalchemy/s3fs.py

@@ -42,11 +42,11 @@ awsathena = "pyathena.sqlalchemy.base:AthenaDialect"
 "awsathena.arrow" = "pyathena.sqlalchemy.arrow:AthenaArrowDialect"
 "awsathena.polars" = "pyathena.sqlalchemy.polars:AthenaPolarsDialect"
 "awsathena.s3fs" = "pyathena.sqlalchemy.s3fs:AthenaS3FSDialect"
-"awsathena.aiorest" = "pyathena.sqlalchemy.async_rest:AthenaAioRestDialect"
-"awsathena.aiopandas" = "pyathena.sqlalchemy.async_pandas:AthenaAioPandasDialect"
-"awsathena.aioarrow" = "pyathena.sqlalchemy.async_arrow:AthenaAioArrowDialect"
-"awsathena.aiopolars" = "pyathena.sqlalchemy.async_polars:AthenaAioPolarsDialect"
-"awsathena.aios3fs" = "pyathena.sqlalchemy.async_s3fs:AthenaAioS3FSDialect"
+"awsathena.aiorest" = "pyathena.aio.sqlalchemy.rest:AthenaAioRestDialect"
+"awsathena.aiopandas" = "pyathena.aio.sqlalchemy.pandas:AthenaAioPandasDialect"
+"awsathena.aioarrow" = "pyathena.aio.sqlalchemy.arrow:AthenaAioArrowDialect"
+"awsathena.aiopolars" = "pyathena.aio.sqlalchemy.polars:AthenaAioPolarsDialect"
+"awsathena.aios3fs" = "pyathena.aio.sqlalchemy.s3fs:AthenaAioS3FSDialect"
 
 [project.optional-dependencies]
 sqlalchemy = ["sqlalchemy>=1.0.0"]
@@ -61,6 +61,9 @@ arrow = [
 polars = [
     "polars>=1.0.0",
 ]
+aiosqlalchemy = [
+    "sqlalchemy[asyncio]>=2.0.0",
+]
 
 [dependency-groups]
 dev = [