Adding reversion.register() decorators to 72 important models

Author: szabozoltan69

api/models.py

@@ -1,3 +1,4 @@
+import reversion
 from django.utils.translation import ugettext_lazy as _
 # from django.db import models
 from django.contrib.gis.db import models
@@ -132,6 +133,7 @@ class Labels:
     # update api_country set record_type=3 where name like '%egion%';
 
 
+@reversion.register()
 class Country(models.Model):
     """ A country """
 
@@ -287,6 +289,7 @@ class Meta:
     def __str__(self):
         return f'{self.admin1} - {self.name}'
 
+
 class CountryGeoms(models.Model):
     """ Admin0 geometries """
     geom = models.MultiPolygonField(srid=4326, blank=True, null=True)
@@ -298,11 +301,13 @@ class DistrictGeoms(models.Model):
     geom = models.MultiPolygonField(srid=4326, blank=True, null=True)
     district = models.OneToOneField(District, verbose_name=_('district'), on_delete=models.DO_NOTHING, primary_key=True)
 
+
 class Admin2Geoms(models.Model):
     """ Admin2 geometries """
     geom = models.MultiPolygonField(srid=4326, blank=True, null=True)
     admin2 = models.OneToOneField(Admin2, verbose_name=_('admin2'), on_delete=models.DO_NOTHING, primary_key=True)
 
+
 class VisibilityChoices(IntEnum):
     MEMBERSHIP = 1
     IFRC = 2
@@ -333,6 +338,7 @@ class VisibilityCharChoices():
 
 # Common parent class for key figures.
 # Country/region variants inherit from this.
+@reversion.register()
 class AdminKeyFigure(models.Model):
     figure = models.CharField(verbose_name=_('figure'), max_length=100)
     deck = models.CharField(verbose_name=_('deck'), max_length=50)
@@ -390,6 +396,7 @@ class Labels:
         TAB_3 = _('Tab 3')
 
 
+@reversion.register()
 class RegionSnippet(models.Model):
     region = models.ForeignKey(Region, verbose_name=_('region'), related_name='snippets', on_delete=models.CASCADE)
     snippet = HTMLField(verbose_name=_('snippet'), null=True, blank=True)
@@ -406,6 +413,7 @@ def __str__(self):
         return self.snippet
 
 
+@reversion.register()
 class RegionEmergencySnippet(models.Model):
     region = models.ForeignKey(Region, verbose_name=_('region'), related_name='emergency_snippets', on_delete=models.CASCADE)
     title = models.CharField(max_length=255, blank=True)
@@ -422,6 +430,7 @@ def __str__(self):
         return self.snippet
 
 
+@reversion.register()
 class RegionPreparednessSnippet(models.Model):
     region = models.ForeignKey(Region, verbose_name=_('region'), related_name='preparedness_snippets', on_delete=models.CASCADE)
     title = models.CharField(max_length=255, blank=True)
@@ -438,6 +447,7 @@ def __str__(self):
         return self.snippet
 
 
+@reversion.register()
 class RegionProfileSnippet(models.Model):
     region = models.ForeignKey(Region, verbose_name=_('region'), related_name='profile_snippets', on_delete=models.CASCADE)
     title = models.CharField(max_length=255, blank=True)
@@ -463,6 +473,7 @@ def __str__(self):
 #         return self.title
 
 
+@reversion.register()
 class CountrySnippet(models.Model):
     country = models.ForeignKey(Country, verbose_name=_('country'), related_name='snippets', on_delete=models.CASCADE)
     snippet = HTMLField(verbose_name=_('snippet'), null=True, blank=True)
@@ -479,6 +490,7 @@ def __str__(self):
         return self.snippet
 
 
+@reversion.register()
 class AdminLink(models.Model):
     title = models.CharField(verbose_name=_('title'), max_length=100)
     url = models.URLField(verbose_name=_('url'), max_length=300)
@@ -505,6 +517,7 @@ class Meta:
         verbose_name_plural = _('country links')
 
 
+@reversion.register()
 class AdminContact(models.Model):
     ctype = models.CharField(verbose_name=_('type'), max_length=100, blank=True)
     name = models.CharField(verbose_name=_('name'), max_length=100)
@@ -546,6 +559,7 @@ class Labels:
         RED = _('Red')
 
 
+@reversion.register()
 class Event(models.Model):
     """ A disaster, which could cover multiple countries """
 
@@ -681,6 +695,7 @@ def __str__(self):
         return self.name
 
 
+@reversion.register()
 class EventFeaturedDocument(models.Model):
     event = models.ForeignKey(
         Event,
@@ -700,6 +715,7 @@ class EventFeaturedDocument(models.Model):
     )
 
 
+@reversion.register()
 class EventLink(models.Model):
     """
     Used in emergency overview.
@@ -716,6 +732,7 @@ class EventLink(models.Model):
     url = models.URLField(verbose_name=_('url'))
 
 
+@reversion.register()
 class EventContact(models.Model):
     """ Contact for event """
 
@@ -734,6 +751,7 @@ def __str__(self):
         return '%s: %s' % (self.name, self.title)
 
 
+@reversion.register()
 class KeyFigure(models.Model):
     event = models.ForeignKey(Event, verbose_name=_('event'), related_name='key_figures', on_delete=models.CASCADE)
     number = models.CharField(verbose_name=_('number'), max_length=100, help_text=_('key figure metric'))
@@ -750,6 +768,7 @@ def snippet_image_path(instance, filename):
     return 'emergencies/%s/%s' % (instance.event.id, filename)
 
 
+@reversion.register()
 class Snippet(models.Model):
     """ Snippet of text """
     snippet = HTMLField(verbose_name=_('snippet'), null=True, blank=True)
@@ -788,6 +807,7 @@ def sitrep_document_path(instance, filename):
     return 'sitreps/%s/%s' % (instance.event.id, filename)
 
 
+@reversion.register()
 class SituationReport(models.Model):
     created_at = models.DateTimeField(verbose_name=_('created at'), auto_now_add=True)
     name = models.CharField(verbose_name=_('name'), max_length=100)
@@ -869,6 +889,7 @@ class Labels:
         ARCHIVED = _('Archived')
 
 
+@reversion.register()
 class Appeal(models.Model):
     """ An appeal for a disaster and country, containing documents """
 
@@ -1025,6 +1046,7 @@ def __str__(self):
         return self.aid
 
 
+@reversion.register()
 class AppealDocument(models.Model):
     # Don't set `auto_now_add` so we can modify it on save
     created_at = models.DateTimeField(verbose_name=_('created at'))
@@ -1048,6 +1070,7 @@ def __str__(self):
         return '%s - %s' % (self.appeal, self.name)
 
 
+@reversion.register()
 class AppealFilter(models.Model):
     name = models.CharField(verbose_name=_('name'), max_length=100)
     value = models.CharField(verbose_name=_('value'), max_length=1000)
@@ -1109,6 +1132,7 @@ class Labels:
         OTHER = _('OTHER')
 
 
+@reversion.register()
 class ExternalPartner(models.Model):
     ''' Dropdown items for COVID Field Reports '''
 
@@ -1177,6 +1201,7 @@ def __str__(self):
         return self.user.get_username()
 
 
+@reversion.register()
 class FieldReport(models.Model):
     """ A field report for a disaster and country, containing documents """
 
@@ -1412,6 +1437,7 @@ def __str__(self):
         return '%s - %s' % (self.id, summary)
 
 
+@reversion.register()
 class FieldReportContact(models.Model):
     """ Contact for field report """
 
@@ -1474,6 +1500,7 @@ class ActionCategory:
     )
 
 
+@reversion.register()
 class Action(models.Model):
     """ Action taken """
     name = models.CharField(verbose_name=_('name'), max_length=400)
@@ -1499,6 +1526,7 @@ def __str__(self):
         return self.name
 
 
+@reversion.register()
 class ActionsTaken(models.Model):
     """ All the actions taken by an organization """
 
@@ -1532,6 +1560,7 @@ def __str__(self):
         return self.name
 
 
+@reversion.register()
 class Source(models.Model):
     """ Source of information """
     stype = models.ForeignKey(SourceType, verbose_name=_('type'), on_delete=models.PROTECT)
@@ -1548,6 +1577,7 @@ def __str__(self):
         return '%s: %s' % (self.stype.name, self.spec)
 
 
+@reversion.register()
 class Profile(models.Model):
     """ Holds location and identifying information about users """
     NTLS = 'NTLS'
@@ -1593,6 +1623,7 @@ def __str__(self):
         return self.user.username
 
 
+@reversion.register()
 class EmergencyOperationsBase(models.Model):
     """Common fields used by EmergencyOperations* Tables"""
     is_validated = models.BooleanField(
@@ -1901,6 +1932,7 @@ def __str__(self):
         return self.raw_file_name
 
 
+@reversion.register()
 class MainContact(models.Model):
     ''' Contacts on the Resources page '''
     extent = models.CharField(verbose_name=_('extent'), max_length=300)
@@ -1930,6 +1962,7 @@ class Labels:
         ERRONEOUS = _('Erroneous')
 
 
+@reversion.register()
 class CronJob(models.Model):
     """ CronJob log row about jobs results """
     name = models.CharField(verbose_name=_('name'), max_length=100, default='')

deployments/models.py

@@ -1,3 +1,4 @@
+import reversion
 from datetime import datetime
 from enumfields import EnumIntegerField
 from enumfields import IntEnum
@@ -68,6 +69,7 @@ class Labels:
         BASECAMP_L = _('Base Camp – L')
 
 
+@reversion.register()
 class ERUOwner(models.Model):
     """ A resource that may or may not be deployed """
 
@@ -88,6 +90,7 @@ def __str__(self):
         return self.national_society_country.name
 
 
+@reversion.register()
 class ERU(models.Model):
     """ A resource that can be deployed """
     type = EnumIntegerField(ERUType, verbose_name=_('type'), default=0)
@@ -123,6 +126,7 @@ def __str__(self):
         return str(self.type.label)
 
 
+@reversion.register()
 class PersonnelDeployment(models.Model):
     country_deployed_to = models.ForeignKey(Country, verbose_name=_('country deployed to'), on_delete=models.CASCADE)
     region_deployed_to = models.ForeignKey(Region, verbose_name=_('region deployed to'), null=True, on_delete=models.SET_NULL)
@@ -156,6 +160,7 @@ def __str__(self):
         return '%s, %s' % (self.country_deployed_to, self.region_deployed_to)
 
 
+@reversion.register()
 class MolnixTag(models.Model):
     '''
     We store tags from molnix in its own model, to make m2m relations
@@ -172,6 +177,7 @@ def __str__(self):
         return self.name
 
 
+@reversion.register()
 class DeployedPerson(models.Model):
     start_date = models.DateTimeField(verbose_name=_('start date'), null=True)
     end_date = models.DateTimeField(verbose_name=_('end date'), null=True)
@@ -246,6 +252,7 @@ def get_tags_for_category(self, molnix_category):
         names = [tag['name'] for tag in tags]
         return ", ".join(names)
 
+@reversion.register()
 class PartnerSocietyActivities(models.Model):
     activity = models.CharField(verbose_name=_('activity'), max_length=50)
 
@@ -369,6 +376,7 @@ class Labels:
         EMERGENCY_OPERATION = _('Emergency Operation')
 
 
+@reversion.register()
 class RegionalProject(models.Model):
     name = models.CharField(verbose_name=_('name'), max_length=100)
     created_at = models.DateTimeField(verbose_name=_('created at'), auto_now_add=True)
@@ -383,6 +391,7 @@ def __str__(self):
 
 
 # 3W
+@reversion.register()
 class Project(models.Model):
     modified_at = models.DateTimeField(verbose_name=_('modified at'), auto_now=True)
     modified_by = models.ForeignKey(
@@ -486,6 +495,7 @@ def get_for(user, queryset=None):
         )
 
 
+@reversion.register()
 class ProjectImport(models.Model):
     """
     Track Project Imports (For Django Admin Panel)
@@ -517,6 +527,7 @@ def __str__(self):
 
 
 # -------------- Emergency 3W [Start]
+@reversion.register()
 class EmergencyProject(models.Model):
     class ActivityLead(TextChoices):
         NATIONAL_SOCIETY = 'national_society', _('National Society')
@@ -624,6 +635,7 @@ def __str__(self):
         return self.title
 
 
+@reversion.register()
 class EmergencyProjectActivitySector(models.Model):
     title = models.CharField(max_length=255, verbose_name=_('title'))
     order = models.SmallIntegerField(default=0)
@@ -632,6 +644,7 @@ def __str__(self):
         return self.title
 
 
+@reversion.register()
 class EmergencyProjectActivityAction(models.Model):
     sector = models.ForeignKey(
         EmergencyProjectActivitySector,
@@ -648,6 +661,7 @@ def __str__(self):
         return self.title
 
 
+@reversion.register()
 class EmergencyProjectActivityActionSupply(models.Model):
     action = models.ForeignKey(
         EmergencyProjectActivityAction,
@@ -662,6 +676,7 @@ def __str__(self):
         return self.title
 
 
+@reversion.register()
 class EmergencyProjectActivityLocation(models.Model):
     # Location Data
     latitude = models.FloatField(verbose_name=_('latitude'))
@@ -672,6 +687,7 @@ def __str__(self):
         return f'{self.latitude} - {self.longitude}'
 
 
+@reversion.register()
 class EmergencyProjectActivity(models.Model):
     class PeopleHouseholds(TextChoices):
         PEOPLE = 'people', _('People'),
@@ -785,6 +801,7 @@ class PeopleHouseholds(TextChoices):
 
 
 # -------------- Emergency 3W [END]
+@reversion.register()
 class ERUReadiness(models.Model):
     """ ERU Readiness concerning personnel and equipment """
     national_society = models.ForeignKey(