mirror of
https://git.sr.ht/~tsileo/microblog.pub
synced 2024-11-14 18:54:27 +00:00
84 lines
2.8 KiB
Python
84 lines
2.8 KiB
Python
|
import datetime
|
||
|
|
from dataclasses import dataclass
|
||
|
|
from typing import Optional
|
||
|
|
|
||
|
|
from loguru import logger
|
||
|
|
|
||
|
|
from app import media
|
||
|
|
from app.models import InboxObject
|
||
|
|
from app.models import Webmention
|
||
|
|
from app.utils.url import make_abs
|
||
|
|
|
||
|
|
|
||
|
|
@dataclass
|
||
|
|
class Face:
|
||
|
|
ap_actor_id: str | None
|
||
|
|
url: str
|
||
|
|
name: str
|
||
|
|
picture_url: str
|
||
|
|
created_at: datetime.datetime
|
||
|
|
|
||
|
|
@classmethod
|
||
|
|
def from_inbox_object(cls, like: InboxObject) -> "Face":
|
||
|
|
return cls(
|
||
|
|
ap_actor_id=like.actor.ap_id,
|
||
|
|
url=like.actor.url, # type: ignore
|
||
|
|
name=like.actor.handle, # type: ignore
|
||
|
|
picture_url=like.actor.resized_icon_url,
|
||
|
|
created_at=like.created_at, # type: ignore
|
||
|
|
)
|
||
|
|
|
||
|
|
@classmethod
|
||
|
|
def from_webmention(cls, webmention: Webmention) -> Optional["Face"]:
|
||
|
|
items = webmention.source_microformats.get("items", []) # type: ignore
|
||
|
|
for item in items:
|
||
|
|
if item["type"][0] == "h-card":
|
||
|
|
try:
|
||
|
|
return cls(
|
||
|
|
ap_actor_id=None,
|
||
|
|
url=webmention.source,
|
||
|
|
name=item["properties"]["name"][0],
|
||
|
|
picture_url=media.resized_media_url(
|
||
|
|
make_abs(
|
||
|
|
item["properties"]["photo"][0], webmention.source
|
||
|
|
), # type: ignore
|
||
|
|
50,
|
||
|
|
),
|
||
|
|
created_at=webmention.created_at, # type: ignore
|
||
|
|
)
|
||
|
|
except Exception:
|
||
|
|
logger.exception(
|
||
|
|
f"Failed to build Face for webmention id={webmention.id}"
|
||
|
|
)
|
||
|
|
break
|
||
|
|
elif item["type"][0] == "h-entry":
|
||
|
|
author = item["properties"]["author"][0]
|
||
|
|
try:
|
||
|
|
return cls(
|
||
|
|
ap_actor_id=None,
|
||
|
|
url=webmention.source,
|
||
|
|
name=author["properties"]["name"][0],
|
||
|
|
picture_url=media.resized_media_url(
|
||
|
|
make_abs(
|
||
|
|
author["properties"]["photo"][0], webmention.source
|
||
|
|
), # type: ignore
|
||
|
|
50,
|
||
|
|
),
|
||
|
|
created_at=webmention.created_at, # type: ignore
|
||
|
|
)
|
||
|
|
except Exception:
|
||
|
|
logger.exception(
|
||
|
|
f"Failed to build Face for webmention id={webmention.id}"
|
||
|
|
)
|
||
|
|
break
|
||
|
|
|
||
|
|
return None
|
||
|
|
|
||
|
|
|
||
|
|
def merge_faces(faces: list[Face]) -> list[Face]:
|
||
|
|
return sorted(
|
||
|
|
faces,
|
||
|
|
key=lambda f: f.created_at,
|
||
|
|
reverse=True,
|
||
|
|
)[:10]
|