如何在Python框架中处理数据库操作？_python操作数据库的库-CSDN博客

本文链接：https://blog.csdn.net/Wyxl990/article/details/136466111

在Python中处理数据库操作是开发中的一个常见需求，无论是在Web开发、数据分析还是自动化脚本中，都可能需要与数据库进行交互。
Python提供了多种方式来实现数据库操作，包括使用标准的数据库API、ORM（对象关系映射）工具等。

下面，我将通过几个示例来探讨如何在Python框架中处理数据库操作：一个使用SQLite数据库的例子，另一个使用SQLAlchemy ORM工具，还有一个是使用Django ORM 工具。

使用SQLite数据库

SQLite是一个轻量级的数据库，它是自包含的、高可靠的、无需配置的SQL数据库引擎。Python内置了对SQLite的支持，通过sqlite3模块可以轻松地进行数据库操作。

import sqlite3

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

# 创建一个表格 user
cursor.execute('CREATE TABLE IF NOT EXISTS user (id VARCHAR(20) PRIMARY KEY, name VARCHAR(20))')

# 插入一条记录
cursor.execute('INSERT INTO user (id, name) VALUES (\'1\', \'Michael\')')

# 通过rowcount获得插入的行数:
print('row count:', cursor.rowcount)

# 查询记录:
cursor.execute('SELECT * FROM user WHERE id=?', ('1',))
values = cursor.fetchall()
print(values)

# 关闭Cursor和Connection:
cursor.close()
conn.commit()
conn.close()

在这个示例代码中，我们首先导入了sqlite3模块，然后创建了一个到SQLite数据库的连接。
CREATE TABLE语句用来创建一个新表，INSERT INTO用来插入数据，SELECT用来查询数据。

使用SQLAlchemy ORM

SQLAlchemy是Python中最流行的ORM工具之一，它提供了一种高层的抽象来处理数据库操作，使开发者可以用Python类和对象来插入、查询、更新和删除数据库记录，而不用写SQL语句。

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

# 定义一个基类
Base = declarative_base()

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

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

# 初始化数据库连接:
engine = create_engine('sqlite:///test.db')
# 创建DBSession类型:
DBSession = sessionmaker(bind=engine)

# 创建session对象:
session = DBSession()
# 创建新User对象:
new_user = User(id='2', name='Bob')
# 添加到session:
session.add(new_user)

# 提交即保存到数据库:
session.commit()
# 关闭session:
session.close()

# 创建Session:
session = DBSession()
# 查询记录:
user = session.query(User).filter(User.id=='2').one()
# 打印类型和对象的name属性:
print('type:', type(user))
print('name:', user.name)
# 关闭Session:
session.close()

在这个例子中，我们首先通过SQLAlchemy定义了一个User类，它对应于数据库中的user表。我们使用了create_engine函数来初始化数据库连接，declarative_base函数来创建了一个基类，它的子类可以自动与一个表关联。通过sessionmaker函数创建了一个DBSession类，进而创建了一个Session实例来进行数据库操作。

使用Django ORM进行数据库操作

Django是一个流行的Python Web框架，它的ORM允许开发者使用Python代码来操作数据库，而无需直接编写SQL。

from django.db import models

# 定义模型
class Blog(models.Model):
    name = models.CharField(max_length=100)
    tagline = models.TextField()

    def __str__(self):
        return self.name

# 使用Django ORM添加数据
b = Blog(name='Beatles Blog', tagline='All the latest Beatles news.')
b.save()

# 使用Django ORM查询数据
blog = Blog.objects.get(name='Beatles Blog')
print(blog.tagline)