Django Benutzermodell, Funktion hinzufügen

Question

Ich möchte dem Standardbenutzermodell von Django eine neue Funktion hinzufügen, um eine verwandte Liste von Modelltypen abzurufen.

Solche Foo Modell:

class Foo(models.Model): 
    owner = models.ForeignKey(User, related_name="owner") 
    likes = models.ForeignKey(User, related_name="likes") 

........

#at some view 
    user = request.user 
    foos= user.get_related_foo_models() 

Wie dies erreicht werden kann?

Answer 1

Sie eine Methode zum

from django.contrib import auth 
auth.models.User.add_to_class('get_related_foo_models', get_related_foo_models) 

Stellen Sie sicher,

User hinzufügen können, müssen Sie den Code innerhalb des models.py oder eine andere Datei, die in der Start von django importiert wird.

Answer 2

Dies ist ein Update der Antwort von @Lakshman Prasad. Aber ein vollständiges Beispiel:

eine Datei monkey_patching.py in einem Ihrer apps erstellen ::

#app/monkey_patching.py 
from django.contrib.auth.models import User 

def get_user_name(self): 
    if self.first_name or self.last_name: 
     return self.first_name + " " + self.last_name 
    return self.username 

User.add_to_class("get_user_name",get_user_name) 

und importieren Sie es in der App __init__.py Datei. dh ::

#app/__init__.py 
import monkey_patching 

Answer 3

Es ist nicht ungewöhnlich Benutzermodell zu ersetzen, wie es in der Dokumentation angegeben: https://docs.djangoproject.com/es/1.9/topics/auth/customizing/#substituting-a-custom-user-model, so, mit diesem Umstand Rechnung, ist es besser, die Benutzermodell-Klasse mit dem folgenden Code zu erhalten:

Anschließend können Sie diese UserModel verwenden, um Funktionen wie @Lakshman Prasad schlägt vor: UserModel.add_to_class('get_related_foo_models', get_related_foo_models).

---

Um nur der Code ausgeführt werden, wenn ich es vorziehen Klassen Django-Anwendung Konfiguration verwendet werden (https://docs.djangoproject.com/es/1.9/ref/applications/), so dass ein voll funktionierendes Beispiel sein:

# myapp/__init__.py 
default_app_config = 'myapp.apps.MyAppConfig' 

# myapp/apps.py 
from django.apps import AppConfig 
from django.contrib.auth import get_user_model 


class MyAppConfig(AppConfig): 
    name = 'myapp' 
    verbose_name = 'MyApp' 

    def ready(self): 
     # Add some functions to user model: 
     def custom_function(self): 
      # Do whatsoever 
      pass 

     UserModel = get_user_model() 
     UserModel.add_to_class('custom_function', custom_function)