f6245cdfd1
- wrap request in AnkiRequestsClient so we can keep track of upload/download bytes without having to monkey patch anything - force a 64kB buffer size instead of the default 8kB - show one decimal point in up/down so small requests still give visual feedback - update add-on downloading and update check to use requests - remove the update throttling in aqt/sync.py, as it's not really necessary anymore
86 lines
2.7 KiB
Python
86 lines
2.7 KiB
Python
# Copyright: Damien Elmes <anki@ichi2.net>
|
|
# -*- coding: utf-8 -*-
|
|
# License: GNU AGPL, version 3 or later; http://www.gnu.org/licenses/agpl.html
|
|
|
|
import time, re, traceback
|
|
from aqt.qt import *
|
|
from anki.sync import AnkiRequestsClient
|
|
from aqt.utils import showWarning
|
|
from anki.hooks import addHook, remHook
|
|
import aqt
|
|
|
|
def download(mw, code):
|
|
"Download addon/deck from AnkiWeb. On success caller must stop progress diag."
|
|
# check code is valid
|
|
try:
|
|
code = int(code)
|
|
except ValueError:
|
|
showWarning(_("Invalid code."))
|
|
return
|
|
# create downloading thread
|
|
thread = Downloader(code)
|
|
def onRecv():
|
|
try:
|
|
mw.progress.update(label="%dKB downloaded" % (thread.recvTotal/1024))
|
|
except NameError:
|
|
# some users report the following error on long downloads
|
|
# NameError: free variable 'mw' referenced before assignment in enclosing scope
|
|
# unsure why this is happening, but guard against throwing the
|
|
# error
|
|
pass
|
|
thread.recv.connect(onRecv)
|
|
thread.start()
|
|
mw.progress.start(immediate=True)
|
|
while not thread.isFinished():
|
|
mw.app.processEvents()
|
|
thread.wait(100)
|
|
if not thread.error:
|
|
# success
|
|
return thread.data, thread.fname
|
|
else:
|
|
mw.progress.finish()
|
|
showWarning(_("Download failed: %s") % thread.error)
|
|
|
|
class Downloader(QThread):
|
|
|
|
recv = pyqtSignal()
|
|
|
|
def __init__(self, code):
|
|
QThread.__init__(self)
|
|
self.code = code
|
|
self.error = None
|
|
|
|
def run(self):
|
|
# setup progress handler
|
|
self.byteUpdate = time.time()
|
|
self.recvTotal = 0
|
|
def recvEvent(bytes):
|
|
self.recvTotal += bytes
|
|
self.recv.emit()
|
|
addHook("httpRecv", recvEvent)
|
|
client = AnkiRequestsClient()
|
|
try:
|
|
resp = client.get(
|
|
aqt.appShared + "download/%d" % self.code)
|
|
if resp.status_code == 200:
|
|
data = client.streamContent(resp)
|
|
elif resp.status_code in (403,404):
|
|
self.error = _("Invalid code")
|
|
return
|
|
else:
|
|
self.error = _("Error downloading: %s" % resp.status_code)
|
|
return
|
|
except Exception as e:
|
|
exc = traceback.format_exc()
|
|
try:
|
|
self.error = str(e[0])
|
|
except:
|
|
self.error = str(exc)
|
|
return
|
|
finally:
|
|
remHook("httpRecv", recvEvent)
|
|
|
|
self.fname = re.match("attachment; filename=(.+)",
|
|
resp.headers['content-disposition']).group(1)
|
|
self.data = data
|