Python 数据库操作 SQLAlchemy示例代码

2019-02-1908:24:57后端程序开发Comments2,186 views字数 8754阅读模式

程序在运行过程中所有的的数据都存储在内存 (RAM) 中,「RAM 是易失性存储器,系统掉电后 RAM 中的所有数据将全部丢失」。在大多数情况下我们希望程序运行中产生的数据能够长久的保存,此时我们就需要将数据保存到磁盘上,无论是保存到本地磁盘,还是通过网络保存到服务器上,最终都会将数据写入磁盘文件。将数据保存在磁盘中我们需要面对一个数据格式的问题,此时就需要引入数据库操作。文章源自菜鸟学院-https://www.cainiaoxueyuan.com/bc/9628.html

数据库是专门用于数据的集中存储和查询的软件,它便于程序保存和读取数据,且能够通过特定的条件查询指定的数据。文章源自菜鸟学院-https://www.cainiaoxueyuan.com/bc/9628.html

Python 的标准数据库接口为 Python DB-API,它为编程人员提供了完善的数据库应用标称接口。但是使用 Python DB-API 需要开发人员自行去拼接 SQL,并把 SQL 做成模板。此时全靠编程人员来保证系统的安全性,完全有人来保证系统的安全性,不可避免的会出现错误,为了减少人为原因产生的错误 ORM 框架应运而生。文章源自菜鸟学院-https://www.cainiaoxueyuan.com/bc/9628.html

ORM 即 Object-Relational Mapping,把关系数据库的表结构映射到对象上面。负责这个转换过程的就是 ORM 框架文章源自菜鸟学院-https://www.cainiaoxueyuan.com/bc/9628.html

Python 中的 ORM 框架主要有 SQLObject、Storm、Django's ORM、peewee、SQLALchemy。每种 ORM 框架都有各自的特点和相应的应用范围,本文主要介绍 SALALchemy,若你对其他框架感兴趣请自行搜索相关内容。文章源自菜鸟学院-https://www.cainiaoxueyuan.com/bc/9628.html

SQLAlchemy 简介文章源自菜鸟学院-https://www.cainiaoxueyuan.com/bc/9628.html

SQLAlchemy 是一个功能强大的开源 Python ORM 工具包。它提供了 “一个知名企业级的持久化模式的,专为高效率和高性能的数据库访问设计的,改编成一个简单的 Python 域语言的完整套件”。它采用了数据映射模式(像 Java 中的 Hibernate)而不是 Active Record 模式(像Ruby on Rails 的 ORM)。文章源自菜鸟学院-https://www.cainiaoxueyuan.com/bc/9628.html

SQLAlchemy官网文章源自菜鸟学院-https://www.cainiaoxueyuan.com/bc/9628.html

SQLAlchemy 的优缺点:文章源自菜鸟学院-https://www.cainiaoxueyuan.com/bc/9628.html

优点:文章源自菜鸟学院-https://www.cainiaoxueyuan.com/bc/9628.html

  • 企业级 API,使得代码有健壮性和适应性。
  • 灵活的设计,使得能轻松完成复杂的数据查询。

缺点:文章源自菜鸟学院-https://www.cainiaoxueyuan.com/bc/9628.html

  • 工作单元概念不常见。
  • 重量级 API,导致长学习曲线。

SQLAlchemy 应用文章源自菜鸟学院-https://www.cainiaoxueyuan.com/bc/9628.html

以下是一段使用 SQLAlchemy 操作 SQLite 数据库的代码文章源自菜鸟学院-https://www.cainiaoxueyuan.com/bc/9628.html

# -*- coding:utf-8 -*-

from sqlalchemy import (
  create_engine,
  Column,
  Integer,
  String,
)

from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy.orm import sessionmaker

engine = create_engine('sqlite:///./sqlalchemy.sqlite', echo=True)

Base = declarative_base()

class User(Base):
  __tablename__ = 'users'

  id = Column(Integer, primary_key=True, autoincrement=True)
  name = Column(String)
  fullname = Column(String)
  nickname = Column(String)

  def __repr__(self):
    return "<User(name='%s', fullname='%s', nickname='%s')>" % (self.name, self.fullname, self.nickname)


db_session = sessionmaker(bind=engine)
session = db_session()

Base.metadata.create_all(engine)

user1 = User(name='ed', fullname='Ed Jones', nickname='edsnickname')
user2 = User(name='wendy', fullname='Wendy Williams', nickname='windy')
user3 = User(name='mary', fullname='Mary Contrary', nickname='mary')

session.add(user1)
session.add(user2)
session.add(user3)
session.commit()

user = session.query(User).filter(User.id<2).all()
print(user)
user = session.query(User).filter(User.id<=5).all()
print(user)

user1.name = 'admin'
session.merge(user1)
user4 = User(name='fred', fullname='Fred Flintstone', nickname='freddy')
session.merge(user4)
session.query(User).filter(User.id==2).update({'name':'test'})
user = session.query(User).filter(User.id<=5).all()
print(user)

在以上代码中我们完成了一下工作:文章源自菜鸟学院-https://www.cainiaoxueyuan.com/bc/9628.html

  1. 连接到数据库「本次我们使用的是 SQLite 数据库」。
  2. 创建数据库表并将其映射到 Python 类上。
  3. 创建数据实例,并将其保存到数据库中。
  4. 对保存在数据库中的数据进行读取和修改。

导入 SQLAlchemy 模块并连接到 SQLite 数据库文章源自菜鸟学院-https://www.cainiaoxueyuan.com/bc/9628.html

SQLAlchemy 通过 create_engine 函数来创建数据库连接。create_engine 函数的第一个参数是数据了 URL,第二个参数 echo 设置为 True 表示在程序的运行过程中我们可以在控制台看到操作所涉及到的 SQL 语句。文章源自菜鸟学院-https://www.cainiaoxueyuan.com/bc/9628.html

在本次示例中我们使用的数据库是 SQLite,你也可以使用其他数据库。只有在调试状态下将 echo 设置为 True,在生产环境请将 echo 设置为 false 或省略 echo 参数。文章源自菜鸟学院-https://www.cainiaoxueyuan.com/bc/9628.html

engine = create_engine('sqlite:///./sqlalchemy.sqlite', echo=True)

create_engine 返回的是一个 Engine 实例,它指向数据库的一些核心接口。SQLAlchemy会根据你选择的数据库配置而调用对应的 DB-API。文章源自菜鸟学院-https://www.cainiaoxueyuan.com/bc/9628.html

create_engine 函数并会不真正建立数据库的 DB-API 连接,当调用 Engine.execute() 或 Engine.connect() 方法时才会建立连接。大多数情况下我们无需关注 Engine,SQLAlchemy 会帮我们处理。文章源自菜鸟学院-https://www.cainiaoxueyuan.com/bc/9628.html

创建数据库表文章源自菜鸟学院-https://www.cainiaoxueyuan.com/bc/9628.html

将 python 类映射到数据库表上,这个 Python 类需要时一个指定基类的子类,这个基类应当含有ORM映射中相关的类和表的信息。这个基类可以通过 declarative_base 方法来创建。
Base = declarative_base()文章源自菜鸟学院-https://www.cainiaoxueyuan.com/bc/9628.html

在这个示例中使用 Base 基类创建了一个 User 的类来作为数据库表。文章源自菜鸟学院-https://www.cainiaoxueyuan.com/bc/9628.html

class User(Base):
  __tablename__ = 'users'

  id = Column(Integer, primary_key=True, autoincrement=True)
  name = Column(String)
  fullname = Column(String)
  nickname = Column(String)

  def __repr__(self):
    return "<User(name='%s', fullname='%s', nickname='%s')>" % (self.name, self.fullname, self.nickname)

在 User 类中我们定义了需要映射到数据库表上的属性,主要包括表名称、列的类型和名称等。这个类至少应包含一个名为 tablename 的属性来给出数据库表的名称,及至少一个给出表的主键「primary key」的列。在 User 类中我们定义了表名称为 users,定义了 id、name、fullname、nickname 四列数据,并设置 id 为表的主键。
创建完成 User 类后,实际在 SQLite 数据库中并不存在该表,此时需要使用 declarative 基类的 Metadata.create_all 在数据库中创建 users 表,在 create_all 方法中我们需要传入参数 Engine。文章源自菜鸟学院-https://www.cainiaoxueyuan.com/bc/9628.html

通过 Metadata.create_all 传入的 Engine 参数,SQLAlchemy 自动实现对数据库的连接。文章源自菜鸟学院-https://www.cainiaoxueyuan.com/bc/9628.html

Base.metadata.create_all(engine)

metadata.create_all 方法执行完成后在 SQLite 数据库即可查到名称为 users 的数据表。文章源自菜鸟学院-https://www.cainiaoxueyuan.com/bc/9628.html

保存数据实例到数据库中文章源自菜鸟学院-https://www.cainiaoxueyuan.com/bc/9628.html

将数据保存到数据库中,我们需要 User 的实例和用于操作数据的 session。文章源自菜鸟学院-https://www.cainiaoxueyuan.com/bc/9628.html

session 是 ORM 数据的接口,可以通过 session 来操作数据库中的数据。文章源自菜鸟学院-https://www.cainiaoxueyuan.com/bc/9628.html

使用已经定义完成的 User 类将数据实例化。文章源自菜鸟学院-https://www.cainiaoxueyuan.com/bc/9628.html

user1 = User(name='ed', fullname='Ed Jones', nickname='edsnickname')
user2 = User(name='wendy', fullname='Wendy Williams', nickname='windy')
user3 = User(name='mary', fullname='Mary Contrary', nickname='mary')

获取 session 首先需要使用 sessionmaker 来得到 session 的工厂类,然后通过工厂类来获取 session。文章源自菜鸟学院-https://www.cainiaoxueyuan.com/bc/9628.html

db_session = sessionmaker(bind=engine)
session = db_session()

session 通过 Engine 与数据库进行关联。session 创建完成后并不会立即打开与数据库的连接,只有当我们第一使用 session 是,才会从 Engine 维护的连接池中取出一个连接来操作数据库,这个连接在我们关闭 session 时会被释放。
获取 session 后可以通过 add 和 commit 方法将数据保存到数据库中。文章源自菜鸟学院-https://www.cainiaoxueyuan.com/bc/9628.html

session.add(user1)
session.add(user2)
session.add(user3)
session.commit()

对数据库中的数据进行查询和修改文章源自菜鸟学院-https://www.cainiaoxueyuan.com/bc/9628.html

SQLAlchemy 通过 query 来对数据进行查询,可以通过 filter 方法对查询结果进行筛选。文章源自菜鸟学院-https://www.cainiaoxueyuan.com/bc/9628.html

user = session.query(User).filter(User.id<2).all()
print(user)
user = session.query(User).filter(User.id<=5).all()
print(user)

以上代码通过 query 获取数据库中所有 User 数据,然后通过 filter 方法筛选出 id 小于 2 和 id 小于等于 5 的数据。文章源自菜鸟学院-https://www.cainiaoxueyuan.com/bc/9628.html

数据库的修改可以通过 merge 和 update 来实现文章源自菜鸟学院-https://www.cainiaoxueyuan.com/bc/9628.html

user1.name = 'admin'
session.merge(user1)
user4 = User(name='fred', fullname='Fred Flintstone', nickname='freddy')
session.merge(user4)
session.query(User).filter(User.id==2).update({'name':'test'})
user = session.query(User).filter(User.id<=5).all()
print(user)

使用 merge 修改数据,当数据中存在该数据时修改,不存在是将当前数据插入数据库中。文章源自菜鸟学院-https://www.cainiaoxueyuan.com/bc/9628.html

代码运行结果文章源自菜鸟学院-https://www.cainiaoxueyuan.com/bc/9628.html

以上示例代码的运行结果如下文章源自菜鸟学院-https://www.cainiaoxueyuan.com/bc/9628.html

2019-02-16 21:45:23,919 INFO sqlalchemy.engine.base.Engine SELECT CAST('test plain returns' AS VARCHAR(60)) AS anon_1
2019-02-16 21:45:23,919 INFO sqlalchemy.engine.base.Engine ()
2019-02-16 21:45:23,919 INFO sqlalchemy.engine.base.Engine SELECT CAST('test unicode returns' AS VARCHAR(60)) AS anon_1
2019-02-16 21:45:23,919 INFO sqlalchemy.engine.base.Engine ()
2019-02-16 21:45:23,920 INFO sqlalchemy.engine.base.Engine PRAGMA table_info("users")
2019-02-16 21:45:23,920 INFO sqlalchemy.engine.base.Engine ()
2019-02-16 21:45:23,921 INFO sqlalchemy.engine.base.Engine
CREATE TABLE users (
  id INTEGER NOT NULL,
  name VARCHAR,
  fullname VARCHAR,
  nickname VARCHAR,
  PRIMARY KEY (id)
)


2019-02-16 21:45:23,921 INFO sqlalchemy.engine.base.Engine ()
2019-02-16 21:45:23,922 INFO sqlalchemy.engine.base.Engine COMMIT
2019-02-16 21:45:23,924 INFO sqlalchemy.engine.base.Engine BEGIN (implicit)
2019-02-16 21:45:23,925 INFO sqlalchemy.engine.base.Engine INSERT INTO users (name, fullname, nickname) VALUES (?, ?, ?)
2019-02-16 21:45:23,925 INFO sqlalchemy.engine.base.Engine ('ed', 'Ed Jones', 'edsnickname')
2019-02-16 21:45:23,926 INFO sqlalchemy.engine.base.Engine INSERT INTO users (name, fullname, nickname) VALUES (?, ?, ?)
2019-02-16 21:45:23,926 INFO sqlalchemy.engine.base.Engine ('wendy', 'Wendy Williams', 'windy')
2019-02-16 21:45:23,926 INFO sqlalchemy.engine.base.Engine INSERT INTO users (name, fullname, nickname) VALUES (?, ?, ?)
2019-02-16 21:45:23,926 INFO sqlalchemy.engine.base.Engine ('mary', 'Mary Contrary', 'mary')
2019-02-16 21:45:23,927 INFO sqlalchemy.engine.base.Engine COMMIT
2019-02-16 21:45:23,929 INFO sqlalchemy.engine.base.Engine BEGIN (implicit)
2019-02-16 21:45:23,929 INFO sqlalchemy.engine.base.Engine SELECT users.id AS users_id, users.name AS users_name, users.fullname AS users_fullname, users.nickname AS users_nickname
FROM users
WHERE users.id < ?
2019-02-16 21:45:23,929 INFO sqlalchemy.engine.base.Engine (2,)
[<User(name='ed', fullname='Ed Jones', nickname='edsnickname')>]
2019-02-16 21:45:23,931 INFO sqlalchemy.engine.base.Engine SELECT users.id AS users_id, users.name AS users_name, users.fullname AS users_fullname, users.nickname AS users_nickname
FROM users
WHERE users.id <= ?
2019-02-16 21:45:23,931 INFO sqlalchemy.engine.base.Engine (5,)
[<User(name='ed', fullname='Ed Jones', nickname='edsnickname')>, <User(name='wendy', fullname='Wendy Williams', nickname='windy')>, <User(name='mary', fullname='Mary Contrary', nickname='mary')>]
2019-02-16 21:45:23,932 INFO sqlalchemy.engine.base.Engine UPDATE users SET name=? WHERE users.id = ?
2019-02-16 21:45:23,932 INFO sqlalchemy.engine.base.Engine ('admin', 1)
2019-02-16 21:45:23,933 INFO sqlalchemy.engine.base.Engine INSERT INTO users (name, fullname, nickname) VALUES (?, ?, ?)
2019-02-16 21:45:23,933 INFO sqlalchemy.engine.base.Engine ('fred', 'Fred Flintstone', 'freddy')
2019-02-16 21:45:23,934 INFO sqlalchemy.engine.base.Engine UPDATE users SET name=? WHERE users.id = ?
2019-02-16 21:45:23,934 INFO sqlalchemy.engine.base.Engine ('test', 2)
2019-02-16 21:45:23,935 INFO sqlalchemy.engine.base.Engine SELECT users.id AS users_id, users.name AS users_name, users.fullname AS users_fullname, users.nickname AS users_nickname
FROM users
WHERE users.id <= ?
2019-02-16 21:45:23,935 INFO sqlalchemy.engine.base.Engine (5,)
[<User(name='admin', fullname='Ed Jones', nickname='edsnickname')>, <User(name='test', fullname='Wendy Williams', nickname='windy')>, <User(name='mary', fullname='Mary Contrary', nickname='mary')>, <User(name='fred', fullname='Fred Flintstone', nickname='freddy')>]

由于我们设置 create_engine 中 echo 为 True,因此在执行结果中包含了 SQLAlchemy 打印的 SQL 语句,我们可以取消 crete_engine 中的 echo文章源自菜鸟学院-https://www.cainiaoxueyuan.com/bc/9628.html

engine = create_engine('sqlite:///./sqlalchemy.sqlite')

此时的执行结果如下:文章源自菜鸟学院-https://www.cainiaoxueyuan.com/bc/9628.html

[<User(name='ed', fullname='Ed Jones', nickname='edsnickname')>]
[<User(name='ed', fullname='Ed Jones', nickname='edsnickname')>, <User(name='wendy', fullname='Wendy Williams', nickname='windy')>, <User(name='mary', fullname='Mary Contrary', nickname='mary')>]
[<User(name='admin', fullname='Ed Jones', nickname='edsnickname')>, <User(name='test', fullname='Wendy Williams', nickname='windy')>, <User(name='mary', fullname='Mary Contrary', nickname='mary')>, <User(name='fred', fullname='Fred Flintstone', nickname='freddy')>]

当取消 echo 后,程序运行结果中原有的 SQL 语句消失。文章源自菜鸟学院-https://www.cainiaoxueyuan.com/bc/9628.html

本文只是对 SQLAlchemy 的使用进行简单的介绍,SQLAlchemy 本身还有很多特性和运用方法我们可以共同探讨。文章源自菜鸟学院-https://www.cainiaoxueyuan.com/bc/9628.html

  • 本站内容整理自互联网,仅提供信息存储空间服务,以方便学习之用。如对文章、图片、字体等版权有疑问,请在下方留言,管理员看到后,将第一时间进行处理。
  • 转载请务必保留本文链接:https://www.cainiaoxueyuan.com/bc/9628.html

Comment

匿名网友 填写信息

:?: :razz: :sad: :evil: :!: :smile: :oops: :grin: :eek: :shock: :???: :cool: :lol: :mad: :twisted: :roll: :wink: :idea: :arrow: :neutral: :cry: :mrgreen:

确定