官方文档:
- https://docs.sqlalchemy.org/en/14/tutorial/engine.html
- https://docs.sqlalchemy.org/en/14/tutorial/dbapi_transactions.html
引擎(Engine)
创建任何 SQLAlchemy 应用程序,都需要先创建一个引擎(Engine)对象。该对象作为连接到特定数据库的中心源,为这些数据库连接提供一个工厂和一个称为连接池的保存空间。
引擎通常是一个全局对象,只为特定的数据库服务器创建一次,并使用一个 URL
字符串进行配置,描述它应该如何连接到数据库主机或后端。
在本教程中,我们将使用一个 SQLite 数据库。这是一种简单的测试方法,不需要预先设置一个实际存在的数据库。引擎是通过使用 create_engine()
方法创建的,指定 create_engine.future
参数为 True
,这样我们就可以完全使用 2.0 风格的语法。
from sqlalchemy import create_engine
engine = create_engine('sqlite+pysqlite:///memory.db', echo=True, future=True)
create_engine()
方法的主要参数是一个字符串 URL,它定义了以下信息:
定义了要链接的数据库类型,本例是
sqlite
。定义了使用什么 DBAPI,Python DBAPI 是一个第三方驱动,SQLAlchemy 用它来与一个特定的数据库进行交互,本例中使用的是
pysqlite
。
3.数据库存放的位置,本例中使用相对路径,将在根目录中创建 memory.db
数据库文件。
我们还指定了一个参数 create_engine.echo
为 True
,它会使标准 SQL 语句在终端中输出。
建立连接(Connection)
从面向用户的角度来看,引擎对象的唯一目的是提供一个与数据库的连接单元,称为 Connection。
我们的教程还没进入到 ORM 部分,所以下面例子中我们依然使用原生 SQL 语句(textual SQL)来操作数据库,text
方法可以让链接对象接受原生 SQL 语句。
下面是一个创建数据库链接,创建数据表(Table),并插入数据的例子。
from sqlalchemy import create_engine, text
engine = create_engine('sqlite+pysqlite:///memory.db', echo=True, future=True)
with engine.connect() as conn:
conn.execute(text("CREATE TABLE some_table (x int, y int)"))
conn.execute(
text("INSERT INTO some_table (x, y) VALUES (:x, :y)"),
[{"x": 1, "y": 1}, {"x": 2, "y": 4}]
)
conn.commit()
执行上述代码后,会在根目录创建 memory.db
数据库文件;CREATE TABLE
语句会新建 some_table
表;INSERT
语句定义了如何插入数据;Connection.commit()
方法提交上述对数据库的操作语句。
除了使用 commit
方法来提交改变之外,还可以使用 Engine.begin()
方法。
with engine.begin() as conn:
conn.execute(text("CREATE TABLE some_table (x int, y int)"))
conn.execute(text(
"INSERT INTO some_table (x, y) VALUES (:x, :y)"),
[{"x": 1, "y": 1}, {"x": 2, "y": 4}]
)
再看一个用 select
语句执行查找的例子:
with engine.connect() as conn:
result = conn.execute(text("SELECT x, y FROM some_table"))
for row in result:
print(f"x: {row.x} y: {row.y}")
输出:
x: 1 y: 1
x: 2 y: 4
使用参数
使用冒号来标记参数,本例中是 :y
;将筛选出 y
小于 2 的行。
with engine.connect() as conn:
result = conn.execute(
text("SELECT x, y FROM some_table WHERE y < :y"),
{"y": 2}
)
for row in result:
print(f"x: {row.x} y: {row.y}")
使用多个参数
with engine.connect() as conn:
conn.execute(
text("INSERT INTO some_table (x, y) VALUES (:x, :y)"),
[{"x": 11, "y": 12}, {"x": 13, "y": 14}]
)
conn.commit()
把参数和语句绑定
stmt = text("SELECT x, y FROM some_table WHERE y > :y ORDER BY x, y").bindparams(y=2)
with engine.connect() as conn:
result = conn.execute(stmt)
for row in result:
print(f"x: {row.x} y: {row.y}")
使用 Session
上面举得使用 Connection
的例子在 ORM 中也能适用,但我们习惯上更多使用 Session
。当使用 ORM 时,基本的数据库交互对象被称为 Session
。在 SQLAlchemy 中,该对象的使用方式与 Connection
非常相似,事实上,在使用 Session
时,它在内部指的是 Connection
。
from sqlalchemy import create_engine, text
from sqlalchemy.orm import Session
stmt = text("SELECT x, y FROM some_table WHERE y > :y ORDER BY x, y").bindparams(y=2)
with Session(engine) as session:
result = session.execute(stmt)
for row in result:
print(f"x: {row.x} y: {row.y}")