jules/spotdl-py
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix Flask app context error in SSE stream route

Browse Source
This commit is contained in:
Jules
2026-03-21 01:30:57 +01:00
parent 5860e4634b
commit ee0b0f3abd
1 changed files with 19 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
app.py
View File

@@ -358,8 +358,11 @@ def api_download():
@login_required @login_required
def stream_download(download_id): def stream_download(download_id):
download = Download.query.get_or_404(download_id) download = Download.query.get_or_404(download_id)
download_id_val = download.id
def generate(): def generate():
with app.app_context():
download = Download.query.get(download_id_val)
last_len = 0 last_len = 0
while True: while True:
db.session.refresh(download) db.session.refresh(download)
@@ -371,7 +374,7 @@ def stream_download(download_id):
yield data yield data
last_len = current_len last_len = current_len
if download.status in ('completed', 'error'): if download.status in ('completed', 'error', 'cancelled'):
data = f"data: {download.id}|{download.status}|{download.progress}|__END__\n\n" data = f"data: {download.id}|{download.status}|{download.progress}|__END__\n\n"
yield data yield data
break break