Selaa lähdekoodia

fix(api): cleanup desecapi.models namespace

Peter Thomassen 4 vuotta sitten
vanhempi
commit
71545dee14
2 muutettua tiedostoa jossa 21 lisäystä ja 22 poistoa
  1. 4 4
      api/desecapi/migrations/0001_initial_squashed_again.py
  2. 17 18
      api/desecapi/models.py

+ 4 - 4
api/desecapi/migrations/0001_initial_squashed_again.py

@@ -41,7 +41,7 @@ class Migration(migrations.Migration):
                 ('name', models.CharField(max_length=191, unique=True, validators=[desecapi.models.validate_lower, django.core.validators.RegexValidator(code='invalid_domain_name', flags=re.RegexFlag['IGNORECASE'], message='Domain names must be labels separated by dots. Labels may consist of up to 63 letters, digits, hyphens, and underscores. The last label may not contain an underscore.', regex='^(([a-z0-9_-]{1,63})\\.)*[a-z0-9-]{1,63}$')])),
                 ('owner', models.ForeignKey(on_delete=django.db.models.deletion.PROTECT, related_name='domains', to=settings.AUTH_USER_MODEL)),
                 ('published', models.DateTimeField(blank=True, null=True)),
-                ('minimum_ttl', models.PositiveIntegerField(default=desecapi.models.get_minimum_ttl_default)),
+                ('minimum_ttl', models.PositiveIntegerField(default=desecapi.models.Domain._minimum_ttl_default)),
                 ('renewal_changed', models.DateTimeField(auto_now_add=True)),
                 ('renewal_state', models.IntegerField(choices=[(1, 'Fresh'), (2, 'Notified'), (3, 'Warned')], default=1)),
             ],
@@ -192,14 +192,14 @@ class Migration(migrations.Migration):
             name='Donation',
             fields=[
                 ('id', models.AutoField(auto_created=True, primary_key=True, serialize=False, verbose_name='ID')),
-                ('created', models.DateTimeField(default=desecapi.models.get_default_value_created)),
+                ('created', models.DateTimeField(default=desecapi.models.Donation._created_default)),
                 ('name', models.CharField(max_length=255)),
                 ('iban', models.CharField(max_length=34)),
                 ('bic', models.CharField(max_length=11)),
                 ('amount', models.DecimalField(decimal_places=2, max_digits=8)),
                 ('message', models.CharField(blank=True, max_length=255)),
-                ('due', models.DateTimeField(default=desecapi.models.get_default_value_due)),
-                ('mref', models.CharField(default=desecapi.models.get_default_value_mref, max_length=32)),
+                ('due', models.DateTimeField(default=desecapi.models.Donation._due_default)),
+                ('mref', models.CharField(default=desecapi.models.Donation._mref_default, max_length=32)),
                 ('email', models.EmailField(blank=True, max_length=255)),
             ],
             options={

+ 17 - 18
api/desecapi/models.py

@@ -200,11 +200,10 @@ validate_domain_name = [
 ]
 
 
-def get_minimum_ttl_default():
-    return settings.MINIMUM_TTL_DEFAULT
-
-
 class Domain(ExportModelOperationsMixin('Domain'), models.Model):
+    @staticmethod
+    def _minimum_ttl_default():
+        return settings.MINIMUM_TTL_DEFAULT
 
     class RenewalState(models.IntegerChoices):
         IMMORTAL = 0
@@ -218,7 +217,7 @@ class Domain(ExportModelOperationsMixin('Domain'), models.Model):
                             validators=validate_domain_name)
     owner = models.ForeignKey(User, on_delete=models.PROTECT, related_name='domains')
     published = models.DateTimeField(null=True, blank=True)
-    minimum_ttl = models.PositiveIntegerField(default=get_minimum_ttl_default)
+    minimum_ttl = models.PositiveIntegerField(default=_minimum_ttl_default.__func__)
     renewal_state = models.IntegerField(choices=RenewalState.choices, default=RenewalState.IMMORTAL)
     renewal_changed = models.DateTimeField(auto_now_add=True)
     _keys = None
@@ -404,27 +403,27 @@ class Token(ExportModelOperationsMixin('Token'), rest_framework.authtoken.models
         return make_password(plain, salt='static', hasher='pbkdf2_sha256_iter1')
 
 
-def get_default_value_created():
-    return timezone.now()
-
-
-def get_default_value_due():
-    return timezone.now() + timedelta(days=7)
-
+class Donation(ExportModelOperationsMixin('Donation'), models.Model):
+    @staticmethod
+    def _created_default():
+        return timezone.now()
 
-def get_default_value_mref():
-    return "ONDON" + str(time.time())
+    @staticmethod
+    def _due_default():
+        return timezone.now() + timedelta(days=7)
 
+    @staticmethod
+    def _mref_default():
+        return "ONDON" + str(time.time())
 
-class Donation(ExportModelOperationsMixin('Donation'), models.Model):
-    created = models.DateTimeField(default=get_default_value_created)
+    created = models.DateTimeField(default=_created_default.__func__)
     name = models.CharField(max_length=255)
     iban = models.CharField(max_length=34)
     bic = models.CharField(max_length=11, blank=True)
     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)
+    due = models.DateTimeField(default=_due_default.__func__)
+    mref = models.CharField(max_length=32, default=_mref_default.__func__)
     email = models.EmailField(max_length=255, blank=True)
 
     class Meta: