python - Django：モデル保存で現在のユーザーを取得

Question

models.pyrequest.userのメソッドで現在ログインしているユーザー（）を取得したい。saveユーザーの役割を確認し、ユーザーの役割に基づいていくつかの操作を実行できるかどうかを確認したいと思います。

models.py：

class TimeSheet(models.Model):
    check_in_time = models.TimeField()
    check_out_time = models.TimeField()

class Tasks(models.Model):
    time_sheet = models.ForeignKey(TimeSheet)
    project = models.ForeignKey(Project)
    start_time = models.TimeField()
    end_time = models.TimeField()

    def save(self, *args,**kwargs):
        project = SpentTime.objects.get(project__project__id = self.project.id)
        start = datetime.datetime.strptime(str(self.start_time), '%H:%M:%S')
        end = datetime.datetime.strptime(str(self.end_time), '%H:%M:%S')
        time = float("{0:.2f}".format((end - start).seconds/3600.0))

        if common.isDesigner(request.user):
            SpentTime.objects.filter(project__project__id = self.project.id).update(design = float(project.design) + time)

        if common.isDeveloper(request.user):
            SpentTime.objects.filter(project__project__id = self.project.id).update(develop = float(project.develop) + time)

        super(Tasks, self).save(*args, **kwargs)

ここでは、Tasksモデルはモデルのインラインとして使用されていTimesheetます。現在ログインしているユーザーの役割を確認し、ユーザーの役割に基づいて別のモデルを更新したいと思います。ここrequest.userで、現在のユーザーの役割を確認する必要があります。私はフォームやテンプレートを使用しておらず、Djangoadminを完全に利用しています。メソッドを取得したり、admin.pyの別のモデルの値を確認および更新したりする方法はありrequest.userますsaveか？

score 41 · Accepted Answer

この問題には別の角度から取り組むことができます。モデルのsaveメソッドを変更する代わりに、AdminSitessave_modelメソッドをオーバーライドする必要があります。そこにリクエストオブジェクトがあり、すでに指摘したようにログインしたユーザーデータにアクセスできます。

ドキュメントのこの章をご覧ください：DjangoModelAdminドキュメントsave_model

score 26 · Accepted Answer

それを行う方法を見つけましたが、ミドルウェアを宣言する必要があります。get_username.py次のコンテンツを使用して、アプリ内で呼び出されるファイルを作成します。

from threading import current_thread

_requests = {}

def get_username():
    t = current_thread()
    if t not in _requests:
         return None
    return _requests[t]

class RequestMiddleware(object):
    def process_request(self, request):
        _requests[current_thread()] = request

を編集して、：settings.pyに追加します。MIDDLEWARE_CLASSES

MIDDLEWARE_CLASSES = (
    ...
    'yourapp.get_username.RequestMiddleware',
)

これで、save()メソッドで、次のように現在のユーザー名を取得できます。

from get_username import get_username

...

def save(self, *args, **kwargs):
    req = get_username()
    print "Your username is: %s" % (req.user)

score 11 · Accepted Answer

@nKnによって提案されたソリューションは良い出発点ですが、今日実装しようとしたときに、2つの問題に直面しました。

現在のDjangoバージョンでは、プレーンオブジェクトとして作成されたミドルウェアは機能しません。
単体テストは失敗しています（通常はシングルスレッドで実行されるため、最初のテストにHTTPリクエストがあり、2番目のテストにHTTPリクエストがない場合、2つの結果のテストの間に「リクエスト」がスタックする可能性があります）。

これが私の更新されたミドルウェアコードです。これはDjango1.10で動作し、単体テストを壊しません。

from threading import current_thread

from django.utils.deprecation import MiddlewareMixin


_requests = {}


def current_request():
    return _requests.get(current_thread().ident, None)


class RequestMiddleware(MiddlewareMixin):

    def process_request(self, request):
        _requests[current_thread().ident] = request

    def process_response(self, request, response):
        # when response is ready, request should be flushed
        _requests.pop(current_thread().ident, None)
        return response

    def process_exception(self, request, exception):
        # if an exception has happened, request should be flushed too
         _requests.pop(current_thread().ident, None)

score 8 · Accepted Answer

正しい方法はthreading.local、で辞書を使用する代わりに、を使用するthreading.current_threadことです。これは、メモリリークが発生するためです。これは、アプリケーションが実行されている間、古い値がそこにとどまるためです。

import threading

request_local = threading.local()

def get_request():
    return getattr(request_local, 'request', None)

class RequestMiddleware():
    def __init__(self, get_response):
        self.get_response = get_response

    def __call__(self, request):
        request_local.request = request
        return self.get_response(request)

    def process_exception(self, request, exception):
        request_local.request = None

    def process_template_response(self, request, response):
        request_local.request = None
        return response

リクエストの代わりにユーザーにアクセスしたい場合は、リクエストのget_request().user代わりにユーザーを保存するか、__call__

score 7 · Accepted Answer

save_modelメソッドのオーバーライドが最良のオプションだとは思いません。たとえば、ユーザー情報を保存したり、ユーザー情報に基づいてモデルを検証したりする場合で、save（）がビューまたは管理サイト自体からのものではないとします。

人々が求めているのは、次のような構造です。

def save(..)
    self.user = current_user()

また

def save(..)
    user = current_user()
    if user.group == 'XPTO':
        error('This user cannot edit this record')

私がこれまでに見つけた最善のアプローチは次のとおりです。

https://bitbucket.org/q/django-current-user/overview

score 4 · Accepted Answer

次のライブラリを使用してみましたか。

https://pypi.org/project/django-currentuser/

ウェブサイトからの使用方法の抜粋：

from django_currentuser.middleware import (get_current_user, get_current_authenticated_user)
# As model field:
from django_currentuser.db.models import CurrentUserField
class Foo(models.Model):
    created_by = CurrentUserField()

python - Django：モデル保存で現在のユーザーを取得

6 に答える 6

Related

Reference