集成 SQLAlchemy

Note

在 PyODPS 0.10.0 中开始支持

PyODPS 支持集成 SQLAlchemy，可以使用 SQLAlchemy 查询 MaxCompute 数据。

创建连接

创建连接可以在连接串中指定 access_id、access_key 和 project 等。

from sqlalchemy import create_engine
engine = create_engine('odps://<access_id>:<access_key>@<project>')

要在连接串中指定 endpoint，可以按如下方式：

from sqlalchemy import create_engine
engine = create_engine('odps://<access_id>:<access_key>@<project>/?endpoint=<endpoint>')

这里把 <access_id> 等替换成相应的账号。

对于已有的 ODPS 对象 o ，调用 o.to_global() 设为全局账号后，在连接串中就不需要指定了。

from sqlalchemy import create_engine
o.to_global()  # set ODPS object as global one
engine = create_engine('odps://')

接着创建连接。

conn = engine.connect()

调用 SQLAlchemy 接口

创建了连接之后，就可以正常调用 SQLAlchemy 接口。以下对建表、写入数据、查询分别举例说明。

建表

from sqlalchemy import Table, Column, Integer, String, MetaData
metadata = MetaData()

users = Table('users', metadata,
    Column('id', Integer),
    Column('name', String),
    Column('fullname', String),
)

metadata.create_all(engine)

写入数据

ins = users.insert().values(id=1, name='jack', fullname='Jack Jones')
conn.execute(ins)

查询数据

>>> from sqlalchemy.sql import select
>>> s = select([users])
>>> result = conn.execute(s)
>>> for row in result:
>>>     print(row)
(1, 'jack', 'Jack Jones')