NoSuchModuleError：无法加载插件：sqlalchemy.dialects：amazonathena

Question

我对 python 和 athena 非常陌生。 我正在尝试使用 python 查询 athena。

代码：

import pandas
from sqlalchemy.engine import create_engine

engine = create_engine("amazonathena:///?AccessKey='<key>'&SecretKey='<secretKey>'&Region='IRELAND'&Database='<dbName>'&S3StagingDirectory='s3://<>'")
df = pandas.read_sql("SELECT doctype FROM dbName.tableName limit 10;", engine)
print(df)

错误：

NoSuchModuleError: Can't load plugin: sqlalchemy.dialects:amazonathena

我也尝试过

pip install --upgrade amazonathena-sqlalchemy

，但没有运气，它说

ERROR: Could not find a version that satisfies the requirement amazonathena-sqlalchemy (from versions: none)
ERROR: No matching distribution found for amazonathena-sqlalchemy

如何解决这个问题？我哪里出错了？

Answer 1

根据 SQLAlchemy 文档 Amazon Athena 的外部方言是 PyAthena

可以通过

安装

pip install PyAthena[SQLAlchemy]

您可能需要转义方括号：

pip install PyAthena\[SQLAlchemy\]