【数据库】对象关系映射SQLAlchemy

1.简介

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

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

SQLAlchemy与数据库关系图

2.操作

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

import sqlalchemy
from sqlalchemy import Column, Integer, String
from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy.orm import Session, sessionmaker
import pymysql
pymysql.install_as_MySQLdb()



# object(oop) ---> mysql(SQL)
# 1). 创建数据库引擎(连接数据库的过程)
# echo=True显示翻译好的SQL语句。
engine = sqlalchemy.create_engine("mysql://root:123456@localhost/Blog",
                         encoding='utf8', echo=True)

# 2). 建立缓存
session = sessionmaker(bind=engine)

# 3). 创建数据库对象需要继承的基类
Base = declarative_base()

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

# 4). 创建类, 一个类就是一个数据库表
class Student(Base):
    # 数据库表的名称
    __tablename__ = 'student_info'
    # 数据库表的属性信息
    id = Column(Integer, primary_key=True, autoincrement=True)
    name = Column(String(20))
    sex = Column(String(10))
    
    # def __repr__(self):
    #     return  'Student<%s>' %(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()

3.用户登录案例

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

import sqlalchemy
from sqlalchemy import Column, Integer, String
from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy.orm import Session, sessionmaker
import pymysql
pymysql.install_as_MySQLdb()

#1.创建数据库引擎（连接数据库的过程）
engine = sqlalchemy.create_engine("mysql://root:123456@localhost/blog",
                                    encoding='utf8', echo=False)

#2.建立缓存
# Session = sessionmaker(bind=engine)
# session = Session()
session = sessionmaker(bind=engine)()

#3.创建数据库对象需要继承的基类
Base = declarative_base()

class Users(Base):
    __tablename__ = 'userManage'
    id = Column(Integer,primary_key=True,autoincrement=True)
    name = Column(String(20),nullable=False,unique=True)  #nullable是否可以为空
    password = Column(String(20),nullable=False)
    email = Column(String(30),unique=True)

if __name__ == '__main__':
    Base.metadata.create_all(engine)

    u1 = Users(name='user1',password='123',email='user1@qq.com')
    u2 = Users(name='user2',password='345')

    session.add_all([u1,u2])
    session.commit()
    print('数据初始化成功')

在数据库中查看：

然后模拟登陆：

from db import Users,session
cookie = {}

class UserManage(object):
    def login(self,name,passwd):
        obj = session.query(Users).filter_by(name=name).filter_by(password=passwd).first()
        if obj:
            cookie['name'] = name
            print('%s用户登陆成功' %(name))
        else:
            print('%s用户登陆失败' %(name))


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