跳到主要内容

数据库应用编程

信息
2024年8月10日 · ·

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表:
# <sqlite3.Cursor object at 0x10f8aa260>
cursor.execute('create table user (id varchar(20) primary key, name varchar(20))')

# 继续执行一条SQL语句,插入一条记录:
# <sqlite3.Cursor object at 0x10f8aa260>
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()
# 执行查询语句:
# <sqlite3.Cursor object at 0x10f8aa340>
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类似;

MySQLSQL占位符是%sSQLite?

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'))

PS:欢迎各路道友阅读评论,感谢道友点赞关注收藏