python - 在 Django 原子事务中提交特定的保存操作

Question

我有一个用原子事务装饰的 View 。但是，我总是想在此 View 中保存特定操作。该操作是调用第三方 API，有时我需要在应用程序发出请求之前刷新 token 。但是，如果此刷新后出现故障，则不会保存新 token ，而是回滚。我怎样才能始终将保存操作提交到数据库？我研究过使用 savepoint但我不确定我是否在正确的轨道上。

views.py

@transaction.atomic
def my_view(request)
    from another_file import fx
    a = Mymodel.objects.first()
    fx(a)

    return a

另一个文件.py

   def fx(obj) 
       from django.db import transaction

       sid = transaction.savepoint()

       obj.token = 'jfkds'
       obj.save()
       transaction.savepoint_commit(sid)

       raise Exception

Answer 1

有一个类似的问题，在进程 A 中的事务提交之前，“进程 B”需要用户 token 来验证从“进程 A”传递的身份验证:

为了解决这个问题，我需要调用“Token.objects.get_or_create(user=user)”来提交它自己的事务，无论周围是否已经有一些正在进行的事务。我认为您正在寻找与您的问题相同的问题。

我使用单独的数据库连接来处理 token 。因此在设置中:

DATABASES = {
    'default': {
        'ENGINE': 'django.db.backends.postgresql_psycopg2',
        'NAME': 'mydb',
        'USER': 'mydbuser',
        'PASSWORD': DBPASSWORD}}
DATABASES['security'] = DATABASES['default']  # allow independent security transactions/connections
DATABASE_ROUTERS = ['security.db_router.SecurityRouter']

路由器确保 token 模型在“安全”数据库中处理。你可以把路由器放在你喜欢的地方，然后使用上面的路由器设置链接它。:

from rest_framework.authtoken.models import Token


class SecurityRouter(object):

    def db_for_read(self, model, **hints):
        return self.mine(model, **hints)

    def db_for_write(self, model, **hints):
        return self.mine(model, **hints)

    def allow_relation(self, obj1, obj2, **hints):
        return True

    def allow_migrate(self, db, app_label, model=None, **hints):
        return True

    @staticmethod
    def mine(model, **hints):
        if model == Token:
            return 'security'

token 对象的所有使用现在都在“安全”数据库中进行。因此，用户 token 是在独立于您的“默认”数据库事务并且是即时的事务中创建的，就像所有 django 连接一样(除非您另外指定)。