import base64 import os import uuid from datetime import timedelta from django.conf import settings from django.contrib.auth.hashers import check_password, make_password from django.contrib.auth.models import User from django.core import signing from django.db import models from django.urls import reverse from django.utils import timezone from hc.lib import emails def month(dt): """ For a given datetime, return the matching first-day-of-month date. """ return dt.date().replace(day=1) class ProfileManager(models.Manager): def for_user(self, user): try: return user.profile except Profile.DoesNotExist: profile = Profile(user=user, team_access_allowed=user.is_superuser) if not settings.USE_PAYMENTS: # If not using payments, set high limits profile.check_limit = 500 profile.sms_limit = 500 profile.save() return profile class Profile(models.Model): # Owner: user = models.OneToOneField(User, models.CASCADE, blank=True, null=True) team_name = models.CharField(max_length=200, blank=True) team_access_allowed = models.BooleanField(default=False) next_report_date = models.DateTimeField(null=True, blank=True) reports_allowed = models.BooleanField(default=True) ping_log_limit = models.IntegerField(default=100) check_limit = models.IntegerField(default=20) token = models.CharField(max_length=128, blank=True) api_key = models.CharField(max_length=128, blank=True) current_team = models.ForeignKey("self", models.SET_NULL, null=True) bill_to = models.TextField(blank=True) last_sms_date = models.DateTimeField(null=True, blank=True) sms_limit = models.IntegerField(default=0) sms_sent = models.IntegerField(default=0) objects = ProfileManager() def __str__(self): return self.team_name or self.user.email def team(self): # compare ids to avoid SQL queries if self.current_team_id and self.current_team_id != self.id: return self.current_team return self def prepare_token(self, salt): token = str(uuid.uuid4()) self.token = make_password(token, salt) self.save() return token def check_token(self, token, salt): return salt in self.token and check_password(token, self.token) def send_instant_login_link(self, inviting_profile=None): token = self.prepare_token("login") path = reverse("hc-check-token", args=[self.user.username, token]) ctx = { "button_text": "Log In", "button_url": settings.SITE_ROOT + path, "inviting_profile": inviting_profile } emails.login(self.user.email, ctx) def send_set_password_link(self): token = self.prepare_token("set-password") path = reverse("hc-set-password", args=[token]) ctx = { "button_text": "Set Password", "button_url": settings.SITE_ROOT + path } emails.set_password(self.user.email, ctx) def send_change_email_link(self): token = self.prepare_token("change-email") path = reverse("hc-change-email", args=[token]) ctx = { "button_text": "Change Email", "button_url": settings.SITE_ROOT + path } emails.change_email(self.user.email, ctx) def set_api_key(self): self.api_key = base64.urlsafe_b64encode(os.urandom(24)) self.save() def send_report(self): # reset next report date first: now = timezone.now() self.next_report_date = now + timedelta(days=30) self.save() token = signing.Signer().sign(uuid.uuid4()) path = reverse("hc-unsubscribe-reports", args=[self.user.username]) unsub_link = "%s%s?token=%s" % (settings.SITE_ROOT, path, token) ctx = { "checks": self.user.check_set.order_by("created"), "now": now, "unsub_link": unsub_link } emails.report(self.user.email, ctx) def invite(self, user): member = Member(team=self, user=user) member.save() # Switch the invited user over to the new team so they # notice the new team on next visit: user.profile.current_team = self user.profile.save() user.profile.send_instant_login_link(self) def sms_sent_this_month(self): # IF last_sms_date was never set, we have not sent any messages yet. if not self.last_sms_date: return 0 # If last sent date is not from this month, we've sent 0 this month. if month(timezone.now()) > month(self.last_sms_date): return 0 return self.sms_sent def authorize_sms(self): """ If monthly limit not exceeded, increase counter and return True """ sent_this_month = self.sms_sent_this_month() if sent_this_month >= self.sms_limit: return False self.sms_sent = sent_this_month + 1 self.last_sms_date = timezone.now() self.save() return True class Member(models.Model): team = models.ForeignKey(Profile, models.CASCADE) user = models.ForeignKey(User, models.CASCADE)