import random from django.db import models, IntegrityError from django.contrib.auth.models import User from django.utils.translation import ugettext as _ from django_lcore.core import LcoreUserProfileMethods, setup_sync_hooks, VPN_AUTH_STORAGE from payments.models import BaseSubUser prng = random.SystemRandom() def random_gift_code(): charset = "123456789ABCDEFGHIJKLMNPQRSTUVWXYZ" return ''.join([prng.choice(charset) for n in range(10)]) class VPNUser(models.Model, BaseSubUser, LcoreUserProfileMethods): class Meta: verbose_name = _("VPN User") verbose_name_plural = _("VPN Users") user = models.OneToOneField(User, on_delete=models.CASCADE) notes = models.TextField(blank=True) expiration = models.DateTimeField(blank=True, null=True) last_expiry_notice = models.DateTimeField(blank=True, null=True) last_vpn_auth_fail_notice = models.DateTimeField(blank=True, null=True) notify_expiration = models.BooleanField(default=True) notify_vpn_auth_fail = models.BooleanField(default=True) trial_periods_given = models.IntegerField(default=0) last_vpn_auth = models.DateTimeField(blank=True, null=True) last_core_sync = models.DateTimeField(blank=True, null=True) referrer = models.ForeignKey(User, blank=True, null=True, on_delete=models.SET_NULL, related_name='referrals') referrer_used = models.BooleanField(default=False) campaign = models.CharField(blank=True, null=True, max_length=64) def clear_fields(self): self.expiration = None self.last_expiry_notice = None self.last_vpn_auth = None self.last_core_sync = None self.referrer = None self.referrer_used = False self.campaign = None def on_payment_confirmed(self, payment): if self.referrer and not self.referrer_used: self.referrer.vpnuser.add_paid_months(1, 'referrer', f"rewarded for {self.user.username} (payment #{payment.id})") self.referrer.vpnuser.save() self.referrer_used = True self.save() def lcore_sync(self): if VPN_AUTH_STORAGE == 'inst': return from lambdainst.tasks import push_user push_user.delay(user_id=self.user.id) def notify_payment(self, payment): return def notify_subscription(self, sub): return def notify_refund(self, payment): return def __str__(self): return self.user.username setup_sync_hooks(User, VPNUser) #from django.db.models.signals import post_save #from django.dispatch import receiver #@receiver(post_save, sender=User) #def create_vpnuser(sender, instance, created, **kwargs): # if created: # try: # VPNUser.objects.create(user=instance) # except IntegrityError: # pass