lib.itmens/journal/models/note.py
2024-06-15 22:02:19 -04:00

257 lines
8.8 KiB
Python

import re
from functools import cached_property
from typing import override
from django.db import models
from django.utils.translation import gettext_lazy as _
from loguru import logger
from catalog.models import Item
from mastodon.api import delete_toot_later
from takahe.utils import Takahe
from .common import Content
from .renderers import render_text
from .shelf import ShelfMember
_progress = re.compile(
r"^\s*(?P<prefix>(p|pg|page|ch|chapter|pt|part|e|ep|episode|trk|track|cycle))?(\s|\.|#)*(?P<value>(\d[\d\:\.\-]*\d|\d))\s*(?P<postfix>(%))?\s*(\s|\n|\.|:)",
re.IGNORECASE,
)
_number = re.compile(r"^[\s\d\:\.]+$")
class Note(Content):
class ProgressType(models.TextChoices):
PAGE = "page", _("Page")
CHAPTER = "chapter", _("Chapter")
# SECTION = "section", _("Section")
# VOLUME = "volume", _("Volume")
PART = "part", _("Part")
EPISODE = "episode", _("Episode")
TRACK = "track", _("Track")
CYCLE = "cycle", _("Cycle")
TIMESTAMP = "timestamp", _("Timestamp")
PERCENTAGE = "percentage", _("Percentage")
title = models.TextField(blank=True, null=True, default=None)
content = models.TextField(blank=False, null=False)
sensitive = models.BooleanField(default=False, null=False)
attachments = models.JSONField(default=list)
progress_type = models.CharField(
max_length=50,
choices=ProgressType.choices,
blank=True,
null=True,
default=None,
)
progress_value = models.CharField(
max_length=500, blank=True, null=True, default=None
)
_progress_display_template = {
ProgressType.PAGE: _("Page {value}"),
ProgressType.CHAPTER: _("Chapter {value}"),
# ProgressType.SECTION: _("Section {value}"),
# ProgressType.VOLUME: _("Volume {value}"),
ProgressType.PART: _("Part {value}"),
ProgressType.EPISODE: _("Episode {value}"),
ProgressType.TRACK: _("Track {value}"),
ProgressType.CYCLE: _("Cycle {value}"),
ProgressType.PERCENTAGE: "{value}%",
ProgressType.TIMESTAMP: "{value}",
}
class Meta:
indexes = [models.Index(fields=["owner", "item", "created_time"])]
@property
def html(self):
return render_text(self.content)
@property
def progress_display(self) -> str:
if not self.progress_value:
return ""
if not self.progress_type:
return str(self.progress_value)
tpl = Note._progress_display_template.get(self.progress_type, None)
if not tpl:
return str(self.progress_value)
if _number.match(self.progress_value):
return tpl.format(value=self.progress_value)
return self.progress_type.display + ": " + self.progress_value
@property
def ap_object(self):
d = {
"id": self.absolute_url,
"type": "Note",
"title": self.title,
"content": self.content,
"sensitive": self.sensitive,
"published": self.created_time.isoformat(),
"updated": self.edited_time.isoformat(),
"attributedTo": self.owner.actor_uri,
"withRegardTo": self.item.absolute_url,
"href": self.absolute_url,
}
if self.progress_value:
d["progress"] = {
"type": self.progress_type,
"value": self.progress_value,
}
return d
@override
@classmethod
def params_from_ap_object(cls, post, obj, piece):
params = {
"title": obj.get("title", post.summary),
"content": obj.get("content", "").strip(),
"sensitive": obj.get("sensitive", post.sensitive),
"attachments": [],
}
progress = obj.get("progress", {})
if progress.get("type"):
params["progress_type"] = progress.get("type")
if progress.get("value"):
params["progress_value"] = progress.get("value")
if post.local:
progress_type, progress_value = cls.extract_progress(params["content"])
print(progress_type, progress_value)
if progress_value:
params["progress_type"] = progress_type
params["progress_value"] = progress_value
if post:
for atta in post.attachments.all():
params["attachments"].append(
{
"type": (atta.mimetype or "unknown").split("/")[0],
"mimetype": atta.mimetype,
"url": atta.full_url().absolute,
"preview_url": atta.thumbnail_url().absolute,
}
)
return params
@override
@classmethod
def update_by_ap_object(cls, owner, item, obj, post):
p = super().update_by_ap_object(owner, item, obj, post)
if (
p
and p.local
and owner.user.preference.mastodon_default_repost
and owner.user.mastodon_username
):
p.sync_to_mastodon()
return p
@cached_property
def shelfmember(self) -> ShelfMember | None:
return ShelfMember.objects.filter(item=self.item, owner=self.owner).first()
def to_mastodon_params(self):
params = {
"spoiler_text": self.title,
"content": self.content,
"sensitive": self.sensitive,
"reply_to_toot_url": (
self.shelfmember.get_mastodon_repost_url() if self.shelfmember else None
),
}
if self.latest_post:
attachments = []
for atta in self.latest_post.attachments.all():
attachments.append((atta.file_display_name, atta.file, atta.mimetype))
if attachments:
params["attachments"] = attachments
return params
def to_post_params(self):
return {
"summary": self.title,
"content": self.content,
"sensitive": self.sensitive,
"reply_to_pk": (
self.shelfmember.latest_post_id if self.shelfmember else None
),
# not passing "attachments" so it won't change
}
@classmethod
def extract_progress(cls, content):
m = _progress.match(content)
if m and m["value"]:
typ_ = "percentage" if m["postfix"] == "%" else m["prefix"]
match typ_:
case "p" | "pg" | "page":
typ = Note.ProgressType.PAGE
case "ch" | "chapter":
typ = Note.ProgressType.CHAPTER
# case "vol" | "volume":
# typ = ProgressType.VOLUME
# case "section":
# typ = ProgressType.SECTION
case "pt" | "part":
typ = Note.ProgressType.PART
case "e" | "ep" | "episode":
typ = Note.ProgressType.EPISODE
case "trk" | "track":
typ = Note.ProgressType.TRACK
case "cycle":
typ = Note.ProgressType.CYCLE
case "percentage":
typ = Note.ProgressType.PERCENTAGE
case _:
typ = "timestamp" if ":" in m["value"] else None
return typ, m["value"]
return None, None
@classmethod
def get_progress_types_by_item(cls, item: Item):
match item.__class__.__name__:
case "Edition":
v = [
Note.ProgressType.PAGE,
Note.ProgressType.CHAPTER,
Note.ProgressType.PERCENTAGE,
]
case "TVShow" | "TVSeason":
v = [
Note.ProgressType.PART,
Note.ProgressType.EPISODE,
Note.ProgressType.PERCENTAGE,
]
case "Movie":
v = [
Note.ProgressType.PART,
Note.ProgressType.TIMESTAMP,
Note.ProgressType.PERCENTAGE,
]
case "Podcast":
v = [
Note.ProgressType.EPISODE,
]
case "TVEpisode" | "PodcastEpisode":
v = []
case "Album":
v = [
Note.ProgressType.TRACK,
Note.ProgressType.TIMESTAMP,
Note.ProgressType.PERCENTAGE,
]
case "Game":
v = [
Note.ProgressType.CYCLE,
]
case "Performance" | "PerformanceProduction":
v = [
Note.ProgressType.PART,
Note.ProgressType.TIMESTAMP,
Note.ProgressType.PERCENTAGE,
]
case _:
v = []
return v