django-avatar/avatar/__init__.py
2008-08-07 06:06:54 +00:00

14 lines
519 B
Python

from django.contrib.auth.models import User
from models import Avatar
from django.db.models import signals
def create_avatar(sender=None, instance=None, **kwargs):
avatar, created = Avatar.objects.get_or_create(user=instance)
avatar.save()
def delete_avatar(sender=None, instance=None, **kwargs):
try:
Avatar.objects.get(user=instance).delete()
except Avatar.DoesNotExist:
pass
signals.post_save.connect(create_avatar, sender=User)
signals.post_save.connect(delete_avatar, sender=User)