diff --git a/archivist/apps/importer/models.py b/archivist/apps/importer/models.py index 12933ae..2286562 100644 --- a/archivist/apps/importer/models.py +++ b/archivist/apps/importer/models.py @@ -7,6 +7,12 @@ from apps.sites.models import Category from apps.user.models import UserProfile class ImportSourceURLs(models.Model): + + SOURCE_TYPES = ( + ('C', 'Complete User'), + ('P', 'Singel Post'), + ) + url = models.URLField(unique=True) category = models.ForeignKey(Category, on_delete=models.CASCADE) @@ -17,6 +23,8 @@ class ImportSourceURLs(models.Model): last_imported = models.DateTimeField(editable=True, blank=True, null=True) + source_type = models.CharField(max_length=1, choices=SOURCE_TYPES, default=None) + class Meta: verbose_name = _("Import Source URL") verbose_name_plural = _("Import Source URLs")