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

Priority downloads — clicked post pauses prefetch

Browse Source 
When a post is activated, prefetch pauses via asyncio.Event.
The clicked post downloads at full speed without competing.
Prefetch resumes after the download completes.
This commit is contained in:
pax 2026-04-05 16:26:29 -05:00
parent ee9d06755f
commit 4512cba629
1 changed files with 6 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
booru_viewer/gui/app.py
View File

@ -219,6 +219,8 @@ class BooruApp(QMainWindow):
self._min_score = 0 self._min_score = 0
self._loading = False self._loading = False
self._last_scroll_page = 0 self._last_scroll_page = 0
self._prefetch_pause = asyncio.Event()
self._prefetch_pause.set() # not paused
self._signals = AsyncSignals() self._signals = AsyncSignals()
self._async_loop = asyncio.new_event_loop() self._async_loop = asyncio.new_event_loop()
@ -932,6 +934,7 @@ class BooruApp(QMainWindow):
self._signals.download_progress.emit(downloaded, total) self._signals.download_progress.emit(downloaded, total)
async def _load(): async def _load():
self._prefetch_pause.clear() # pause prefetch
try: try:
path = await download_image(post.file_url, progress_callback=_progress) path = await download_image(post.file_url, progress_callback=_progress)
info = f"#{post.id} {post.width}x{post.height} score:{post.score} [{post.rating}] {Path(post.file_url.split('?')[0]).suffix.lstrip('.').upper() if post.file_url else ''}" info = f"#{post.id} {post.width}x{post.height} score:{post.score} [{post.rating}] {Path(post.file_url.split('?')[0]).suffix.lstrip('.').upper() if post.file_url else ''}"
@ -939,6 +942,8 @@ class BooruApp(QMainWindow):
except Exception as e: except Exception as e:
log.error(f"Image download failed: {e}") log.error(f"Image download failed: {e}")
self._signals.image_error.emit(str(e)) self._signals.image_error.emit(str(e))
finally:
self._prefetch_pause.set() # resume prefetch
self._run_async(_load) self._run_async(_load)
@ -977,6 +982,7 @@ class BooruApp(QMainWindow):
async def _prefetch_spiral(): async def _prefetch_spiral():
for adj in order: for adj in order:
await self._prefetch_pause.wait() # yield to active downloads
if 0 <= adj < len(self._posts) and self._posts[adj].file_url: if 0 <= adj < len(self._posts) and self._posts[adj].file_url:
self._signals.prefetch_progress.emit(adj, 0.0) self._signals.prefetch_progress.emit(adj, 0.0)
try: try: