GitHub/Anki
1
0
Fork 0
mirror of https://github.com/ankitects/anki.git synced 2025-09-24 08:46:37 -04:00
Code Issues Projects Releases Packages 1 Wiki Activity Actions

new code to download remote links and replace with local ones

Browse source
This commit is contained in:
Damien Elmes 2010-12-25 12:42:28 +09:00
parent 50622a3127
commit f0bf1989bc
1 changed files with 45 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

48
anki/media.py
View file

@ -8,7 +8,7 @@ Media support
""" """
__docformat__ = 'restructuredtext' __docformat__ = 'restructuredtext'
import os, shutil, re, urllib2, time import os, shutil, re, urllib2, time, tempfile
from anki.db import * from anki.db import *
from anki.utils import checksum, genID from anki.utils import checksum, genID
from anki.lang import _ from anki.lang import _
@ -117,11 +117,14 @@ def removeUnusedMedia(deck):
# String manipulation # String manipulation
########################################################################## ##########################################################################
def mediaFiles(string): def mediaFiles(string, remote=False):
l = [] l = []
for reg in regexps: for reg in regexps:
for (full, fname) in re.findall(reg, string): for (full, fname) in re.findall(reg, string):
if not re.match("(https?|ftp)://", fname.lower()): isLocal = not re.match("(https?|ftp)://", fname.lower())
if not remote and isLocal:
l.append(fname)
elif remote and not isLocal:
l.append(fname) l.append(fname)
return l return l
@ -222,3 +225,42 @@ def downloadMissing(deck):
deck.updateProgress(label=_("File %d...") % (grabbed+missing)) deck.updateProgress(label=_("File %d...") % (grabbed+missing))
deck.finishProgress() deck.finishProgress()
return (True, grabbed, missing) return (True, grabbed, missing)
# Convert remote links to local ones
##########################################################################
def downloadRemote(deck):
mdir = deck.mediaDir(create=True)
refs = {}
deck.startProgress()
for (question, answer) in deck.s.all(
"select question, answer from cards"):
for txt in (question, answer):
for f in mediaFiles(txt, remote=True):
refs[f] = True
tmpdir = tempfile.mkdtemp(prefix="anki")
failed = []
passed = []
for c, link in enumerate(refs.keys()):
try:
path = os.path.join(tmpdir, os.path.basename(link))
url = urllib2.urlopen(link)
open(path, "wb").write(url.read())
newpath = copyToMedia(deck, path)
passed.append([link, newpath])
except:
failed.append(link)
deck.updateProgress(label=_("Download %d...") % c)
for (url, name) in passed:
deck.s.statement(
"update fields set value = replace(value, :url, :name)",
url=url, name=name)
deck.updateProgress(label=_("Updating references..."))
deck.updateProgress(label=_("Updating cards..."))
# rebuild entire q/a cache
for m in deck.models:
deck.updateCardsFromModel(m, dirty=True)
deck.finishProgress()
deck.flushMod()
return (passed, failed)