「Python 基础」数据库应用编程

Python 定义了一套 DB-API，任何数据库要连接到 Python，只需要提供符合 Python 标准的数据库驱动即可；

1. 连接 SQLite

签入式数据库，本身是一个文件，C写的，体积很小，常集成在应用程序；

操作关系数据库

• 连接数据库（Connection）；
• 打开游标，通过游标执行 SQL，获取结果；

1. 建表、插入数据

# 导入SQLite驱动:
import sqlite3

# 连接到 SQLite 数据库 test.db
# 如果文件不存在，会自动在当前目录创建:
conn = sqlite3.connect('test.db')

# 创建一个 Cursor:
cursor = conn.cursor()

# 执行一条SQL语句，创建user表:
# 
cursor.execute('create table user (id varchar(20) primary key, name varchar(20))')

# 继续执行一条SQL语句，插入一条记录:
# 
cursor.execute('insert into user (id, name) values (\'1\', \'Michael\')')

# 通过 rowcount 获得插入(影响)的行数:
cursor.rowcount
# 1

# 关闭Cursor:
cursor.close()

# 提交事务:
conn.commit()

# 关闭Connection:
conn.close()

2. 查询数据

conn = sqlite3.connect('test.db')
cursor = conn.cursor()
# 执行查询语句:
# 
cursor.execute('select * from user where id=?', ('1',))
# 获得查询结果集:
values = cursor.fetchall()
# [('1', 'Michael')]
cursor.close()
conn.close()

2. 连接 MySQL

1. 安装驱动

# 官方驱动
$ pip install mysql-connector-python --allow-external mysql-connector-python

# 或者其他驱动
$ pip install mysql-connector

2. 演示连接

# 导入MySQL驱动:
import mysql.connector
# 注意把password设为你的root口令:
conn = mysql.connector.connect(user='root', password='password', database='test')
cursor = conn.cursor()
# 创建user表:
cursor.execute('create table user (id varchar(20) primary key, name varchar(20))')
# 插入一行记录，注意MySQL的占位符是%s:
# cursor.rowcount 为 1
cursor.execute('insert into user (id, name) values (%s, %s)', ['1', 'Michael'])
# 提交事务:
conn.commit()
cursor.close()

# 运行查询:
cursor = conn.cursor()
cursor.execute('select * from user where id = %s', ('1',))
# [('1', 'Michael')]
values = cursor.fetchall()
# 关闭Cursor和Connection:
# True
cursor.close()
conn.close()

Python 的DB-API定义是通用的，所有操作MySQL的数据库代码和SQLite类似；

MySQL的SQL占位符是%s，SQLite是?；

3. SQLAlchemy

Python 中最有名的ORM框架；

ORM Object-Relational Mapping，把关系数据库的表结构映射到对象上；

1. 安装

$ pip install sqlalchemy

2. DBSession

# 导入:
from sqlalchemy import Column, String, create_engine
from sqlalchemy.orm import sessionmaker
from sqlalchemy.ext.declarative import declarative_base

# 创建对象的基类:
Base = declarative_base()

# 定义User对象:
class User(Base):
    # 表的名字:
    __tablename__ = 'user'

    # 表的结构:
    id = Column(String(20), primary_key=True)
    name = Column(String(20))

# 初始化数据库连接:
# '数据库类型+数据库驱动名称://用户名:口令@机器地址:端口号/数据库名'
engine = create_engine('mysql+mysqlconnector://root:password@localhost:3306/test')
# 创建DBSession类型:
DBSession = sessionmaker(bind=engine)

3. add

# 创建session对象:
session = DBSession()
# 创建新User对象:
new_user = User(id='5', name='Bob')
# 添加到session:
session.add(new_user)
# 提交即保存到数据库:
session.commit()
# 关闭session:
session.close()

4. query

# 创建Session:
session = DBSession()
# 创建Query查询，filter是where条件，最后调用one()返回唯一行，如果调用all()则返回所有行:
user = session.query(User).filter(User.id=='5').one()
# 打印类型和对象的name属性:
print('type:', type(user))
print('name:', user.name)
# 关闭Session:
session.close()

5. relationship

class User(Base):
    __tablename__ = 'user'

    id = Column(String(20), primary_key=True)
    name = Column(String(20))
    # 一对多:
    books = relationship('Book')

class Book(Base):
    __tablename__ = 'book'

    id = Column(String(20), primary_key=True)
    name = Column(String(20))
    # “多”的一方的book表是通过外键关联到user表的:
    user_id = Column(String(20), ForeignKey('user.id'))

---

• 上一篇：「Python 基础」网络编程、电子邮件
• 专栏：《Python 基础》

PS：欢迎各路道友阅读与评论，感谢道友点赞、关注、收藏！