Django créer userprofile si n'existe pas

Question

J'utilise request.user.get_profile() dans un certain nombre d'endroits dans mon code. Cependant, tous les utilisateurs n'ont pas un profil utilisateur assosié à leur compte. Est-il possible de créer un profil utilisateur automatiquement lorsque j'appelle le request.user.get_profile() s'il n'en a pas un sans avoir à changer chacun de mes request.user.get_profile() appels. Peut-être en utilisant des signaux ou quelque chose. Le problème que j'ai est que j'utilise un identifiant de connexion unique, donc si quelqu'un se connecte sur un autre système puis va sur mon site, il est connecté mais ne crée pas de profil utilisateur (je crée un utilisateur profil quand ils se connectent).

Answer 1

Généralement, les objets de profil utilisateur sont créés juste après la création de l'instance utilisateur. Ceci est facilement accompli en utilisant des signaux. Apparemment, l'utilisation de signaux au moment de la création de l'utilisateur ne fonctionnera pas pour vous. Une façon d'accomplir ce que vous essayez de faire serait de remplacer les appels à request.user.get_profile() avec une fonction personnalisée. Dites get_or_create_user_profile(request). Cette fonction peut essayer de récupérer le profil d'un utilisateur et en créer un à la volée s'il n'en existe pas.

Pour exemple:

def get_or_create_user_profile(request): 
    profile = None 
    user = request.user 
    try: 
     profile = user.get_profile() 
    except UserProfile.DoesNotExist: 
     profile = UserProfile.objects.create(user, ...) 
    return profile 

Answer 2

J'ai trouvé cette solution que je préfère utiliser sur ce site: http://www.turnkeylinux.org/blog/django-profile

User.profile = property(lambda u: UserProfile.objects.get_or_create(user=u)[0]) 

En utilisant une propriété comme cela semble plus propre que la définition d'une nouvelle méthode, mais je suis d'accord ce serait difficile de passer en revue votre base de code et de changer user.get_profile() références à user.profile.

Là encore, n'est-ce pas ce que votre fonction globale de recherche et de remplacement de fichiers est destinée?

Answer 3

Miser sur la réponse de bskinner et en réponse au commentaire de sqwerty, vous pouvez utiliser ceci:

def get_or_create_profile(user): 
    """ 
    Return the UserProfile for the given user, creating one if it does not exist. 

    This will also set user.profile to cache the result. 
    """ 
    user.profile, c = UserProfile.objects.get_or_create(user=user) 
    return user.profile 

User.profile = property(get_or_create_profile) 

Donc cela va créer un profil si nécessaire, et puis User.profile.field = "deep"User.profile.save() œuvres.

Answer 4

Révisée comme ci-dessous. Sinon, "AttributeError: impossible de définir l'attribut" sera affiché. La permission d'assignation du gardien est facultative ici.

 
def get_or_create_profile(user): 
    profile, created = UserProfile.objects.get_or_create(user=user) 
    assign('change_userprofile', user, profile) 
    assign('delete_userprofile', user, profile) 
    return profile 

User.profile = property(get_or_create_profile) 

Answer 5

Voici une commande de gestion rapide qui peut être utilisée pour créer des profils par défaut pour les utilisateurs qui en sont dépourvus. Avec cela en place, vous pouvez exécuter ./manage.py missing_profiles à tout moment, ou l'ajouter à une tâche cron, etc.

from django.core.management.base import BaseCommand 
from django.contrib.auth.models import User 
from people.models import Profile 

''' 
Create default profiles for any User objects that lack them. 
''' 


class Command(BaseCommand): 
    help = "Create default profiles for any User objects that lack them." 

    def handle(self, *args, **options): 
     users = User.objects.filter(profile=None) 
     for u in users: 
      Profile.objects.create(user=u) 
      print("Created profile for {u}".format(u=u))