moviewyrm/bookwyrm/models/author.py

71 lines
2.3 KiB
Python
Raw Normal View History

2021-03-08 16:49:10 +00:00
""" database schema for info about authors """
2021-12-05 20:28:17 +00:00
import re
from django.contrib.postgres.indexes import GinIndex
2022-01-05 22:33:10 +00:00
from django.core.cache import cache
from django.core.cache.utils import make_template_fragment_key
from django.db import models
from bookwyrm import activitypub
2020-11-28 21:40:09 +00:00
from bookwyrm.settings import DOMAIN
from .book import BookDataModel
2020-11-30 22:40:26 +00:00
from . import fields
class Author(BookDataModel):
2021-04-26 16:15:42 +00:00
"""basic biographic info"""
2021-03-08 16:49:10 +00:00
2020-12-13 02:06:48 +00:00
wikipedia_link = fields.CharField(
2021-03-08 16:49:10 +00:00
max_length=255, blank=True, null=True, deduplication_field=True
)
2021-04-07 00:46:06 +00:00
isni = fields.CharField(
max_length=255, blank=True, null=True, deduplication_field=True
)
viaf_id = fields.CharField(
max_length=255, blank=True, null=True, deduplication_field=True
)
gutenberg_id = fields.CharField(
max_length=255, blank=True, null=True, deduplication_field=True
)
# idk probably other keys would be useful here?
2020-11-30 22:40:26 +00:00
born = fields.DateTimeField(blank=True, null=True)
died = fields.DateTimeField(blank=True, null=True)
name = fields.CharField(max_length=255)
2020-11-30 22:40:26 +00:00
aliases = fields.ArrayField(
models.CharField(max_length=255), blank=True, default=list
)
2020-12-17 00:47:05 +00:00
bio = fields.HtmlField(null=True, blank=True)
2022-01-05 22:33:10 +00:00
def save(self, *args, **kwargs):
"""clear related template caches"""
# clear template caches
if self.id:
cache_keys = [
make_template_fragment_key("titleby", [book])
for book in self.book_set.values_list("id", flat=True)
]
cache.delete_many(cache_keys)
2022-01-05 22:33:10 +00:00
return super().save(*args, **kwargs)
2021-12-05 20:28:17 +00:00
@property
def isni_link(self):
"""generate the url from the isni id"""
clean_isni = re.sub(r"\s", "", self.isni)
2021-12-05 21:24:40 +00:00
return f"https://isni.org/isni/{clean_isni}"
2021-12-05 20:28:17 +00:00
@property
def openlibrary_link(self):
"""generate the url from the openlibrary id"""
return f"https://openlibrary.org/authors/{self.openlibrary_key}"
2020-11-28 21:40:09 +00:00
def get_remote_id(self):
2021-04-26 16:15:42 +00:00
"""editions and works both use "book" instead of model_name"""
2021-09-18 18:32:00 +00:00
return f"https://{DOMAIN}/author/{self.id}"
2020-11-28 21:40:09 +00:00
activity_serializer = activitypub.Author
class Meta:
"""sets up postgres GIN index field"""
indexes = (GinIndex(fields=["search_vector"]),)