Skip to content

Commit

Permalink
fixes #162 with a temporary patch : a deep code rethink about signals…
Browse files Browse the repository at this point in the history
… is required
  • Loading branch information
DylannCordel committed Mar 18, 2015
1 parent 9490e4c commit ded6891
Show file tree
Hide file tree
Showing 4 changed files with 54 additions and 6 deletions.
13 changes: 10 additions & 3 deletions pybb/signals.py
Expand Up @@ -21,9 +21,16 @@ def post_saved(instance, **kwargs):


def post_deleted(instance, **kwargs):
profile = util.get_pybb_profile(instance.user)
profile.post_count = instance.user.posts.count()
profile.save()
Profile = util.get_pybb_profile_model()
User = compat.get_user_model()
try:
profile = util.get_pybb_profile(instance.user)
except (Profile.DoesNotExist, User.DoesNotExist) as e:
#When we cascade delete an user, profile and posts are also deleted
pass
else:
profile.post_count = instance.user.posts.count()
profile.save()


def user_saved(instance, created, **kwargs):
Expand Down
20 changes: 19 additions & 1 deletion pybb/tests.py
Expand Up @@ -31,7 +31,10 @@
raise Exception('PyBB requires lxml for self testing')

from pybb import defaults
from pybb.models import Topic, TopicReadTracker, Forum, ForumReadTracker, Post, Category, PollAnswer, Profile
from pybb.models import Topic, TopicReadTracker, Forum, ForumReadTracker, Post, Category, PollAnswer


Profile = util.get_pybb_profile_model()

__author__ = 'zeus'

Expand Down Expand Up @@ -1854,3 +1857,18 @@ def test_profile_autocreation_middleware(self):
self.assertIsNotNone(profile)
self.assertEqual(type(profile), util.get_pybb_profile_model())
user.delete()

def test_user_delete_cascade(self):
user = User.objects.create_user('cronos', 'cronos@localhost', 'cronos')
profile = getattr(user, defaults.PYBB_PROFILE_RELATED_NAME, None)
self.assertIsNotNone(profile)
post = Post(topic=self.topic, user=user, body='I \'ll be back')
post.save()
user_pk = user.pk
profile_pk = profile.pk
post_pk = post.pk

user.delete()
self.assertFalse(User.objects.filter(pk=user_pk).exists())
self.assertFalse(Profile.objects.filter(pk=profile_pk).exists())
self.assertFalse(Post.objects.filter(pk=post_pk).exists())
1 change: 1 addition & 0 deletions runtests.py
Expand Up @@ -82,6 +82,7 @@
STATIC_URL='/static/',
TEMPLATE_DIRS=(os.path.join(os.path.dirname(os.path.abspath(__file__)), 'test/test_project/templates'), ),
PYBB_ATTACHMENT_ENABLE=True,
PYBB_PROFILE_RELATED_NAME='pybb_customprofile',
MIDDLEWARE_CLASSES=MIDDLEWARE_CLASSES,
TEMPLATE_CONTEXT_PROCESSORS=TEMPLATE_CONTEXT_PROCESSORS,
AUTH_USER_MODEL='test_project.CustomUser',
Expand Down
26 changes: 24 additions & 2 deletions test/test_project/models.py
@@ -1,8 +1,13 @@
from __future__ import unicode_literals

import django
from django.core.urlresolvers import reverse
from django.db import models

from pybb.compat import get_user_model_path, get_username_field
from pybb.profiles import PybbProfile

if django.VERSION[:2] >= (1, 5):
from django.db import models
from django.contrib.auth.models import AbstractBaseUser, PermissionsMixin, BaseUserManager
from django.utils import timezone

Expand Down Expand Up @@ -30,4 +35,21 @@ class CustomUser(AbstractBaseUser, PermissionsMixin):
objects = CustomUserManager()

class Meta:
abstract = False
abstract = False


class CustomProfile(PybbProfile):
user = models.OneToOneField(get_user_model_path(),
verbose_name='linked account',
related_name='pybb_customprofile',
blank=False, null=False,)

class Meta(object):
verbose_name = 'Profile'
verbose_name_plural = 'Profiles'

def get_absolute_url(self):
return reverse('pybb:user', kwargs={'username': getattr(self.user, get_username_field())})

def get_display_name(self):
return self.user.get_username()

0 comments on commit ded6891

Please sign in to comment.