Adds link domain table

This commit is contained in:
Mouse Reeve
2022-01-09 13:19:29 -08:00
parent 4931e5cad2
commit bae355e8d2
2 changed files with 66 additions and 5 deletions

View File

@ -1,5 +1,6 @@
""" outlink data """
from django.db import models
from django.utils.translation import gettext_lazy as _
from bookwyrm import activitypub
from .activitypub_mixin import ActivitypubMixin
@ -11,7 +12,6 @@ class Link(ActivitypubMixin, BookWyrmModel):
"""a link to a website"""
url = fields.URLField(max_length=255, activitypub_field="href")
name = fields.CharField(max_length=255)
activity_serializer = activitypub.Link
reverse_unfurl = True
@ -31,3 +31,24 @@ class FileLink(Link):
"Book", on_delete=models.CASCADE, related_name="file_links", null=True
)
filetype = fields.CharField(max_length=5, activitypub_field="mediaType")
StatusChoices = [
("approved", _("Approved")),
("blocked", _("Blocked")),
("pending", _("Pending")),
]
class LinkDomain(BookWyrmModel):
"""List of domains used in links"""
domain = models.CharField(max_length=255)
status = models.CharField(max_length=50, choices=StatusChoices, default="pending")
name = models.CharField(max_length=100)
def save(self, *args, **kwargs):
"""set a default name"""
if not self.name:
self.name = self.domain
super().save(*args, **kwargs)