Mark several package columns as non-NULL

These have been around for a long time now so they don't need to be
NULL-able anymore. We can also add a custom field type for our numbers
to at least get a check constraint at the Django level.

Signed-off-by: Dan McGee <dan@archlinux.org>

1 files changed, 14 insertions, 4 deletions

diff --git a/main/models.py b/main/models.py
index 926e737..57e9650 100644
--- a/main/models.py
+++ b/main/models.py
@@ -11,6 +11,17 @@ from datetime import datetime
 from itertools import groupby
 import pytz
 
+class PositiveBigIntegerField(models.BigIntegerField):
+    _south_introspects = True
+
+    def get_internal_type(self):
+        return "PositiveBigIntegerField"
+
+    def formfield(self, **kwargs):
+        defaults = {'min_value': 0}
+        defaults.update(kwargs)
+        return super(PositiveBigIntegerField, self).formfield(**defaults)
+
 def validate_pgp_key_length(value):
     if len(value) not in (8, 16, 40):
         raise ValidationError(
@@ -131,11 +142,10 @@ class Package(models.Model):
     pkgdesc = models.CharField(max_length=255, null=True)
     url = models.CharField(max_length=255, null=True)
     filename = models.CharField(max_length=255)
-    # TODO: it would be nice to have the >0 check constraint back here
-    compressed_size = models.BigIntegerField(null=True)
-    installed_size = models.BigIntegerField(null=True)
+    compressed_size = PositiveBigIntegerField()
+    installed_size = PositiveBigIntegerField()
     build_date = models.DateTimeField(null=True)
-    last_update = models.DateTimeField(null=True, blank=True)
+    last_update = models.DateTimeField()
     files_last_update = models.DateTimeField(null=True, blank=True)
     packager_str = models.CharField(max_length=255)
     packager = models.ForeignKey(User, null=True,