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):
|
||||
|
||||
def for_user(self, user):
|
||||
try:
|
||||
profile = self.get(user_id=user.id)
|
||||
except Profile.DoesNotExist:
|
||||
profile = Profile(user=user)
|
||||
profile.save()
|
||||
|
||||
profile, created = Profile.objects.get_or_create(user_id=user.id)
|
||||
return profile
|
||||
|
||||
|
||||
|
@ -5,12 +5,7 @@ from django.db import models
|
||||
class SubscriptionManager(models.Manager):
|
||||
|
||||
def for_user(self, user):
|
||||
try:
|
||||
sub = self.get(user_id=user.id)
|
||||
except Subscription.DoesNotExist:
|
||||
sub = Subscription(user=user)
|
||||
sub.save()
|
||||
|
||||
sub, created = Subscription.objects.get_or_create(user_id=user.id)
|
||||
return sub
|
||||
|
||||
|
||||
|
Loading…
x
Reference in New Issue
Block a user