diff --git a/accounts/models.py b/accounts/models.py index b9cd09b..1c242e6 100644 --- a/accounts/models.py +++ b/accounts/models.py @@ -115,16 +115,35 @@ class UserProfile(models.Model): @property def is_a_team(self): - return True if get_object_or_None(Team, user=self.user) else False + if get_object_or_None(Team, user=self.user, disabled=False): + return True + else: + return False def teams(self): - teams_owned = Team.objects.filter(owner=self.user) - teams_in = Team.objects.filter(members=self.user) + teams_owned = Team.objects.filter(owner=self.user, disabled=False) + teams_in = Team.objects.filter(members=self.user, disabled=False) return list(chain(teams_owned, teams_in)) + @property + def has_teams(self): + if (len(self.teams()) > 0): + return True + else: + return False + def get_account_age(self): delta = datetime.now().replace(tzinfo=None) - \ self.user.date_joined.replace(tzinfo=None) return delta.days + @property + def has_pro(self): + if (self.is_pro or + self.has_teams or + self.is_a_team): + return True + else: + return False + User.profile = property(lambda u: UserProfile.objects.get_or_create(user=u)[0]) diff --git a/accounts/templates/account.html b/accounts/templates/account.html index 29a9377..5dbab94 100644 --- a/accounts/templates/account.html +++ b/accounts/templates/account.html @@ -26,7 +26,7 @@
  • Editor
    • -
  • +
  • Billing
    • diff --git a/blogs/middleware.py b/blogs/middleware.py index 644032c..f630740 100644 --- a/blogs/middleware.py +++ b/blogs/middleware.py @@ -34,6 +34,7 @@ class BlogMiddleware: get_object_or_404(User, username__iexact=blog_user) if request.blog_user is None: + # TODO: This needs to check profile.has_pro() instead. pro_users = User.objects.filter(userprofile__is_pro=True) for pro_user in pro_users: diff --git a/blogs/templates/blogs/themes/default/base.html b/blogs/templates/blogs/themes/default/base.html index 4ca2dea..8e348af 100644 --- a/blogs/templates/blogs/themes/default/base.html +++ b/blogs/templates/blogs/themes/default/base.html @@ -77,7 +77,7 @@ {% endif %} - {% if not blog_user.profile.is_pro %} + {% if not blog_user.profile.has_pro %}