forked from GithubBackups/healthchecks
Profile and Subscription managers use get_or_create to avoid some boilerplate
This commit is contained in:
parent
636710f4f0
commit
bb193cf4f1
@ -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
|
||||||
|
|
||||||
|
|
||||||
|
@ -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
|
||||||
|
|
||||||
|
|
||||||
|
Loading…
x
Reference in New Issue
Block a user