PonyConf/accounts/signals.py

29 lines
1.0 KiB
Python

from django.contrib import messages
from django.contrib.auth.models import User
from django.contrib.auth.signals import user_logged_in, user_logged_out
from django.contrib.sites.shortcuts import get_current_site
from django.db.models.signals import post_save
from django.dispatch import receiver
from .models import Participation, Profile
@receiver(user_logged_in)
def on_user_logged_in(sender, request, user, **kwargs):
proposition, created = Participation.on_site.get_or_create(user=user, site=get_current_site(request))
if created:
messages.info(request, "Please check your profile!\n", fail_silently=True) # FIXME
messages.success(request, 'Welcome!', fail_silently=True) # FIXME
@receiver(user_logged_out)
def on_user_logged_out(sender, request, **kwargs):
messages.success(request, 'Goodbye!', fail_silently=True) # FIXME
def create_profile(sender, instance, created, **kwargs):
if created:
Profile.objects.create(user=instance)
post_save.connect(create_profile, sender=User, weak=False, dispatch_uid='create_profile')