Profile and Subscription managers use get_or_create to avoid some boilerplate

This commit is contained in:
Pēteris Caune 2016-03-11 12:42:46 +02:00
parent 636710f4f0
commit bb193cf4f1
2 changed files with 2 additions and 12 deletions

View File

@ -16,12 +16,7 @@ from hc.lib import emails
class ProfileManager(models.Manager): class ProfileManager(models.Manager):
def for_user(self, user): def for_user(self, user):
try: profile, created = Profile.objects.get_or_create(user_id=user.id)
profile = self.get(user_id=user.id)
except Profile.DoesNotExist:
profile = Profile(user=user)
profile.save()
return profile return profile

View File

@ -5,12 +5,7 @@ from django.db import models
class SubscriptionManager(models.Manager): class SubscriptionManager(models.Manager):
def for_user(self, user): def for_user(self, user):
try: sub, created = Subscription.objects.get_or_create(user_id=user.id)
sub = self.get(user_id=user.id)
except Subscription.DoesNotExist:
sub = Subscription(user=user)
sub.save()
return sub return sub