from django.conf import settings from django.db import models from django.contrib.auth.models import ( BaseUserManager, AbstractBaseUser ) from django.utils import timezone import requests import json import subprocess import os import datetime, time class MyUserManager(BaseUserManager): def create_user(self, email, password=None): """ Creates and saves a User with the given email, date of birth and password. """ if not email: raise ValueError('Users must have an email address') user = self.model( email=self.normalize_email(email), ) user.set_password(password) user.save(using=self._db) return user def create_superuser(self, email, password): """ Creates and saves a superuser with the given email, date of birth and password. """ user = self.create_user(email, password=password ) user.is_admin = True user.save(using=self._db) return user class User(AbstractBaseUser): email = models.EmailField( verbose_name='email address', max_length=191, unique=True, ) is_active = models.BooleanField(default=True) is_admin = models.BooleanField(default=False) objects = MyUserManager() USERNAME_FIELD = 'email' REQUIRED_FIELDS = [] def get_full_name(self): return self.email def get_short_name(self): return self.email def __str__(self): return self.email def has_perm(self, perm, obj=None): "Does the user have a specific permission?" # Simplest possible answer: Yes, always return True def has_module_perms(self, app_label): "Does the user have permissions to view the app `app_label`?" # Simplest possible answer: Yes, always return True @property def is_staff(self): "Is the user a member of staff?" # Simplest possible answer: All admins are staff return self.is_admin class Domain(models.Model): created = models.DateTimeField(auto_now_add=True) updated = models.DateTimeField(null=True) name = models.CharField(max_length=191, unique=True) arecord = models.CharField(max_length=255, blank=True) aaaarecord = models.CharField(max_length=1024, blank=True) dyn = models.BooleanField(default=False) owner = models.ForeignKey(settings.AUTH_USER_MODEL, related_name='domains') headers = { 'User-Agent': 'desecapi', 'X-API-Key': settings.POWERDNS_API_TOKEN, } def save(self, *args, **kwargs): if self.id is None: self.pdnsCreate() if self.arecord or self.aaaarecord: self.pdnsUpdate() else: orig = Domain.objects.get(id=self.id) if self.arecord != orig.arecord or self.aaaarecord != orig.aaaarecord: self.pdnsUpdate() self.updated = timezone.now() super(Domain, self).save(*args, **kwargs) # Call the "real" save() method. def pdnsCreate(self): payload = { "name": self.name + ".", "kind": "NATIVE", "masters": [], "nameservers": [ "ns1.desec.io.", "ns2.desec.io." ] } r = requests.post(settings.POWERDNS_API + '/zones', data=json.dumps(payload), headers=self.headers) if r.status_code < 200 or r.status_code >= 300: raise Exception(r.text) def pdnsUpdate(self): if self.arecord: a = \ { "records": [ { "type": "A", "name": self.name + ".", "disabled": False, "content": self.arecord, } ], "ttl": 60, "changetype": "REPLACE", "type": "A", "name": self.name + ".", } else: a = \ { "changetype": "DELETE", "type": "A", "name": self.name + "." } if self.aaaarecord: aaaa = \ { "records": [ { "type": "AAAA", "name": self.name + ".", "disabled": False, "content": self.aaaarecord, } ], "ttl": 60, "changetype": "REPLACE", "type": "AAAA", "name": self.name + ".", } else: aaaa = \ { "changetype": "DELETE", "type": "AAAA", "name": self.name + "." } payload = { "rrsets": [a, aaaa] } r = requests.patch(settings.POWERDNS_API + '/zones/' + self.name, data=json.dumps(payload), headers=self.headers) if r.status_code < 200 or r.status_code >= 300: raise Exception(r) class Meta: ordering = ('created',) def get_default_value_created(): return timezone.now() def get_default_value_due(): return timezone.now() + datetime.timedelta(days=7) def get_default_value_mref(): return "ONDON" + str((timezone.now() - timezone.datetime(1970,1,1,tzinfo=timezone.utc)).total_seconds()) class Donation(models.Model): created = models.DateTimeField(default=get_default_value_created) name = models.CharField(max_length=255) iban = models.CharField(max_length=34) bic = models.CharField(max_length=11) amount = models.DecimalField(max_digits=8,decimal_places=2) message = models.CharField(max_length=255, blank=True) due = models.DateTimeField(default=get_default_value_due) mref = models.CharField(max_length=32,default=get_default_value_mref) email = models.EmailField(max_length=255, blank=True) def save(self, *args, **kwargs): self.iban = self.iban[:6] + "xxx" # do NOT save account details super(Donation, self).save(*args, **kwargs) # Call the "real" save() method. class Meta: ordering = ('created',)