Python中的对象关系映射SQLAlchemy使用

简介

什么是ORM?
ORM，即Object-Relational Mapping（对象关系映射），它的作用是在关系型数据库和业务实体对象之间作一个映射，这样，我们在具体的操作业务对象的时候，就不需要再去和复杂的SQL语句打交道，只需简单的操作对象的属性和方法。

SQLAlchemy：SQLAlchemy是Python编程语言下的一款开源软件。提供了SQL工具包及对象关系映射（ORM）工具，为高效和高性能的数据库访问设计，实现了完整的企业级持久模型。

SQLAlchemy与数据库关系图

基本操作

安装sqlalchemy
1.需要一台有安装mysql数据库的计算机
2.安装模块：在命令行中输入 pip3 install sqlalchemy
3.查看版本信息

1.连接数据库
在sqlalchemy中，session用于创建程序与数据库之间的会话。所有对象的载入和保存都需要通过session对象。具体代码如下：

from sqlalchemy import create_engine, Column, Integer, SmallInteger, String
from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy.orm import sessionmaker
# python2: MySQLdb
# python3: pymysql
# 使用create_engine建立同数据库的连接，返回的是一个Engine实例
# 指向数据库的一些核心的接口
# echo=True， 可以在控制台看到操作涉及的SQL语言
engine = create_engine("mysql+pymysql://root:westos@172.25.254.35/pymysql",
                       encoding='utf8',
                       echo=True)
# 创建缓存对象
Session = sessionmaker(bind=engine)
session = Session()
# 声明基类
Base = declarative_base()

2.创建映射(创建表)
一个映射对应着一个Python类，用来表示一个表的结构。代码如下：

# 基于这个基类来创建我们的自定义类，一个类就是一个数据库表
class Person(Base):
    __tablename__ =  'persons'
    id = Column(Integer, primary_key=True, autoincrement=True)
    name = Column(String(20), nullable=False)
    email = Column(String(30), unique=True)

    def __repr__(self):
        return  self.name

if __name__ == '__main__':
    # 检查表的存在性，如果不存在的话会执行表的创建工作
    Base.metadata.create_all(engine)

3.添加单条数据

    # 添加单条数据
    p = Person(id=1, name="小明", email='xiaoming@westos.org')
    # 先将数据写入缓存;
    session.add(p)
    # 将缓存中的数据提交到数据库并执行
    session.commit()

在数据库中查看创建的表：

查看添加的数据：

4.添加多条数据，代码如下：

    #  添加多条数据
    p1 = Person(name='A')
    p2 = Person(name='B')
    # 先将数据写入缓存;
    session.add_all((p1, p2))
    # 将缓存中的数据提交到数据库并执行
    session.commit()

在数据库中查看表中添加的数据：

5.查找数据，代码如下：

# 查找数据
    # select * from students where ()
    results = session.query(Person).all()  # 查找表的所有数据
    print(results)

    results = session.query(Person).first()  # 查找表的第一条数据
    print(results)

    obj = session.query(Person).filter_by(name='小明').first()
    print("邮箱地址:", obj.email)

查找的输出结果如下：

6.删除数据，代码如下：

    # 删除数据，没有first/all方法， 默认返回的是要执行的sql语句;
    obj = session.query(Person).filter_by(name='B').first()
    print(obj)
    session.delete(obj)
    session.commit()

查看删除数据后的表：

7.更新数据，代码如下：

    # 更新数据: 更新姓名为A的邮箱为hello@qq.com
    obj = session.query(Person).filter_by(name='A').first()
    obj.email = 'hello.qq.com'
    session.add(obj)
    session.commit()

查看更新后的表：

下面通过一个实际案例对SQLAlchemy进行应用，我们实现一个用户登录系统数据库版本，首先创建一个表，在表中添加用户信息，代码如下：

from sqlalchemy import create_engine, Column, Integer, SmallInteger, String
from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy.orm import sessionmaker

engine = create_engine("mysql+pymysql://root:westos@172.25.254.35/pymysql",
                       encoding='utf8',
                       # echo=True
                       )
# 创建缓存对象
Session = sessionmaker(bind=engine)
session = Session()

# 声明基类
Base = declarative_base()

class User(Base):
    __tablename__ = 'users'
    id = Column(Integer, primary_key=True, autoincrement=True)
    name = Column(String(20), nullable=False, unique=True)
    password = Column(String(20), nullable=False)
    email = Column(String(30), unique=True)

    def __repr__(self):
        return self.name

if __name__ == '__main__':
    # 创建数据表
    Base.metadata.create_all(engine)
    # 创建初始化数据
    u1 = User(name='user1', password='123')
    u2 = User(name='user2', password='123')
    session.add_all([u1, u2])
    session.commit()
    print("初始化数据成功........")

在数据库中查看创建的表，以及查看表中的数据：

然后模拟登录系统，并进行测试：

cookie = {}
class UserManage(object):
    def login(self, name, password):  # name='root', password='123'
        # 如果找到(用户名和密码正确)， 则返回找到的用户对象; 如果没有找到， 返回None;
        obj = session.query(User).filter_by(name=name).filter_by(password=password).first()
        if obj:
            cookie['name'] = name
            print("用户%s登陆成功" %(name))
        else:
            print("用户%s登陆失败" %(name))

    def logout(self):
        cookie.pop('name')
        print("注销成功.....")

if __name__ == '__main__':
    um = UserManage()
    um.login('user1', '123')

输出结果如下：