Pyramid SQLAlchemy, scoped_session – 原始 SQL INSERT 不会写入数据库

在本文中，我们将介绍在使用Pyramid框架和SQLAlchemy库时，使用scoped_session进行原始SQL INSERT操作时可能遇到的问题。

阅读更多：Pyramid 教程

SQLAlchemy简介

SQLAlchemy是Python中最受欢迎的ORM(Object Relational Mapping)库之一。它提供了高度灵活的数据库访问接口，可以与多种关系数据库进行交互。Pyramid是一个轻量级的Python Web框架，它提供了强大灵活的工具和库，方便开发者构建易于维护和可扩展的Web应用。将Pyramid和SQLAlchemy结合使用，可以在开发Web应用时方便地进行数据库操作。

scoped_session和原始SQL INSERT操作

在使用SQLAlchemy时，我们通常使用session对象来执行CRUD操作。在Pyramid中，可以使用scoped_session来创建一个每个请求使用的会话对象，这样可以确保每个请求都有独立的数据库会话，避免线程安全问题。

然而，在使用scoped_session执行原始SQL INSERT操作时，可能会遇到一个问题。当我们在请求处理函数中使用scoped_session执行INSERT语句后，提交事务并没有将数据写入数据库中。这是由于scoped_session在写入数据后并没有自动关闭数据库连接，而是等到请求结束时自动回滚事务。

例如，假设我们有一个用户注册的功能，请求处理函数如下所示：

from pyramid.view import view_config
from sqlalchemy.orm import scoped_session

@view_config(route_name='register', renderer='json')
def register(request):
    session = scoped_session(request.dbsession_factory)
    email = request.json_body['email']
    password = request.json_body['password']
    # 使用原始SQL INSERT插入数据
    session.execute("INSERT INTO users (email, password) VALUES (:email, :password)",
                    {'email': email, 'password': password})
    session.commit()
    # 其他操作...
    return {'message': 'Register success'}

上述代码中，我们使用scoped_session创建了一个会话对象，并执行原始SQL INSERT插入数据。然而，即使我们使用session.commit()提交了事务，数据仍然不会写入数据库。

解决方法

要解决上述问题，我们需要手动关闭数据库连接，以确保数据能够写入数据库。可以使用session.close()方法来关闭会话对象，从而关闭数据库连接。

修改上述代码如下：

from pyramid.view import view_config
from sqlalchemy.orm import scoped_session

@view_config(route_name='register', renderer='json')
def register(request):
    session = scoped_session(request.dbsession_factory)
    email = request.json_body['email']
    password = request.json_body['password']
    try:
        # 使用原始SQL INSERT插入数据
        session.execute("INSERT INTO users (email, password) VALUES (:email, :password)",
                        {'email': email, 'password': password})
        session.commit()
        # 其他操作...
        return {'message': 'Register success'}
    finally:
        session.close()

通过添加try-finally代码块，我们确保无论是否出现异常，都会调用session.close()来关闭数据库连接。这样就能够保证使用scoped_session执行原始SQL INSERT后，数据能够正确写入数据库。

总结

在Pyramid和SQLAlchemy结合使用时，使用scoped_session进行原始SQL INSERT操作时需要注意。由于scoped_session不会自动关闭数据库连接，在执行完INSERT语句后需要手动调用session.close()来关闭会话对象，以确保数据写入数据库。在实际开发中，建议结合使用ORM操作和原始SQL操作，以便充分利用SQLAlchemy提供的便利性和灵活性。