acronomy/acros/models/Weblink.py

from urllib.parse import urlparse

from django.db import models
from simple_history.models import HistoricalRecords

from acros.models import Acronym, Host
from acros.utils.apis import get_website_title


class Weblink(models.Model):
    acronym = models.ForeignKey(Acronym, on_delete=models.CASCADE, related_name="links")
    url = models.URLField()
    host = models.ForeignKey(Host, on_delete=models.CASCADE, editable=False)
    title = models.CharField(max_length=500, blank=True)
    history = HistoricalRecords()

    def __str__(self):
        return self.url

    def save(self, *args, **kwargs):
        uri = urlparse(self.url)
        self.host, created = Host.objects.get_or_create(host=uri.hostname)
        self.title = get_website_title(self.url)
        super(Weblink, self).save(*args, **kwargs)