基于当前登录用户的Django数据库路由

Question

在view课堂上，您可以self.request.user基于此调用和执行操作。就我而言，我希望能够根据当前登录的用户切换数据库。无论如何，是否可以将self.request调用注入db_for_read()和db_for_write()方法，如下所示？

class DataBaseRouter(object):

    def db_for_read(self, model, **hints):
        user = self.request.user
        if user.id == 1:
            return "master"
        return "default"

    def db_for_write(self, model, **hints):
        user = self.request.user
        if user.id == 1:
            return "master"
        return "default"

Answer 1

您可以使用 aRouterMiddlewear检查用户是否登录，然后将所有用户重定向到您选择的特定数据库，如果您正在使用查询的执行，queries这将很有帮助。view based

class RouterMiddleware (object):

    def process_view( self, request, view_func, args, kwargs ):
        # Check the user logged in
        user = self.request.user
        # Call your functions to set the database by passing the user.id



    def process_response( self, request, response ):
        # Make the database to default here if you wish to use it no longer

        return response


class DataBaseRouter(object):

    def db_for_read(self, model, user_id=None, **hints):
       if user.id == 1:
            return "master"
        return "default"

    def db_for_write(self, model, user_id=None, **hints):
        if user.id == 1:
            return "master"
        return "default"

这是我根据您的要求修改的链接。

确保将 the 添加RouterMiddleware到您的MIDDLEWARE_CLASSES.