new bsky sensitive contents

This commit is contained in:
2024-11-12 17:23:46 +08:00
parent 282cf46d58
commit afc31430e7
2 changed files with 14 additions and 1 deletions
+9 -1
View File
@@ -11,6 +11,7 @@ if TYPE_CHECKING:
bsky_api_url = 'https://public.api.bsky.app/xrpc/app.bsky.feed.getPostThread' bsky_api_url = 'https://public.api.bsky.app/xrpc/app.bsky.feed.getPostThread'
SENSITIVE_TAG = {'sexual', 'nudity', 'porn', 'graphic-media'}
class BskyMedia: class BskyMedia:
__slots__ = ('_url', '_thumb', '_type', '__dict__') __slots__ = ('_url', '_thumb', '_type', '__dict__')
@@ -101,7 +102,7 @@ class ProcessBsky:
author_id=self._bsky['author']['handle'], author_id=self._bsky['author']['handle'],
text=self._bsky['record']['text'], text=self._bsky['record']['text'],
media=self._bsky_media, media=self._bsky_media,
sensitive=False # TODO: implement sensitive sensitive=self._sensitive
) )
async def __aexit__(self, exc_type, exc_val, exc_tb): async def __aexit__(self, exc_type, exc_val, exc_tb):
@@ -152,3 +153,10 @@ class ProcessBsky:
] ]
case _: case _:
raise NotImplementedError(f"Unknown Bsky embed type: {embed['$type']}") raise NotImplementedError(f"Unknown Bsky embed type: {embed['$type']}")
@property
def _sensitive(self) -> bool:
return any(
tag in self._bsky['labels']
for tag in SENSITIVE_TAG
)
+5
View File
@@ -31,6 +31,7 @@ class BskyPost(TypedDict):
author: BskyAuthor author: BskyAuthor
record: BskyPostRecord record: BskyPostRecord
embed: BskyEmbedImages | BskyEmbedVideo | BskyEmbedExternal embed: BskyEmbedImages | BskyEmbedVideo | BskyEmbedExternal
labels: list[BskyLabel]
class BskyAuthor(TypedDict): class BskyAuthor(TypedDict):
@@ -42,6 +43,10 @@ class BskyPostRecord(TypedDict):
text: str text: str
class BskyLabel(TypedDict):
val: str
class BskyEmbedImage(TypedDict): class BskyEmbedImage(TypedDict):
thumb: str thumb: str
fullsize: str fullsize: str