bornhack-website/utils/models.py

44 lines
1.1 KiB
Python

import uuid
from django.core.exceptions import ValidationError
from django.db import models
class CleanedModel(models.Model):
class Meta:
abstract = True
def save(self, **kwargs):
try:
# call this models full_clean() method before saving,
# which in turn calls .clean_fields(), .clean() and .validate_unique()
self.full_clean()
except ValidationError as e:
message = "Got ValidationError while saving: %s" % e
if hasattr(self, 'request'):
messages.error(self.request, message)
print(message)
# dont save, just return
return
super(CleanedModel, self).save(**kwargs)
class UUIDModel(CleanedModel):
class Meta:
abstract = True
uuid = models.UUIDField(
primary_key=True,
default=uuid.uuid4,
editable=False,
)
class CreatedUpdatedModel(CleanedModel):
class Meta:
abstract = True
created = models.DateTimeField(auto_now_add=True)
updated = models.DateTimeField(auto_now=True)