mirror of
https://github.com/ankitects/anki.git
synced 2025-09-19 22:42:25 -04:00
b23b6fbe35
Earlier today I pushed a change that split this code up into multiple repos, but that has proved to complicate things too much. So we're back to a single repo, except the individual submodules are better separated than they were before. The README files need updating again; I will push them out soon. Aside from splitting out the different modules, the sound code has moved from from anki to aqt.
83 lines
2.2 KiB
Python
83 lines
2.2 KiB
Python
# Copyright: Ankitects Pty Ltd and contributors
|
|
# -*- coding: utf-8 -*-
|
|
# License: GNU AGPL, version 3 or later; http://www.gnu.org/licenses/agpl.html
|
|
|
|
import re
|
|
import time
|
|
|
|
import aqt
|
|
from anki.hooks import addHook, remHook
|
|
from anki.lang import _
|
|
from anki.sync import AnkiRequestsClient
|
|
from aqt.qt import *
|
|
|
|
|
|
def download(mw, code):
|
|
"Download addon from AnkiWeb. Caller must start & stop progress diag."
|
|
# create downloading thread
|
|
thread = Downloader(code)
|
|
done = False
|
|
|
|
def onRecv():
|
|
if done:
|
|
return
|
|
mw.progress.update(label="%dKB downloaded" % (thread.recvTotal / 1024))
|
|
|
|
thread.recv.connect(onRecv)
|
|
thread.start()
|
|
while not thread.isFinished():
|
|
mw.app.processEvents()
|
|
thread.wait(100)
|
|
|
|
# make sure any posted events don't fire after we return
|
|
done = True
|
|
|
|
if not thread.error:
|
|
# success
|
|
return thread.data, thread.fname
|
|
else:
|
|
return "error", 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/%s?v=2.1" % self.code)
|
|
if resp.status_code == 200:
|
|
data = client.streamContent(resp)
|
|
elif resp.status_code in (403, 404):
|
|
self.error = _(
|
|
"Invalid code, or add-on not available for your version of Anki."
|
|
)
|
|
return
|
|
else:
|
|
self.error = _("Unexpected response code: %s" % resp.status_code)
|
|
return
|
|
except Exception as e:
|
|
self.error = _("Please check your internet connection.") + "\n\n" + str(e)
|
|
return
|
|
finally:
|
|
remHook("httpRecv", recvEvent)
|
|
|
|
self.fname = re.match(
|
|
"attachment; filename=(.+)", resp.headers["content-disposition"]
|
|
).group(1)
|
|
self.data = data
|