# -*- coding: utf-8 -*- # Copyright: Damien Elmes # License: GNU GPL, version 3 or later; http://www.gnu.org/copyleft/gpl.html from PyQt4.QtGui import * from PyQt4.QtCore import * from PyQt4.QtSvg import * # fixme: obsolete? import re, os, sys, tempfile, urllib2, ctypes, simplejson from anki.utils import stripHTML from anki.sound import play from anki.hooks import addHook, removeHook, runHook, runFilter from aqt.sound import getAudio from aqt.webview import AnkiWebView from aqt.utils import shortcut import anki.js # fixme: add code to escape from text field _html = """

""" # fixme: use shortcut() for mac shortcuts if sys.platform.startswith("win32"): ActivateKeyboardLayout = ctypes.windll.user32.ActivateKeyboardLayout ActivateKeyboardLayout.restype = ctypes.c_void_p ActivateKeyboardLayout.argtypes = [ctypes.c_void_p, ctypes.c_uint] GetKeyboardLayout = ctypes.windll.user32.GetKeyboardLayout GetKeyboardLayout.restype = ctypes.c_void_p GetKeyboardLayout.argtypes = [ctypes.c_uint] class Editor(object): def __init__(self, mw, widget): self.widget = widget self.mw = mw self.fact = None self.onChange = None self._loaded = False # to be handled js side #self.lastFocusedEdit = None self.changeTimer = None # current card, for card layout self.card = None addHook("deckClosed", self.deckClosedHook) addHook("guiReset", self.refresh) addHook("colourChanged", self.colourChanged) self.setupOuter() self.setupButtons() self.setupWeb() self.setupTags() def close(self): removeHook("deckClosed", self.deckClosedHook) removeHook("guiReset", self.refresh) removeHook("colourChanged", self.colourChanged) # Initial setup ############################################################ def setupOuter(self): l = QVBoxLayout()#self.widget) l.setMargin(0) l.setSpacing(3) self.widget.setLayout(l) self.outerLayout = l def _addButton(self, name, func, key=None, tip=None, size=True, text="", check=False): b = QPushButton(text) if check: b.connect(b, SIGNAL("clicked(bool)"), func) else: b.connect(b, SIGNAL("clicked()"), func) if size: b.setFixedHeight(20) b.setFixedWidth(20) b.setStyle(self.plastiqueStyle) b.setFocusPolicy(Qt.NoFocus) if not text: b.setIcon(QIcon(":/icons/%s.png" % name)) if key: b.setShortcut(key) if tip: b.setToolTip(tip) if check: b.setCheckable(True) self.iconsBox.addWidget(b) self._buttons[name] = b return b def setupButtons(self): self._buttons = {} # button styles for mac self.plastiqueStyle = QStyleFactory.create("plastique") self.widget.setStyle(self.plastiqueStyle) # icons self.iconsBox = QHBoxLayout() self.iconsBox.setMargin(0) self.iconsBox.setSpacing(0) self.outerLayout.addLayout(self.iconsBox) # align to right self.iconsBox.addItem(QSpacerItem(20,1, QSizePolicy.Expanding)) b = self._addButton b("layout", self.onCardLayout, "Ctrl+l", shortcut(_("Layout (Ctrl+l)")), size=False, text=_("Layout...")) b("text_bold", self.toggleBold, "Ctrl+b", _("Bold text (Ctrl+b)"), check=True) b("text_italic", self.toggleItalic, "Ctrl+i", _("Italic text (Ctrl+i)"), check=True) b("text_under", self.toggleUnderline, "Ctrl+u", _("Underline text (Ctrl+u)"), check=True) b("text_super", self.toggleSuper, "Ctrl+=", _("Superscript (Ctrl+=)"), check=True) b("text_sub", self.toggleSub, "Ctrl+Shift+=", _("Subscript (Ctrl+Shift+=)"), check=True) #self.setupForegroundButton() but = b("cloze", self.onCloze, "F9", _("Cloze (F9)"), text="[...]") but.setFixedWidth(24) # fixme: better image names but = b("colors", self.onAddPicture, "F3", _("Add picture (F3)")) but = b("text-speak", self.onAddSound, "F3", _("Add audio/video (F4)")) but = b("media-record", self.onRecSound, "F5", _("Record audio (F5)")) but = b("tex", self.latexMenu, "Ctrl+t", _("LaTeX (Ctrl+t)")) # insertLatex, insertLatexEqn, insertLatexMathEnv but = b("text-xml", self.onHtmlEdit, "Ctrl+x", _("Source (Ctrl+x)")) def setupForegroundButton(self): # foreground color self.foreground = QPushButton() self.foreground.connect(self.foreground, SIGNAL("clicked()"), self.onForeground) self.foreground.setToolTip( _("Set colour (F7; repeat to choose next; F6 to use)")) self.foreground.setShortcut(_("F7")) self.foreground.setFocusPolicy(Qt.NoFocus) self.foreground.setEnabled(False) self.foreground.setFixedWidth(20) self.foreground.setFixedHeight(20) self.foregroundFrame = QFrame() self.foregroundFrame.setAutoFillBackground(True) self.colourChanged() hbox = QHBoxLayout() hbox.addWidget(self.foregroundFrame) hbox.setMargin(5) self.foreground.setLayout(hbox) self.iconsBox.addWidget(self.foreground) self.foreground.setStyle(self.plastiqueStyle) self.iconsBox.addItem(QSpacerItem(5,1, QSizePolicy.Fixed)) def setupWeb(self): self.web = AnkiWebView(self.widget) self.web.setBridge(self.bridge) self.outerLayout.addWidget(self.web) # pick up the window colour p = self.web.palette() p.setBrush(QPalette.Base, Qt.transparent) self.web.page().setPalette(p) self.web.setAttribute(Qt.WA_OpaquePaintEvent, False) self.web.setHtml(_html % anki.js.all, loadCB=self._loadFinished) def bridge(self, str): print str if str.startswith("focus") or str.startswith("key"): (type, num, txt) = str.split(":", 2) self.fact._fields[int(num)] = txt if type == "focus": runHook("editor.focusLost", self.fact) else: runHook("editor.keyPressed", self.fact) self.fact.flush() elif str.startswith("state"): (cmd, txt) = str.split(":", 1) r = simplejson.loads(txt) self._buttons['text_bold'].setChecked(r['bold']) self._buttons['text_italic'].setChecked(r['italic']) self._buttons['text_under'].setChecked(r['under']) self._buttons['text_super'].setChecked(r['super']) self._buttons['text_sub'].setChecked(r['sub']) def _loadFinished(self, w): self._loaded = True if self.fact: self.loadFact() def setupTags(self): return # # scrollarea # self.fieldsScroll = QScrollArea() # self.fieldsScroll.setWidgetResizable(True) # self.fieldsScroll.setLineWidth(0) # self.fieldsScroll.setFrameStyle(0) # self.fieldsScroll.setFocusPolicy(Qt.NoFocus) # self.fieldsBox.addWidget(self.fieldsScroll) # # tags # self.tagsBox = QHBoxLayout() # self.tagsLabel = QLabel(_("Tags")) # self.tagsBox.addWidget(self.tagsLabel) # import aqt.tagedit # self.tags = aqt.tagedit.TagEdit(self.parent) # self.tags.connect(self.tags, SIGNAL("lostFocus"), # self.onTagChange) # self.tagsBox.addWidget(self.tags) # self.fieldsBox.addLayout(self.tagsBox) # update available tags self.tags.setDeck(self.deck) tagsw = self.tagsLabel.sizeHint().width() self.tagsLabel.setFixedWidth(max(tagsw, max(*([ l.width() for l in self.labels] + [0]))) + extra) pass def setFact(self, fact): "Make FACT the current fact." self.fact = fact if self.changeTimer: self.changeTimer.stop() self.changeTimer = None if self.fact: self.loadFact() else: self.widget.hide() def loadFact(self): if not self._loaded: # will be loaded when page is ready return # fixme: focus on first widget self.web.eval("setFields(%s);" % simplejson.dumps(self.fact.items())) self.widget.show() def refresh(self): if self.fact: self.fact.load() # fixme: what if fact is deleted? self.setFact(self.fact) def initMedia(self): os.chdir(self.deck.mediaDir(create=True)) def deckClosedHook(self): self.setFact(None) # if field.fieldModel.features: # w.setLayoutDirection(Qt.RightToLeft) # else: # w.setLayoutDirection(Qt.LeftToRight) # catch changes w.connect(w, SIGNAL("lostFocus"), lambda w=w: self.onFocusLost(w)) w.connect(w, SIGNAL("textChanged()"), self.onTextChanged) w.connect(w, SIGNAL("currentCharFormatChanged(QTextCharFormat)"), lambda w=w: self.formatChanged(w)) return w def loadFields(self, check=True, font=True): "Update field text (if changed) and font/colours." # text for field in self.fact.fields: w = self.fields[field.name][1] self.fields[field.name] = (field, w) self.widgets[w] = field new = self.fact[field.name] #old = tidyHTML(unicode(w.toHtml())) # only update if something has changed if new != old: cur = w.textCursor() w.setHtml('' + new) w.setTextCursor(cur) if font: # apply fonts font = QFont() # family family = field.fieldModel.quizFontFamily if family: font.setFamily(family) # size size = field.fieldModel.editFontSize if size: font.setPixelSize(size) w.setFont(font) self.tags.blockSignals(True) self.tags.setText(self.fact.tags) self.tags.blockSignals(False) if check: self.checkValid() def saveFields(self): "Save field text into fact." modified = False n = _("Edit") self.deck.setUndoStart(n, merge=True) for (w, f) in self.widgets.items(): #v = tidyHTML(unicode(w.toHtml())) if self.fact[f.name] != v: self.fact[f.name] = v modified = True if modified: self.fact.setModified(textChanged=True, deck=self.deck) if not self.fact.isNew(): self.deck.setModified() self.deck.setUndoEnd(n) return modified def onFocusLost(self, widget): from aqt import mw if not self.fact: # editor or deck closed return if mw.inDbHandler: return modified = self.saveFields() field = self.widgets[widget] self.fact.focusLost(field) self.fact.setModified(textChanged=True, deck=self.deck) self.loadFields(font=False) if modified: aqt.mw.reset(runHooks=False) def onTextChanged(self): interval = 250 if self.changeTimer: self.changeTimer.setInterval(interval) else: self.changeTimer = QTimer(self.parent) self.changeTimer.setSingleShot(True) self.changeTimer.start(interval) self.parent.connect(self.changeTimer, SIGNAL("timeout()"), self.onChangeTimer) def onChangeTimer(self): from aqt import mw interval = 250 if not self.fact: return if mw.inDbHandler: self.changeTimer.start(interval) return self.checkValid() self.changeTimer = None def saveFieldsNow(self): "Must call this before adding cards, closing dialog, etc." if not self.fact: return # disable timer if self.changeTimer: self.changeTimer.stop() self.changeTimer = None if self.onChange: self.onChange('field') # save fields and run features w = self.focusedEdit() if w: self.onFocusLost(w) self.onTagChange() # ensure valid self.checkValid() def checkValid(self): empty = [] dupe = [] for field in self.fact.fields: p = QPalette() p.setColor(QPalette.Text, QColor("#000000")) if not self.fieldValid(field): empty.append(field) p.setColor(QPalette.Base, QColor("#ffffcc")) self.fields[field.name][1].setPalette(p) elif not self.fieldUnique(field): dupe.append(field) p.setColor(QPalette.Base, QColor("#ffcccc")) self.fields[field.name][1].setPalette(p) else: p.setColor(QPalette.Base, QColor("#ffffff")) self.fields[field.name][1].setPalette(p) def textForField(self, field): "Current edited value for field." w = self.fields[field.name][1] #v = tidyHTML(unicode(w.toHtml())) return v def fieldValid(self, field): return not (field.fieldModel.required and not self.textForField(field).strip()) def fieldUnique(self, field): if not field.fieldModel.unique: return True req = ("select value from fields " "where fieldModelId = :fmid and value = :val and id != :id " "and chksum = :chk") val = self.textForField(field) return not self.deck.db.scalar( req, val=val, fmid=field.fieldModel.id, id=field.id, chk=fieldChecksum(val)) def onTagChange(self): if not self.fact: return old = self.fact.tags self.fact.tags = canonifyTags(unicode(self.tags.text())) if old != self.fact.tags: self.deck.db.flush() self.deck.updateFactTags([self.fact.id]) self.fact.setModified(textChanged=True, deck=self.deck) self.deck.flushMod() aqt.mw.reset(runHooks=False) if self.onChange: self.onChange('tag') def focusField(self, fieldName): self.fields[fieldName][1].setFocus() def formatChanged(self, fmt): w = self.focusedEdit() if not w: return else: l = self.bold, self.italic, self.underline for b in l: b.blockSignals(True) self.bold.setChecked(w.fontWeight() == QFont.Bold) self.italic.setChecked(w.fontItalic()) self.underline.setChecked(w.fontUnderline()) for b in l: b.blockSignals(False) def resetFormatButtons(self): for b in self.bold, self.italic, self.underline: b.blockSignals(True) b.setChecked(False) b.blockSignals(False) def enableButtons(self, val=True): self.bold.setEnabled(val) self.italic.setEnabled(val) self.underline.setEnabled(val) self.foreground.setEnabled(val) self.addPicture.setEnabled(val) self.addSound.setEnabled(val) self.latex.setEnabled(val) self.latexEqn.setEnabled(val) self.latexMathEnv.setEnabled(val) self.cloze.setEnabled(val) self.htmlEdit.setEnabled(val) self.recSound.setEnabled(val) def disableButtons(self): self.enableButtons(False) def focusedEdit(self): for (name, (field, w)) in self.fields.items(): if w.hasFocus(): return w return None def toggleBold(self, bool): self.web.eval("setFormat('bold');") def toggleItalic(self, bool): self.web.eval("setFormat('italic');") def toggleUnderline(self, bool): self.web.eval("setFormat('underline');") def toggleSuper(self, bool): self.web.eval("setFormat('superscript');") def toggleSub(self, bool): self.web.eval("setFormat('subscript');") def _updateForegroundButton(self, txtcol): self.foregroundFrame.setPalette(QPalette(QColor(txtcol))) self.foregroundFrame.setStyleSheet("* {background-color: %s}" % txtcol) def colourChanged(self): recent = aqt.mw.config['recentColours'] self._updateForegroundButton(recent[-1]) def onForeground(self): self.lastFocusedEdit = self.focusedEdit() p = ColourPopup(self.parent) p.move(self.foreground.mapToGlobal(QPoint(0,0))) g = QGridLayout(p) g.setMargin(4) g.setSpacing(0) p.setLayout(g) lastWidget = None self.colourNext = QShortcut(QKeySequence("F7"), p) p.connect(self.colourNext, SIGNAL("activated()"), self.onNextColour) self.colourChoose = QShortcut(QKeySequence("F6"), p) p.connect(self.colourChoose, SIGNAL("activated()"), self.onChooseColourKey) for n, c in enumerate(reversed(aqt.mw.config['recentColours'])): col = QToolButton() col.setAutoRaise(True) col.setFixedWidth(64) col.setFixedHeight(16) col.setAutoFillBackground(True) col.setPalette(QPalette(QColor(c))) col.setStyleSheet("* {background-color: %s}" % c) col.connect(col, SIGNAL("clicked()"), lambda c=c: self.onChooseColour(c)) g.addWidget(col, n, 0) if lastWidget: p.setTabOrder(lastWidget, col) lastWidget = col but = QPushButton("X") but.setFixedWidth(16) but.setFixedHeight(16) but.setAutoDefault(False) but.connect(but, SIGNAL("clicked()"), lambda c=c: self.onRemoveColour(c)) g.addWidget(but, n, 1) spc = QSpacerItem(5,10, QSizePolicy.Fixed) g.addItem(spc, n+1, 0) cb = QPushButton(_("+")) cb.connect(cb, SIGNAL("clicked()"), self.onNewColour) cb.setFixedWidth(80) cb.setFixedHeight(16) cb.setAutoDefault(False) g.addWidget(cb, n+2, 0, 1, 2) self.colourDiag = p p.show() def onRemoveColour(self, colour): recent = aqt.mw.config['recentColours'] recent.remove(colour) if not recent: recent.append("#000000") self.colourDiag.close() self.onForeground() runHook("colourChanged") def onNextColour(self): try: self.colourDiag.focusWidget().nextInFocusChain().setFocus() except: ui.utils.showInfo("Your Qt version is too old to support this.") def onChooseColourKey(self): try: self.colourDiag.focusWidget().click() except: # dialog focused pass def onChooseColour(self, colour): recent = aqt.mw.config['recentColours'] recent.remove(colour) recent.append(colour) w = self.lastFocusedEdit w.setTextColor(QColor(colour)) self.colourDiag.close() runHook("colourChanged") def onNewColour(self): new = QColorDialog.getColor(Qt.white, self.parent) self.parent.raise_() recent = aqt.mw.config['recentColours'] if new.isValid(): txtcol = unicode(new.name()) if txtcol not in recent: recent.append(txtcol) runHook("colourChanged") self.onChooseColour(txtcol) def latexMenu(self): pass def insertLatex(self): w = self.focusedEdit() if w: selected = w.textCursor().selectedText() self.deck.mediaDir(create=True) cur = w.textCursor() pos = cur.position() w.insertHtml("[latex]%s[/latex]" % selected) cur.setPosition(pos+7) w.setTextCursor(cur) def insertLatexEqn(self): w = self.focusedEdit() if w: selected = w.textCursor().selectedText() self.deck.mediaDir(create=True) cur = w.textCursor() pos = cur.position() w.insertHtml("[$]%s[/$]" % selected) cur.setPosition(pos+3) w.setTextCursor(cur) def insertLatexMathEnv(self): w = self.focusedEdit() if w: selected = w.textCursor().selectedText() self.deck.mediaDir(create=True) cur = w.textCursor() pos = cur.position() w.insertHtml("[$$]%s[/$$]" % selected) cur.setPosition(pos+4) w.setTextCursor(cur) def onMore(self, toggle=None): if toggle is None: toggle = not self.latex.isVisible() aqt.mw.config['factEditorAdvanced'] = toggle self.latex.setShown(toggle) self.latexEqn.setShown(toggle) self.latexMathEnv.setShown(toggle) self.htmlEdit.setShown(toggle) def onCardLayout(self): from aqt.clayout import CardLayout if self.card: type = 1; ord = self.card.ord else: type = 0; ord = 0 CardLayout(self.mw, self.fact, type=type, ord=ord, parent=self.widget) # FIXME: in some future version, we should use a different delimiter, as # [sound] et al conflicts def onCloze(self): src = self.focusedEdit() if not src: return re1 = "\[(?:<.+?>)?.+?(:(.+?))?\](?:)?" re2 = "\[(?:<.+?>)?(.+?)(:.+?)?\](?:)?" # add brackets because selected? cursor = src.textCursor() oldSrc = None if cursor.hasSelection(): oldSrc = src.toHtml() s = cursor.selectionStart() e = cursor.selectionEnd() cursor.setPosition(e) cursor.insertText("]]") cursor.setPosition(s) cursor.insertText("[[") re1 = "\[" + re1 + "\]" re2 = "\[" + re2 + "\]" dst = None for field in self.fact.fields: w = self.fields[field.name][1] if w.hasFocus(): dst = False continue if dst is False: dst = w break if not dst: dst = self.fields[self.fact.fields[0].name][1] if dst == w: return # check if there's alredy something there if not oldSrc: oldSrc = src.toHtml() oldDst = dst.toHtml() if unicode(dst.toPlainText()): if (self.lastCloze and self.lastCloze[1] == oldSrc and self.lastCloze[2] == oldDst): src.setHtml(self.lastCloze[0]) dst.setHtml("") self.lastCloze = None self.saveFields() return else: ui.utils.showInfo( _("Next field must be blank."), help="ClozeDeletion", parent=self.parent) return # check if there's anything to change if not re.search("\[.+?\]", unicode(src.toPlainText())): ui.utils.showInfo( _("You didn't specify anything to occlude."), help="ClozeDeletion", parent=self.parent) return # create s = unicode(src.toHtml()) def repl(match): exp = "" if match.group(2): exp = match.group(2) return '[...%s]' % ( clozeColour, exp) new = re.sub(re1, repl, s) old = re.sub(re2, '\\1' % clozeColour, s) src.setHtml(new) dst.setHtml(old) self.lastCloze = (oldSrc, unicode(src.toHtml()), unicode(dst.toHtml())) self.saveFields() def onHtmlEdit(self): def helpRequested(): aqt.openHelp("HtmlEditor") w = self.focusedEdit() if w: self.saveFields() d = QDialog(self.parent) form = aqt.forms.edithtml.Ui_Dialog() form.setupUi(d) d.connect(form.buttonBox, SIGNAL("helpRequested()"), helpRequested) form.textEdit.setPlainText(self.widgets[w].value) form.textEdit.moveCursor(QTextCursor.End) d.exec_() w.setHtml(unicode(form.textEdit.toPlainText()).\ replace("\n", "")) self.saveFields() def fieldsAreBlank(self): for (field, widget) in self.fields.values(): #value = tidyHTML(unicode(widget.toHtml())) if value: return False return True def onAddPicture(self): # get this before we open the dialog w = self.focusedEdit() key = (_("Images") + " (*.jpg *.png *.gif *.tiff *.svg *.tif *.jpeg)") file = ui.utils.getFile(self.parent, _("Add an image"), "picture", key) if not file: return if file.lower().endswith(".svg"): # convert to a png s = QSvgRenderer(file) i = QImage(s.defaultSize(), QImage.Format_ARGB32_Premultiplied) p = QPainter() p.begin(i) s.render(p) p.end() (fd, name) = tempfile.mkstemp(prefix="anki", suffix=".png") file = unicode(name, sys.getfilesystemencoding()) i.save(file) self._addPicture(file, widget=w) def _addPicture(self, file, widget=None): self.initMedia() if widget: w = widget else: w = self.focusedEdit() path = self._addMedia(file) self.maybeDelete(path, file) w.insertHtml('

' % path) def _addMedia(self, file): try: return self.deck.addMedia(file) except (IOError, OSError), e: ui.utils.showWarning(_("Unable to add media: %s") % unicode(e), parent=self.parent) def onAddSound(self): # get this before we open the dialog w = self.focusedEdit() key = (_("Sounds/Videos") + " (*.mp3 *.ogg *.wav *.avi *.ogv *.mpg *.mpeg *.mov *.mp4 " + "*.mkv *.ogx *.ogv *.oga *.flv *.swf *.flac)") file = ui.utils.getFile(self.parent, _("Add audio"), "audio", key) if not file: return self._addSound(file, widget=w) def _addSound(self, file, widget=None, copy=True): self.initMedia() if widget: w = widget else: w = self.focusedEdit() if copy: path = self._addMedia(file) self.maybeDelete(path, file) else: path = file anki.sound.play(path) w.insertHtml('[sound:%s]' % path) def maybeDelete(self, new, old): if not aqt.mw.config['deleteMedia']: return if new == os.path.basename(old): return try: os.unlink(old) except: pass def onRecSound(self): self.initMedia() w = self.focusedEdit() try: file = getAudio(self.parent) except: if sys.platform.startswith("darwin"): ui.utils.showInfo(_('''\ Please install lame to enable recording.'''), parent=self.parent) return raise if file: self._addSound(file, w, copy=False) class FactEdit(QTextEdit): def __init__(self, parent, *args): QTextEdit.__init__(self, *args) self.parent = parent self._tmpDir = None if sys.platform.startswith("win32"): self._ownLayout = None def canInsertFromMimeData(self, source): return (source.hasUrls() or source.hasText() or source.hasImage() or source.hasHtml()) def insertFromMimeData(self, source): if self._insertFromMimeData(source): self.emit(SIGNAL("lostFocus")) def _insertFromMimeData(self, source): pics = ("jpg", "jpeg", "png", "tif", "tiff", "gif") audio = ("wav", "mp3", "ogg", "flac") errtxt = _("An error occured while opening %s") if source.hasHtml() and "qrichtext" in unicode(source.html()): self.insertHtml(source.html()) return True if source.hasText() and (aqt.mw.config['stripHTML'] or not source.hasHtml()): txt = unicode(source.text()) l = txt.lower() if l.startswith("http://") or l.startswith("file://"): hadN = False if "\n" in txt: txt = txt.split("\n")[0] hadN = True if "\r" in txt: txt = txt.split("\r")[0] hadN = True if not source.hasImage() or hadN: # firefox on linux just gives us a url ext = txt.split(".")[-1].lower() try: if ext in pics: name = self._retrieveURL(txt) self.parent._addPicture(name, widget=self) elif ext in audio: name = self._retrieveURL(txt) self.parent._addSound(name, widget=self) else: # not image or sound, treat as plain text self.insertPlainText(source.text()) return True except urllib2.URLError, e: ui.utils.showWarning(errtxt % e) else: self.insertPlainText(source.text()) return True if source.hasImage(): im = QImage(source.imageData()) if im.hasAlphaChannel(): (fd, name) = tempfile.mkstemp(prefix="paste", suffix=".png") uname = unicode(name, sys.getfilesystemencoding()) im.save(uname) else: (fd, name) = tempfile.mkstemp(prefix="paste", suffix=".jpg") uname = unicode(name, sys.getfilesystemencoding()) im.save(uname, None, 95) self.parent._addPicture(uname, widget=self) return True if source.hasUrls(): for url in source.urls(): url = unicode(url.toString()) ext = url.split(".")[-1].lower() try: if ext in pics: name = self._retrieveURL(url) self.parent._addPicture(name, widget=self) elif ext in audio: name = self._retrieveURL(url) self.parent._addSound(name, widget=self) except urllib2.URLError, e: ui.utils.showWarning(errtxt % e) return True if source.hasHtml(): self.insertHtml(self.simplifyHTML(unicode(source.html()))) return True def _retrieveURL(self, url): req = urllib2.Request(url, None, { 'User-Agent': 'Mozilla/5.0 (compatible; Anki/%s)' % aqt.appVersion }) filecontents = urllib2.urlopen(req).read() path = os.path.join(self.tmpDir(), os.path.basename(url)) file = open(path, "wb") file.write(filecontents) file.close() return path def tmpDir(self): if not self._tmpDir: self._tmpDir = tempfile.mkdtemp(prefix="anki") return self._tmpDir def simplifyHTML(self, html): "Remove all style information and P tags." # fixme if not aqt.mw.config['stripHTML']: return html html = re.sub("\n", " ", html) html = re.sub("
", "\n", html) html = re.sub("

", "\n\n", html) html = re.sub('', "", html) html = stripHTML(html) html = html.replace("\n", "
") html = html.strip() return html def focusOutEvent(self, evt): QTextEdit.focusOutEvent(self, evt) self.parent.lastFocusedEdit = self self.parent.resetFormatButtons() self.parent.disableButtons() if aqt.mw.config['preserveKeyboard'] and sys.platform.startswith("win32"): self._ownLayout = GetKeyboardLayout(0) ActivateKeyboardLayout(self._programLayout, 0) self.emit(SIGNAL("lostFocus")) def focusInEvent(self, evt): if (self.parent.lastFocusedEdit and self.parent.lastFocusedEdit is not self): # remove selection from previous widget try: cur = self.parent.lastFocusedEdit.textCursor() cur.clearSelection() self.parent.lastFocusedEdit.setTextCursor(cur) except RuntimeError: # old widget was deleted pass self.lastFocusedEdit = None QTextEdit.focusInEvent(self, evt) self.parent.formatChanged(None) self.parent.enableButtons() if aqt.mw.config['preserveKeyboard'] and sys.platform.startswith("win32"): self._programLayout = GetKeyboardLayout(0) if self._ownLayout == None: self._ownLayout = self._programLayout ActivateKeyboardLayout(self._ownLayout, 0) class ColourPopup(QDialog): def __init__(self, parent): QDialog.__init__(self, parent, Qt.FramelessWindowHint) def event(self, evt): if evt.type() == QEvent.WindowDeactivate: self.close() return QDialog.event(self, evt)