将 SqlAlchemy 集成到 Django 项目中的最佳方法

Question

我将我的 Django 应用程序更改为使用 SQLAlchemy，它现在可以工作了。

但我想知道我应该把这些行放在哪里：

engine = sqlalchemy.create_engine(settings.DATABASE_URL)
Session = sqlalchemy.orm.sessionmaker(bind=engine)
session = Session()

我问的原因是因为我想在很多地方使用 SQLAlchemy，我认为每次我需要使用数据库时调用这三行代码并不正确/强大/编写良好。

我需要 SA 的地方是：

• 在我看来，当然
• 在我写的一些中间件中
• 在我的模型中。就像在 get_all_tags 中的 BlogPost 模型一样。

我认为正确的是获取会话，如果会话关闭则重新连接到数据库，或者如果存在则返回当前连接的会话。

如何在我的 Django 应用中正确使用 SQLAlchemy？

感谢您的帮助！

注意：我已经按照本教程将 SA 实现到我的 Django 应用程序中，但是这篇并没有告诉我将这 3 行放在哪里 (http://lethain.com/entry/2008/jul/23/replacing-django-s-orm-with-sqlalchemy/)。

Answer 1

前两个engine和Session可以放在settings.py；毕竟，它们是配置。

实际上创建会话需要稍微小心一点，因为session 本质上是一个“事务”。最简单的做法是在需要时在每个视图函数中创建它，并在返回之前提交它们。如果你想要更多的魔法，或者如果你想/需要在视图函数之外使用会话，你应该定义一些middleware，比如

class MySQLAlchemySessionMiddleware(object):
    def process_request(self, request):
        request.db_session = settings.Session()

    def process_response(self, request, response):
        try:
            session = request.db_session
        except AttributeError:
            return response
        try:
            session.commit()
            return response
        except:
            session.rollback()
            raise

    def process_exception(self, request, exception):
        try:
            session = request.db_session
        except AttributeError:
            return
        session.rollback()

然后，每个视图的请求中都会有一个db_session 属性，他们可以根据需要使用该属性，并且在响应完成后将提交任何添加的内容。

不要忘记将中间件添加到MIDDLEWARE_CLASSES