def download_addon(client: HttpClient, id: int) -> Union[DownloadOk, DownloadError]: "Fetch a single add-on from AnkiWeb." try: resp = client.get(aqt.appShared + f"download/{id}?v=2.1&p={current_point_version}") if resp.status_code != 200: return DownloadError(status_code=resp.status_code) data = client.streamContent(resp) fname = re.match("attachment; filename=(.+)", resp.headers["content-disposition"]).group(1) meta = extract_meta_from_download_url(resp.url) return DownloadOk( data=data, filename=fname, mod_time=meta.mod_time, min_point_version=meta.min_point_version, max_point_version=meta.max_point_version, branch_index=meta.branch_index, ) except Exception as e: return DownloadError(exception=e)
def download(self): idx = self.lst.currentIndex() if not idx.isValid(): QMessageBox.show(self, self.windowTitle(), 'Please select a language.') return url = idx.data(Qt.UserRole) client = HttpClient() resp = client.get(url) if resp.status_code != 200: QMessageBox.information( self, self.windowTitle(), 'Downloading %s data failed.' % self.mode_str) return data = client.streamContent(resp) dir_path = os.path.join(addon_path, 'user_files', 'db', self.mode_str) os.makedirs(dir_path, exist_ok=True) dst_path = os.path.join(dir_path, '%s.json' % self.dst_lang) with open(dst_path, 'wb') as f: f.write(data) if self.mode == self.Mode.Freq: msg = 'Imported frequency data for "%s".\n\nNote that the frequency data is only applied to newly imported dictionaries for this language.' % self.dst_lang else: msg = 'Imported conjugation data for "%s".' % self.dst_lang QMessageBox.information(self, self.windowTitle(), msg) self.accept()
def download_index(server_url=DEFAULT_SERVER): server_url = normalize_url(server_url) index_url = server_url + '/index.json' client = HttpClient() resp = client.get(index_url) if resp.status_code != 200: return None data = client.streamContent(resp) return json.loads(data)
def download_addon(client: HttpClient, id: int) -> Union[DownloadOk, DownloadError]: "Fetch a single add-on from AnkiWeb." try: resp = client.get(aqt.appShared + f"download/{id}?v=2.1&p={pointVersion}") if resp.status_code != 200: return DownloadError(status_code=resp.status_code) data = client.streamContent(resp) fname = re.match( "attachment; filename=(.+)", resp.headers["content-disposition"] ).group(1) return DownloadOk(data=data, filename=fname) except Exception as e: return DownloadError(exception=e)
def run(self): from .dictionaryManager import importDict client = HttpClient() num_dicts = 0 num_installed = 0 def update_dict_progress(amt): progress = 0 if num_dicts > 0: progress = num_installed / num_dicts progress += amt / num_dicts progress_percent = round(progress * 100) self.progress_update.emit(progress_percent) for l in self.install_index: num_dicts += len(l.get('dictionaries', [])) self.log_update.emit('Installing %d dictionaries...' % num_dicts) freq_path = os.path.join(addon_path, 'user_files', 'db', 'frequency') os.makedirs(freq_path, exist_ok=True) conj_path = os.path.join(addon_path, 'user_files', 'db', 'conjugation') os.makedirs(conj_path, exist_ok=True) for l in self.install_index: if self.cancel_requested: return lname = self.force_lang if not lname: lname = l.get('name_en') if not lname: continue # Create Language try: aqt.mw.miDictDB.addLanguages([lname]) except Exception as e: # Lanugage already exists pass # Install frequency data if self.install_freq: furl = l.get('frequency_url') if furl: self.log_update.emit( 'Installing %s frequency data...' % lname) furl = self.construct_url(furl) dl_resp = client.get(furl) if dl_resp.status_code == 200: fdata = client.streamContent(dl_resp) dst_path = os.path.join(freq_path, '%s.json' % lname) with open(dst_path, 'wb') as f: f.write(fdata) else: self.log_update.emit( ' ERROR: Download failed (%d).' % dl_resp.status_code) # Install conjugation data if self.install_conj: curl = l.get('conjugation_url') if curl: self.log_update.emit( 'Installing %s conjugation data...' % lname) curl = self.construct_url(curl) dl_resp = client.get(curl) if dl_resp.status_code == 200: cdata = client.streamContent(dl_resp) dst_path = os.path.join(conj_path, '%s.json' % lname) with open(dst_path, 'wb') as f: f.write(cdata) else: self.log_update.emit( ' ERROR: Download failed (%d).' % dl_resp.status_code) # Install dictionaries for d in l.get('dictionaries', []): if self.cancel_requested: return dname = d.get('name') durl = self.construct_url(d.get('url')) self.log_update.emit('Installing %s...' % dname) self.log_update.emit(' Downloading %s...' % durl) dl_resp = client.get(durl) if dl_resp.status_code == 200: update_dict_progress(0.5) self.log_update.emit(' Importing...') ddata = client.streamContent(dl_resp) try: importDict(lname, io.BytesIO(ddata), dname) except ValueError as e: self.log_update.emit(' ERROR: %s' % str(e)) else: self.log_update.emit(' ERROR: Download failed (%d).' % dl_resp.status_code) update_dict_progress(1.0) num_installed += 1 # Only once language can be installed when language is forced if self.force_lang: # Should never happen break self.progress_update.emit(100) self.log_update.emit('All done.')