pax/booru-viewer
1
0
Fork 0
Code Issues Releases 18 Activity

Add Animated checkbox — filters to only show video/gif/animated posts

Browse Source 
Client-side filter by file extension. Works with backfill and
infinite scroll. Unchecked shows everything, checked shows only
gif/mp4/webm/mkv/mov/zip(ugoira).
This commit is contained in:
pax 2026-04-05 19:00:03 -05:00
parent 05e0a1783f
commit c577d7005a
1 changed files with 13 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
booru_viewer/gui/app.py
View File

@ -325,6 +325,11 @@ class BooruApp(QMainWindow):
score_up.clicked.connect(lambda: self._score_spin.setValue(self._score_spin.value() + 1)) score_up.clicked.connect(lambda: self._score_spin.setValue(self._score_spin.value() + 1))
top.addWidget(score_up) top.addWidget(score_up)
from PySide6.QtWidgets import QCheckBox
self._animated_only = QCheckBox("Animated")
self._animated_only.setToolTip("Only show animated/video posts")
top.addWidget(self._animated_only)
self._search_bar = SearchBar(db=self._db) self._search_bar = SearchBar(db=self._db)
self._search_bar.search_requested.connect(self._on_search) self._search_bar.search_requested.connect(self._on_search)
self._search_bar.autocomplete_requested.connect(self._request_autocomplete) self._search_bar.autocomplete_requested.connect(self._request_autocomplete)
@ -619,12 +624,16 @@ class BooruApp(QMainWindow):
bl_tags = set(self._db.get_blacklisted_tags()) bl_tags = set(self._db.get_blacklisted_tags())
bl_posts = self._db.get_blacklisted_posts() bl_posts = self._db.get_blacklisted_posts()
shown_ids = getattr(self, '_shown_post_ids', set()).copy() shown_ids = getattr(self, '_shown_post_ids', set()).copy()
animated_only = self._animated_only.isChecked()
_ANIM_EXTS = {".gif", ".mp4", ".webm", ".mkv", ".mov", ".avi", ".zip"}
def _filter(posts): def _filter(posts):
if bl_tags: if bl_tags:
posts = [p for p in posts if not bl_tags.intersection(p.tag_list)] posts = [p for p in posts if not bl_tags.intersection(p.tag_list)]
if bl_posts: if bl_posts:
posts = [p for p in posts if p.file_url not in bl_posts] posts = [p for p in posts if p.file_url not in bl_posts]
if animated_only:
posts = [p for p in posts if Path(p.file_url.split("?")[0]).suffix.lower() in _ANIM_EXTS]
posts = [p for p in posts if p.id not in shown_ids] posts = [p for p in posts if p.id not in shown_ids]
return posts return posts
@ -731,12 +740,16 @@ class BooruApp(QMainWindow):
bl_tags = set(self._db.get_blacklisted_tags()) bl_tags = set(self._db.get_blacklisted_tags())
bl_posts = self._db.get_blacklisted_posts() bl_posts = self._db.get_blacklisted_posts()
shown_ids = getattr(self, '_shown_post_ids', set()).copy() shown_ids = getattr(self, '_shown_post_ids', set()).copy()
animated_only = self._animated_only.isChecked()
_ANIM_EXTS = {".gif", ".mp4", ".webm", ".mkv", ".mov", ".avi", ".zip"}
def _filter(posts): def _filter(posts):
if bl_tags: if bl_tags:
posts = [p for p in posts if not bl_tags.intersection(p.tag_list)] posts = [p for p in posts if not bl_tags.intersection(p.tag_list)]
if bl_posts: if bl_posts:
posts = [p for p in posts if p.file_url not in bl_posts] posts = [p for p in posts if p.file_url not in bl_posts]
if animated_only:
posts = [p for p in posts if Path(p.file_url.split("?")[0]).suffix.lower() in _ANIM_EXTS]
# Skip posts already shown on previous pages # Skip posts already shown on previous pages
posts = [p for p in posts if p.id not in shown_ids] posts = [p for p in posts if p.id not in shown_ids]
return posts return posts