2012-12-21 08:51:59 +01:00
|
|
|
# -*- coding: utf-8 -*-
|
|
|
|
# Copyright: Damien Elmes <anki@ichi2.net>
|
|
|
|
# License: GNU AGPL, version 3 or later; http://www.gnu.org/licenses/agpl.html
|
2013-11-13 06:56:37 +01:00
|
|
|
import re
|
|
|
|
import os
|
|
|
|
import urllib2
|
|
|
|
import ctypes
|
|
|
|
import urllib
|
2012-12-21 08:51:59 +01:00
|
|
|
|
2013-11-13 06:56:37 +01:00
|
|
|
from anki.lang import _
|
2012-12-21 08:51:59 +01:00
|
|
|
from aqt.qt import *
|
2013-01-14 22:25:21 +01:00
|
|
|
from anki.utils import stripHTML, isWin, isMac, namedtmp, json, stripHTMLMedia
|
2013-07-11 10:21:16 +02:00
|
|
|
import anki.sound
|
2012-12-21 08:51:59 +01:00
|
|
|
from anki.hooks import runHook, runFilter
|
|
|
|
from aqt.sound import getAudio
|
|
|
|
from aqt.webview import AnkiWebView
|
|
|
|
from aqt.utils import shortcut, showInfo, showWarning, getBase, getFile, \
|
2015-09-28 15:09:30 +02:00
|
|
|
openHelp, tooltip, downArrow
|
2012-12-21 08:51:59 +01:00
|
|
|
import aqt
|
|
|
|
import anki.js
|
|
|
|
from BeautifulSoup import BeautifulSoup
|
|
|
|
|
2015-03-14 19:00:00 +01:00
|
|
|
pics = ("jpg", "jpeg", "png", "tif", "tiff", "gif", "svg", "webp")
|
|
|
|
audio = ("wav", "mp3", "ogg", "flac", "mp4", "swf", "mov", "mpeg", "mkv", "m4a", "3gp", "spx", "oga")
|
2012-12-21 08:51:59 +01:00
|
|
|
|
|
|
|
_html = """
|
|
|
|
<html><head>%s<style>
|
|
|
|
.field {
|
|
|
|
border: 1px solid #aaa; background:#fff; color:#000; padding: 5px;
|
|
|
|
}
|
|
|
|
/* prevent floated images from being displayed outside field */
|
|
|
|
.field:after {
|
2013-03-03 01:07:35 +01:00
|
|
|
content: "";
|
2012-12-21 08:51:59 +01:00
|
|
|
display: block;
|
|
|
|
height: 0;
|
|
|
|
clear: both;
|
|
|
|
visibility: hidden;
|
|
|
|
}
|
|
|
|
.fname { vertical-align: middle; padding: 0; }
|
|
|
|
img { max-width: 90%%; }
|
|
|
|
body { margin: 5px; }
|
|
|
|
</style><script>
|
|
|
|
%s
|
|
|
|
|
|
|
|
var currentField = null;
|
|
|
|
var changeTimer = null;
|
|
|
|
var dropTarget = null;
|
|
|
|
|
|
|
|
String.prototype.format = function() {
|
|
|
|
var args = arguments;
|
|
|
|
return this.replace(/\{\d+\}/g, function(m){
|
|
|
|
return args[m.match(/\d+/)]; });
|
|
|
|
};
|
|
|
|
|
|
|
|
function onKey() {
|
|
|
|
// esc clears focus, allowing dialog to close
|
|
|
|
if (window.event.which == 27) {
|
|
|
|
currentField.blur();
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
clearChangeTimer();
|
|
|
|
if (currentField.innerHTML == "<div><br></div>") {
|
|
|
|
// fix empty div bug. slight flicker, but must be done in a timer
|
|
|
|
changeTimer = setTimeout(function () {
|
|
|
|
currentField.innerHTML = "<br>";
|
|
|
|
sendState();
|
|
|
|
saveField("key"); }, 1);
|
|
|
|
} else {
|
|
|
|
changeTimer = setTimeout(function () {
|
|
|
|
sendState();
|
|
|
|
saveField("key"); }, 600);
|
|
|
|
}
|
|
|
|
};
|
|
|
|
|
|
|
|
function sendState() {
|
|
|
|
var r = {
|
|
|
|
'bold': document.queryCommandState("bold"),
|
|
|
|
'italic': document.queryCommandState("italic"),
|
|
|
|
'under': document.queryCommandState("underline"),
|
|
|
|
'super': document.queryCommandState("superscript"),
|
|
|
|
'sub': document.queryCommandState("subscript"),
|
|
|
|
'col': document.queryCommandValue("forecolor")
|
|
|
|
};
|
|
|
|
py.run("state:" + JSON.stringify(r));
|
|
|
|
};
|
|
|
|
|
|
|
|
function setFormat(cmd, arg, nosave) {
|
|
|
|
document.execCommand(cmd, false, arg);
|
|
|
|
if (!nosave) {
|
|
|
|
saveField('key');
|
|
|
|
}
|
|
|
|
};
|
|
|
|
|
|
|
|
function clearChangeTimer() {
|
|
|
|
if (changeTimer) {
|
|
|
|
clearTimeout(changeTimer);
|
|
|
|
changeTimer = null;
|
|
|
|
}
|
|
|
|
};
|
|
|
|
|
|
|
|
function onFocus(elem) {
|
|
|
|
currentField = elem;
|
|
|
|
py.run("focus:" + currentField.id.substring(1));
|
|
|
|
// don't adjust cursor on mouse clicks
|
|
|
|
if (mouseDown) { return; }
|
|
|
|
// do this twice so that there's no flicker on newer versions
|
|
|
|
caretToEnd();
|
|
|
|
// need to do this in a timeout for older qt versions
|
|
|
|
setTimeout(function () { caretToEnd() }, 1);
|
|
|
|
// scroll if bottom of element off the screen
|
|
|
|
function pos(obj) {
|
|
|
|
var cur = 0;
|
|
|
|
do {
|
|
|
|
cur += obj.offsetTop;
|
|
|
|
} while (obj = obj.offsetParent);
|
|
|
|
return cur;
|
|
|
|
}
|
|
|
|
var y = pos(elem);
|
|
|
|
if ((window.pageYOffset+window.innerHeight) < (y+elem.offsetHeight) ||
|
|
|
|
window.pageYOffset > y) {
|
|
|
|
window.scroll(0,y+elem.offsetHeight-window.innerHeight);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
function focusField(n) {
|
|
|
|
$("#f"+n).focus();
|
|
|
|
}
|
|
|
|
|
|
|
|
function onDragOver(elem) {
|
|
|
|
// if we focus the target element immediately, the drag&drop turns into a
|
|
|
|
// copy, so note it down for later instead
|
|
|
|
dropTarget = elem;
|
|
|
|
}
|
|
|
|
|
|
|
|
function caretToEnd() {
|
|
|
|
var r = document.createRange()
|
|
|
|
r.selectNodeContents(currentField);
|
|
|
|
r.collapse(false);
|
|
|
|
var s = document.getSelection();
|
|
|
|
s.removeAllRanges();
|
|
|
|
s.addRange(r);
|
|
|
|
};
|
|
|
|
|
|
|
|
function onBlur() {
|
|
|
|
if (currentField) {
|
|
|
|
saveField("blur");
|
|
|
|
}
|
|
|
|
clearChangeTimer();
|
|
|
|
// if we lose focus, assume the last field is still targeted
|
|
|
|
//currentField = null;
|
|
|
|
};
|
|
|
|
|
|
|
|
function saveField(type) {
|
|
|
|
if (!currentField) {
|
|
|
|
// no field has been focused yet
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
// type is either 'blur' or 'key'
|
|
|
|
py.run(type + ":" + currentField.innerHTML);
|
|
|
|
clearChangeTimer();
|
|
|
|
};
|
|
|
|
|
|
|
|
function wrappedExceptForWhitespace(text, front, back) {
|
|
|
|
var match = text.match(/^(\s*)([^]*?)(\s*)$/);
|
|
|
|
return match[1] + front + match[2] + back + match[3];
|
|
|
|
};
|
|
|
|
|
|
|
|
function wrap(front, back) {
|
|
|
|
var s = window.getSelection();
|
|
|
|
var r = s.getRangeAt(0);
|
|
|
|
var content = r.cloneContents();
|
|
|
|
var span = document.createElement("span")
|
|
|
|
span.appendChild(content);
|
|
|
|
var new_ = wrappedExceptForWhitespace(span.innerHTML, front, back);
|
|
|
|
setFormat("inserthtml", new_);
|
|
|
|
if (!span.innerHTML) {
|
|
|
|
// run with an empty selection; move cursor back past postfix
|
|
|
|
r = s.getRangeAt(0);
|
|
|
|
r.setStart(r.startContainer, r.startOffset - back.length);
|
|
|
|
r.collapse(true);
|
|
|
|
s.removeAllRanges();
|
|
|
|
s.addRange(r);
|
|
|
|
}
|
|
|
|
};
|
|
|
|
|
|
|
|
function setFields(fields, focusTo) {
|
|
|
|
var txt = "";
|
|
|
|
for (var i=0; i<fields.length; i++) {
|
|
|
|
var n = fields[i][0];
|
|
|
|
var f = fields[i][1];
|
|
|
|
if (!f) {
|
|
|
|
f = "<br>";
|
|
|
|
}
|
|
|
|
txt += "<tr><td class=fname>{0}</td></tr><tr><td width=100%%>".format(n);
|
|
|
|
txt += "<div id=f{0} onkeydown='onKey();' onmouseup='onKey();'".format(i);
|
|
|
|
txt += " onfocus='onFocus(this);' onblur='onBlur();' class=field ";
|
2013-06-14 06:11:48 +02:00
|
|
|
txt += "ondragover='onDragOver(this);' ";
|
2012-12-21 08:51:59 +01:00
|
|
|
txt += "contentEditable=true class=field>{0}</div>".format(f);
|
|
|
|
txt += "</td></tr>";
|
|
|
|
}
|
|
|
|
$("#fields").html("<table cellpadding=0 width=100%%>"+txt+"</table>");
|
|
|
|
if (!focusTo) {
|
|
|
|
focusTo = 0;
|
|
|
|
}
|
|
|
|
if (focusTo >= 0) {
|
|
|
|
$("#f"+focusTo).focus();
|
|
|
|
}
|
|
|
|
};
|
|
|
|
|
|
|
|
function setBackgrounds(cols) {
|
|
|
|
for (var i=0; i<cols.length; i++) {
|
|
|
|
$("#f"+i).css("background", cols[i]);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
function setFonts(fonts) {
|
|
|
|
for (var i=0; i<fonts.length; i++) {
|
|
|
|
$("#f"+i).css("font-family", fonts[i][0]);
|
|
|
|
$("#f"+i).css("font-size", fonts[i][1]);
|
|
|
|
$("#f"+i)[0].dir = fonts[i][2] ? "rtl" : "ltr";
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
function showDupes() {
|
|
|
|
$("#dupes").show();
|
|
|
|
}
|
|
|
|
|
|
|
|
function hideDupes() {
|
|
|
|
$("#dupes").hide();
|
|
|
|
}
|
|
|
|
|
|
|
|
var mouseDown = 0;
|
|
|
|
|
|
|
|
$(function () {
|
|
|
|
document.body.onmousedown = function () {
|
|
|
|
mouseDown++;
|
|
|
|
}
|
2013-06-14 06:11:48 +02:00
|
|
|
|
2012-12-21 08:51:59 +01:00
|
|
|
document.body.onmouseup = function () {
|
|
|
|
mouseDown--;
|
|
|
|
}
|
2013-06-14 06:11:48 +02:00
|
|
|
|
2012-12-21 08:51:59 +01:00
|
|
|
document.onclick = function (evt) {
|
|
|
|
var src = window.event.srcElement;
|
|
|
|
if (src.tagName == "IMG") {
|
|
|
|
// image clicked; find contenteditable parent
|
|
|
|
var p = src;
|
|
|
|
while (p = p.parentNode) {
|
|
|
|
if (p.className == "field") {
|
|
|
|
$("#"+p.id).focus();
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
2013-06-14 06:11:48 +02:00
|
|
|
|
2012-12-21 08:51:59 +01:00
|
|
|
});
|
|
|
|
|
|
|
|
</script></head><body>
|
|
|
|
<div id="fields"></div>
|
|
|
|
<div id="dupes"><a href="#" onclick="py.run('dupes');return false;">%s</a></div>
|
|
|
|
</body></html>
|
|
|
|
"""
|
|
|
|
|
|
|
|
# caller is responsible for resetting note on reset
|
|
|
|
class Editor(object):
|
|
|
|
def __init__(self, mw, widget, parentWindow, addMode=False):
|
|
|
|
self.mw = mw
|
|
|
|
self.widget = widget
|
|
|
|
self.parentWindow = parentWindow
|
|
|
|
self.note = None
|
|
|
|
self.stealFocus = True
|
|
|
|
self.addMode = addMode
|
|
|
|
self._loaded = False
|
|
|
|
self.currentField = 0
|
|
|
|
# current card, for card layout
|
|
|
|
self.card = None
|
|
|
|
self.setupOuter()
|
|
|
|
self.setupButtons()
|
|
|
|
self.setupWeb()
|
|
|
|
self.setupTags()
|
|
|
|
self.setupKeyboard()
|
|
|
|
|
|
|
|
# Initial setup
|
|
|
|
############################################################
|
|
|
|
|
|
|
|
def setupOuter(self):
|
|
|
|
l = QVBoxLayout()
|
|
|
|
l.setMargin(0)
|
|
|
|
l.setSpacing(0)
|
|
|
|
self.widget.setLayout(l)
|
|
|
|
self.outerLayout = l
|
|
|
|
|
|
|
|
def setupWeb(self):
|
|
|
|
self.web = EditorWebView(self.widget, self)
|
|
|
|
self.web.allowDrops = True
|
|
|
|
self.web.setBridge(self.bridge)
|
|
|
|
self.outerLayout.addWidget(self.web, 1)
|
|
|
|
# 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)
|
|
|
|
|
|
|
|
# Top buttons
|
|
|
|
######################################################################
|
|
|
|
|
|
|
|
def _addButton(self, name, func, key=None, tip=None, size=True, text="",
|
|
|
|
check=False, native=False, canDisable=True):
|
|
|
|
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)
|
|
|
|
if not native:
|
2013-11-13 06:56:37 +01:00
|
|
|
if self.plastiqueStyle:
|
|
|
|
b.setStyle(self.plastiqueStyle)
|
2012-12-21 08:51:59 +01:00
|
|
|
b.setFocusPolicy(Qt.NoFocus)
|
|
|
|
else:
|
|
|
|
b.setAutoDefault(False)
|
|
|
|
if not text:
|
|
|
|
b.setIcon(QIcon(":/icons/%s.png" % name))
|
|
|
|
if key:
|
|
|
|
b.setShortcut(QKeySequence(key))
|
|
|
|
if tip:
|
|
|
|
b.setToolTip(shortcut(tip))
|
|
|
|
if check:
|
|
|
|
b.setCheckable(True)
|
|
|
|
self.iconsBox.addWidget(b)
|
|
|
|
if canDisable:
|
|
|
|
self._buttons[name] = b
|
|
|
|
return b
|
|
|
|
|
|
|
|
def setupButtons(self):
|
|
|
|
self._buttons = {}
|
|
|
|
# button styles for mac
|
2013-11-13 06:56:37 +01:00
|
|
|
if not isMac:
|
|
|
|
self.plastiqueStyle = QStyleFactory.create("plastique")
|
|
|
|
if not self.plastiqueStyle:
|
|
|
|
# plastique was removed in qt5
|
|
|
|
self.plastiqueStyle = QStyleFactory.create("fusion")
|
|
|
|
self.widget.setStyle(self.plastiqueStyle)
|
|
|
|
else:
|
|
|
|
self.plastiqueStyle = None
|
2012-12-21 08:51:59 +01:00
|
|
|
# icons
|
|
|
|
self.iconsBox = QHBoxLayout()
|
|
|
|
if not isMac:
|
|
|
|
self.iconsBox.setMargin(6)
|
2013-11-13 06:56:37 +01:00
|
|
|
self.iconsBox.setSpacing(0)
|
2012-12-21 08:51:59 +01:00
|
|
|
else:
|
|
|
|
self.iconsBox.setMargin(0)
|
2013-11-13 06:56:37 +01:00
|
|
|
self.iconsBox.setSpacing(14)
|
2012-12-21 08:51:59 +01:00
|
|
|
self.outerLayout.addLayout(self.iconsBox)
|
|
|
|
b = self._addButton
|
|
|
|
b("fields", self.onFields, "",
|
|
|
|
shortcut(_("Customize Fields")), size=False, text=_("Fields..."),
|
|
|
|
native=True, canDisable=False)
|
|
|
|
self.iconsBox.addItem(QSpacerItem(6,1, QSizePolicy.Fixed))
|
|
|
|
b("layout", self.onCardLayout, _("Ctrl+L"),
|
|
|
|
shortcut(_("Customize Cards (Ctrl+L)")),
|
|
|
|
size=False, text=_("Cards..."), native=True, canDisable=False)
|
|
|
|
# align to right
|
|
|
|
self.iconsBox.addItem(QSpacerItem(20,1, QSizePolicy.Expanding))
|
|
|
|
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)
|
2013-04-11 07:26:09 +02:00
|
|
|
b("text_super", self.toggleSuper, _("Ctrl+Shift+="),
|
|
|
|
_("Superscript (Ctrl+Shift+=)"), check=True)
|
|
|
|
b("text_sub", self.toggleSub, _("Ctrl+="),
|
|
|
|
_("Subscript (Ctrl+=)"), check=True)
|
2012-12-21 08:51:59 +01:00
|
|
|
b("text_clear", self.removeFormat, _("Ctrl+R"),
|
|
|
|
_("Remove formatting (Ctrl+R)"))
|
|
|
|
but = b("foreground", self.onForeground, _("F7"), text=" ")
|
|
|
|
but.setToolTip(_("Set foreground colour (F7)"))
|
|
|
|
self.setupForegroundButton(but)
|
|
|
|
but = b("change_colour", self.onChangeCol, _("F8"),
|
2015-09-28 15:09:30 +02:00
|
|
|
_("Change colour (F8)"), text=downArrow())
|
2012-12-21 08:51:59 +01:00
|
|
|
but.setFixedWidth(12)
|
|
|
|
but = b("cloze", self.onCloze, _("Ctrl+Shift+C"),
|
|
|
|
_("Cloze deletion (Ctrl+Shift+C)"), text="[...]")
|
|
|
|
but.setFixedWidth(24)
|
|
|
|
s = self.clozeShortcut2 = QShortcut(
|
|
|
|
QKeySequence(_("Ctrl+Alt+Shift+C")), self.parentWindow)
|
|
|
|
s.connect(s, SIGNAL("activated()"), self.onCloze)
|
|
|
|
# fixme: better image names
|
|
|
|
b("mail-attachment", self.onAddMedia, _("F3"),
|
|
|
|
_("Attach pictures/audio/video (F3)"))
|
|
|
|
b("media-record", self.onRecSound, _("F5"), _("Record audio (F5)"))
|
2015-09-28 15:09:30 +02:00
|
|
|
b("adv", self.onAdvanced, text=downArrow())
|
2012-12-21 08:51:59 +01:00
|
|
|
s = QShortcut(QKeySequence("Ctrl+T, T"), self.widget)
|
|
|
|
s.connect(s, SIGNAL("activated()"), self.insertLatex)
|
|
|
|
s = QShortcut(QKeySequence("Ctrl+T, E"), self.widget)
|
|
|
|
s.connect(s, SIGNAL("activated()"), self.insertLatexEqn)
|
|
|
|
s = QShortcut(QKeySequence("Ctrl+T, M"), self.widget)
|
|
|
|
s.connect(s, SIGNAL("activated()"), self.insertLatexMathEnv)
|
|
|
|
s = QShortcut(QKeySequence("Ctrl+Shift+X"), self.widget)
|
|
|
|
s.connect(s, SIGNAL("activated()"), self.onHtmlEdit)
|
2013-05-17 07:32:50 +02:00
|
|
|
# tags
|
|
|
|
s = QShortcut(QKeySequence("Ctrl+Shift+T"), self.widget)
|
|
|
|
s.connect(s, SIGNAL("activated()"), lambda: self.tags.setFocus())
|
2012-12-21 08:51:59 +01:00
|
|
|
runHook("setupEditorButtons", self)
|
|
|
|
|
|
|
|
def enableButtons(self, val=True):
|
|
|
|
for b in self._buttons.values():
|
|
|
|
b.setEnabled(val)
|
|
|
|
|
|
|
|
def disableButtons(self):
|
|
|
|
self.enableButtons(False)
|
|
|
|
|
|
|
|
def onFields(self):
|
|
|
|
from aqt.fields import FieldDialog
|
|
|
|
self.saveNow()
|
|
|
|
FieldDialog(self.mw, self.note, parent=self.parentWindow)
|
|
|
|
|
|
|
|
def onCardLayout(self):
|
|
|
|
from aqt.clayout import CardLayout
|
|
|
|
self.saveNow()
|
|
|
|
if self.card:
|
|
|
|
ord = self.card.ord
|
|
|
|
else:
|
|
|
|
ord = 0
|
2013-07-23 15:35:00 +02:00
|
|
|
# passing parentWindow leads to crash on windows at the moment
|
|
|
|
if isWin:
|
|
|
|
parent=None
|
|
|
|
else:
|
|
|
|
parent=self.parentWindow
|
|
|
|
CardLayout(self.mw, self.note, ord=ord, parent=parent,
|
2012-12-21 08:51:59 +01:00
|
|
|
addMode=self.addMode)
|
|
|
|
self.loadNote()
|
2013-07-23 15:35:00 +02:00
|
|
|
if isWin:
|
|
|
|
self.parentWindow.activateWindow()
|
2012-12-21 08:51:59 +01:00
|
|
|
|
|
|
|
# JS->Python bridge
|
|
|
|
######################################################################
|
|
|
|
|
|
|
|
def bridge(self, str):
|
|
|
|
if not self.note or not runHook:
|
|
|
|
# shutdown
|
|
|
|
return
|
|
|
|
# focus lost or key/button pressed?
|
|
|
|
if str.startswith("blur") or str.startswith("key"):
|
|
|
|
(type, txt) = str.split(":", 1)
|
|
|
|
txt = self.mungeHTML(txt)
|
|
|
|
# misbehaving apps may include a null byte in the text
|
|
|
|
txt = txt.replace("\x00", "")
|
2014-08-01 02:37:23 +02:00
|
|
|
# reverse the url quoting we added to get images to display
|
2014-08-01 02:42:28 +02:00
|
|
|
txt = self.mw.col.media.escapeImages(txt, unescape=True)
|
2012-12-21 08:51:59 +01:00
|
|
|
self.note.fields[self.currentField] = txt
|
|
|
|
if not self.addMode:
|
|
|
|
self.note.flush()
|
|
|
|
self.mw.requireReset()
|
|
|
|
if type == "blur":
|
|
|
|
self.disableButtons()
|
|
|
|
# run any filters
|
|
|
|
if runFilter(
|
|
|
|
"editFocusLost", False, self.note, self.currentField):
|
|
|
|
# something updated the note; schedule reload
|
|
|
|
def onUpdate():
|
Avoid calling checkValid on an empty note
Fix error message when editFocusLost is called after editor destruction
https://anki.tenderapp.com/discussions/ankidesktop/11175-bug-in-browser-revealed-by-add-on?unresolve=true
To reproduce:
1. Install the attached add-on. (All it does is register a function on the editFocusLost hook, and set the flag to 1 to signify that a field has been updated).
2. Run Anki. Enter the Browse window. Press Enter to display your deck.
3. Select one card.
4. Click inside one of the fields for that card. Now, when you leave that field, the editFocusLost hook will be called.
5. Now, select 2 notes from the browser, with your mouse, without clicking anywhere else before.
You will observe this error message:
Traceback (most recent call last):
File "aqt/progress.pyc", line 69, in handler
File "aqt/editor.pyc", line 467, in onUpdate
File "aqt/editor.pyc", line 572, in checkValid AttributeError: 'NoneType' object has no attribute 'fields'
2014-12-31 15:36:12 +01:00
|
|
|
if not self.note:
|
|
|
|
return
|
2013-01-30 09:00:17 +01:00
|
|
|
self.stealFocus = True
|
2012-12-21 08:51:59 +01:00
|
|
|
self.loadNote()
|
|
|
|
self.checkValid()
|
|
|
|
self.mw.progress.timer(100, onUpdate, False)
|
|
|
|
else:
|
|
|
|
self.checkValid()
|
|
|
|
else:
|
|
|
|
runHook("editTimer", self.note)
|
|
|
|
self.checkValid()
|
|
|
|
# focused into field?
|
|
|
|
elif str.startswith("focus"):
|
|
|
|
(type, num) = str.split(":", 1)
|
|
|
|
self.enableButtons()
|
|
|
|
self.currentField = int(num)
|
2013-05-21 00:19:43 +02:00
|
|
|
runHook("editFocusGained", self.note, self.currentField)
|
2012-12-21 08:51:59 +01:00
|
|
|
# state buttons changed?
|
|
|
|
elif str.startswith("state"):
|
|
|
|
(cmd, txt) = str.split(":", 1)
|
|
|
|
r = json.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'])
|
|
|
|
elif str.startswith("dupes"):
|
|
|
|
self.showDupes()
|
|
|
|
else:
|
|
|
|
print str
|
|
|
|
|
|
|
|
def mungeHTML(self, txt):
|
|
|
|
if txt == "<br>":
|
|
|
|
txt = ""
|
2013-07-11 10:21:16 +02:00
|
|
|
return self._filterHTML(txt, localize=False)
|
2012-12-21 08:51:59 +01:00
|
|
|
|
|
|
|
# Setting/unsetting the current note
|
|
|
|
######################################################################
|
|
|
|
|
|
|
|
def _loadFinished(self, w):
|
|
|
|
self._loaded = True
|
|
|
|
if self.note:
|
|
|
|
self.loadNote()
|
|
|
|
|
|
|
|
def setNote(self, note, hide=True, focus=False):
|
|
|
|
"Make NOTE the current note."
|
|
|
|
self.note = note
|
|
|
|
self.currentField = 0
|
2013-01-08 03:01:32 +01:00
|
|
|
self.disableButtons()
|
2013-05-22 05:54:29 +02:00
|
|
|
if focus:
|
|
|
|
self.stealFocus = True
|
2012-12-21 08:51:59 +01:00
|
|
|
# change timer
|
|
|
|
if self.note:
|
|
|
|
self.web.setHtml(_html % (
|
|
|
|
getBase(self.mw.col), anki.js.jquery,
|
|
|
|
_("Show Duplicates")), loadCB=self._loadFinished)
|
|
|
|
self.updateTags()
|
|
|
|
self.updateKeyboard()
|
|
|
|
else:
|
|
|
|
self.hideCompleters()
|
|
|
|
if hide:
|
|
|
|
self.widget.hide()
|
|
|
|
|
|
|
|
def loadNote(self):
|
|
|
|
if not self.note:
|
|
|
|
return
|
|
|
|
if self.stealFocus:
|
|
|
|
field = self.currentField
|
|
|
|
else:
|
|
|
|
field = -1
|
|
|
|
if not self._loaded:
|
|
|
|
# will be loaded when page is ready
|
|
|
|
return
|
|
|
|
data = []
|
|
|
|
for fld, val in self.note.items():
|
|
|
|
data.append((fld, self.mw.col.media.escapeImages(val)))
|
|
|
|
self.web.eval("setFields(%s, %d);" % (
|
|
|
|
json.dumps(data), field))
|
|
|
|
self.web.eval("setFonts(%s);" % (
|
|
|
|
json.dumps(self.fonts())))
|
|
|
|
self.checkValid()
|
|
|
|
self.widget.show()
|
|
|
|
if self.stealFocus:
|
|
|
|
self.web.setFocus()
|
2013-05-22 05:54:29 +02:00
|
|
|
self.stealFocus = False
|
2012-12-21 08:51:59 +01:00
|
|
|
|
|
|
|
def focus(self):
|
|
|
|
self.web.setFocus()
|
|
|
|
|
|
|
|
def fonts(self):
|
|
|
|
return [(f['font'], f['size'], f['rtl'])
|
|
|
|
for f in self.note.model()['flds']]
|
|
|
|
|
|
|
|
def saveNow(self):
|
|
|
|
"Must call this before adding cards, closing dialog, etc."
|
|
|
|
if not self.note:
|
|
|
|
return
|
|
|
|
self.saveTags()
|
|
|
|
if self.mw.app.focusWidget() != self.web:
|
|
|
|
# if no fields are focused, there's nothing to save
|
|
|
|
return
|
|
|
|
# move focus out of fields and save tags
|
|
|
|
self.parentWindow.setFocus()
|
|
|
|
# and process events so any focus-lost hooks fire
|
|
|
|
self.mw.app.processEvents()
|
|
|
|
|
|
|
|
def checkValid(self):
|
|
|
|
cols = []
|
|
|
|
err = None
|
|
|
|
for f in self.note.fields:
|
|
|
|
cols.append("#fff")
|
|
|
|
err = self.note.dupeOrEmpty()
|
|
|
|
if err == 2:
|
|
|
|
cols[0] = "#fcc"
|
|
|
|
self.web.eval("showDupes();")
|
|
|
|
else:
|
|
|
|
self.web.eval("hideDupes();")
|
|
|
|
self.web.eval("setBackgrounds(%s);" % json.dumps(cols))
|
|
|
|
|
|
|
|
def showDupes(self):
|
2013-01-14 22:25:21 +01:00
|
|
|
contents = stripHTMLMedia(self.note.fields[0])
|
2012-12-21 08:51:59 +01:00
|
|
|
browser = aqt.dialogs.open("Browser", self.mw)
|
|
|
|
browser.form.searchEdit.lineEdit().setText(
|
2013-05-16 07:17:07 +02:00
|
|
|
'"dupe:%s,%s"' % (self.note.model()['id'],
|
|
|
|
contents))
|
2012-12-21 08:51:59 +01:00
|
|
|
browser.onSearch()
|
|
|
|
|
|
|
|
def fieldsAreBlank(self):
|
|
|
|
if not self.note:
|
|
|
|
return True
|
2013-05-24 05:04:28 +02:00
|
|
|
m = self.note.model()
|
|
|
|
for c, f in enumerate(self.note.fields):
|
|
|
|
if f and not m['flds'][c]['sticky']:
|
2012-12-21 08:51:59 +01:00
|
|
|
return False
|
|
|
|
return True
|
|
|
|
|
|
|
|
# HTML editing
|
|
|
|
######################################################################
|
|
|
|
|
|
|
|
def onHtmlEdit(self):
|
|
|
|
self.saveNow()
|
|
|
|
d = QDialog(self.widget)
|
|
|
|
form = aqt.forms.edithtml.Ui_Dialog()
|
|
|
|
form.setupUi(d)
|
|
|
|
d.connect(form.buttonBox, SIGNAL("helpRequested()"),
|
|
|
|
lambda: openHelp("editor"))
|
|
|
|
form.textEdit.setPlainText(self.note.fields[self.currentField])
|
|
|
|
form.textEdit.moveCursor(QTextCursor.End)
|
|
|
|
d.exec_()
|
|
|
|
html = form.textEdit.toPlainText()
|
|
|
|
# filter html through beautifulsoup so we can strip out things like a
|
|
|
|
# leading </div>
|
|
|
|
html = unicode(BeautifulSoup(html))
|
|
|
|
self.note.fields[self.currentField] = html
|
|
|
|
self.loadNote()
|
|
|
|
# focus field so it's saved
|
|
|
|
self.web.setFocus()
|
|
|
|
self.web.eval("focusField(%d);" % self.currentField)
|
|
|
|
|
|
|
|
# Tag handling
|
|
|
|
######################################################################
|
|
|
|
|
|
|
|
def setupTags(self):
|
|
|
|
import aqt.tagedit
|
|
|
|
g = QGroupBox(self.widget)
|
|
|
|
g.setFlat(True)
|
|
|
|
tb = QGridLayout()
|
|
|
|
tb.setSpacing(12)
|
|
|
|
tb.setMargin(6)
|
|
|
|
# tags
|
|
|
|
l = QLabel(_("Tags"))
|
|
|
|
tb.addWidget(l, 1, 0)
|
|
|
|
self.tags = aqt.tagedit.TagEdit(self.widget)
|
|
|
|
self.tags.connect(self.tags, SIGNAL("lostFocus"),
|
|
|
|
self.saveTags)
|
2013-05-17 07:32:50 +02:00
|
|
|
self.tags.setToolTip(shortcut(_("Jump to tags with Ctrl+Shift+T")))
|
2012-12-21 08:51:59 +01:00
|
|
|
tb.addWidget(self.tags, 1, 1)
|
|
|
|
g.setLayout(tb)
|
|
|
|
self.outerLayout.addWidget(g)
|
|
|
|
|
|
|
|
def updateTags(self):
|
|
|
|
if self.tags.col != self.mw.col:
|
|
|
|
self.tags.setCol(self.mw.col)
|
|
|
|
if not self.tags.text() or not self.addMode:
|
|
|
|
self.tags.setText(self.note.stringTags().strip())
|
|
|
|
|
|
|
|
def saveTags(self):
|
|
|
|
if not self.note:
|
|
|
|
return
|
2013-05-17 06:51:49 +02:00
|
|
|
self.note.tags = self.mw.col.tags.canonify(
|
|
|
|
self.mw.col.tags.split(self.tags.text()))
|
|
|
|
self.tags.setText(self.mw.col.tags.join(self.note.tags).strip())
|
2012-12-21 08:51:59 +01:00
|
|
|
if not self.addMode:
|
|
|
|
self.note.flush()
|
|
|
|
runHook("tagsUpdated", self.note)
|
|
|
|
|
|
|
|
def saveAddModeVars(self):
|
|
|
|
if self.addMode:
|
|
|
|
# save tags to model
|
|
|
|
m = self.note.model()
|
|
|
|
m['tags'] = self.note.tags
|
|
|
|
self.mw.col.models.save(m)
|
|
|
|
|
|
|
|
def hideCompleters(self):
|
|
|
|
self.tags.hideCompleter()
|
|
|
|
|
|
|
|
# Format buttons
|
|
|
|
######################################################################
|
|
|
|
|
|
|
|
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 removeFormat(self):
|
|
|
|
self.web.eval("setFormat('removeFormat');")
|
|
|
|
|
|
|
|
def onCloze(self):
|
|
|
|
# check that the model is set up for cloze deletion
|
2014-03-09 19:12:39 +01:00
|
|
|
if not re.search('{{(.*:)*cloze:',self.note.model()['tmpls'][0]['qfmt']):
|
2012-12-21 08:51:59 +01:00
|
|
|
if self.addMode:
|
2013-05-17 08:57:22 +02:00
|
|
|
tooltip(_("Warning, cloze deletions will not work until "
|
|
|
|
"you switch the type at the top to Cloze."))
|
2012-12-21 08:51:59 +01:00
|
|
|
else:
|
|
|
|
showInfo(_("""\
|
|
|
|
To make a cloze deletion on an existing note, you need to change it \
|
|
|
|
to a cloze type first, via Edit>Change Note Type."""))
|
2013-05-17 08:57:22 +02:00
|
|
|
return
|
2012-12-21 08:51:59 +01:00
|
|
|
# find the highest existing cloze
|
|
|
|
highest = 0
|
|
|
|
for name, val in self.note.items():
|
|
|
|
m = re.findall("\{\{c(\d+)::", val)
|
|
|
|
if m:
|
|
|
|
highest = max(highest, sorted([int(x) for x in m])[-1])
|
|
|
|
# reuse last?
|
|
|
|
if not self.mw.app.keyboardModifiers() & Qt.AltModifier:
|
|
|
|
highest += 1
|
|
|
|
# must start at 1
|
|
|
|
highest = max(1, highest)
|
|
|
|
self.web.eval("wrap('{{c%d::', '}}');" % highest)
|
|
|
|
|
|
|
|
# Foreground colour
|
|
|
|
######################################################################
|
|
|
|
|
|
|
|
def setupForegroundButton(self, but):
|
|
|
|
self.foregroundFrame = QFrame()
|
|
|
|
self.foregroundFrame.setAutoFillBackground(True)
|
|
|
|
self.foregroundFrame.setFocusPolicy(Qt.NoFocus)
|
|
|
|
self.fcolour = self.mw.pm.profile.get("lastColour", "#00f")
|
|
|
|
self.onColourChanged()
|
|
|
|
hbox = QHBoxLayout()
|
|
|
|
hbox.addWidget(self.foregroundFrame)
|
|
|
|
hbox.setMargin(5)
|
|
|
|
but.setLayout(hbox)
|
|
|
|
|
|
|
|
# use last colour
|
|
|
|
def onForeground(self):
|
|
|
|
self._wrapWithColour(self.fcolour)
|
|
|
|
|
|
|
|
# choose new colour
|
|
|
|
def onChangeCol(self):
|
|
|
|
new = QColorDialog.getColor(QColor(self.fcolour), None)
|
|
|
|
# native dialog doesn't refocus us for some reason
|
|
|
|
self.parentWindow.activateWindow()
|
|
|
|
if new.isValid():
|
|
|
|
self.fcolour = new.name()
|
|
|
|
self.onColourChanged()
|
|
|
|
self._wrapWithColour(self.fcolour)
|
|
|
|
|
|
|
|
def _updateForegroundButton(self):
|
|
|
|
self.foregroundFrame.setPalette(QPalette(QColor(self.fcolour)))
|
|
|
|
|
|
|
|
def onColourChanged(self):
|
|
|
|
self._updateForegroundButton()
|
|
|
|
self.mw.pm.profile['lastColour'] = self.fcolour
|
|
|
|
|
|
|
|
def _wrapWithColour(self, colour):
|
|
|
|
self.web.eval("setFormat('forecolor', '%s')" % colour)
|
|
|
|
|
|
|
|
# Audio/video/images
|
|
|
|
######################################################################
|
|
|
|
|
|
|
|
def onAddMedia(self):
|
|
|
|
key = (_("Media") +
|
|
|
|
" (*.jpg *.png *.gif *.tiff *.svg *.tif *.jpeg "+
|
|
|
|
"*.mp3 *.ogg *.wav *.avi *.ogv *.mpg *.mpeg *.mov *.mp4 " +
|
|
|
|
"*.mkv *.ogx *.ogv *.oga *.flv *.swf *.flac)")
|
|
|
|
def accept(file):
|
|
|
|
self.addMedia(file, canDelete=True)
|
|
|
|
file = getFile(self.widget, _("Add Media"), accept, key, key="media")
|
|
|
|
self.parentWindow.activateWindow()
|
|
|
|
|
|
|
|
def addMedia(self, path, canDelete=False):
|
|
|
|
html = self._addMedia(path, canDelete)
|
|
|
|
self.web.eval("setFormat('inserthtml', %s);" % json.dumps(html))
|
|
|
|
|
|
|
|
def _addMedia(self, path, canDelete=False):
|
2013-07-11 10:21:16 +02:00
|
|
|
"Add to media folder and return local img or sound tag."
|
2012-12-21 08:51:59 +01:00
|
|
|
# copy to media folder
|
2013-07-11 10:21:16 +02:00
|
|
|
fname = self.mw.col.media.addFile(path)
|
2012-12-21 08:51:59 +01:00
|
|
|
# remove original?
|
|
|
|
if canDelete and self.mw.pm.profile['deleteMedia']:
|
2013-07-11 10:21:16 +02:00
|
|
|
if os.path.abspath(fname) != os.path.abspath(path):
|
2012-12-21 08:51:59 +01:00
|
|
|
try:
|
|
|
|
os.unlink(path)
|
|
|
|
except:
|
|
|
|
pass
|
|
|
|
# return a local html link
|
2013-07-11 10:21:16 +02:00
|
|
|
return self.fnameToLink(fname)
|
2012-12-21 08:51:59 +01:00
|
|
|
|
|
|
|
def onRecSound(self):
|
|
|
|
try:
|
|
|
|
file = getAudio(self.widget)
|
|
|
|
except Exception, e:
|
|
|
|
showWarning(_(
|
|
|
|
"Couldn't record audio. Have you installed lame and sox?") +
|
2013-10-20 03:57:42 +02:00
|
|
|
"\n\n" + repr(str(e)))
|
2012-12-21 08:51:59 +01:00
|
|
|
return
|
|
|
|
self.addMedia(file)
|
|
|
|
|
2013-07-11 10:21:16 +02:00
|
|
|
# Media downloads
|
|
|
|
######################################################################
|
|
|
|
|
|
|
|
def urlToLink(self, url):
|
|
|
|
fname = self.urlToFile(url)
|
|
|
|
if not fname:
|
|
|
|
return ""
|
|
|
|
return self.fnameToLink(fname)
|
|
|
|
|
|
|
|
def fnameToLink(self, fname):
|
|
|
|
ext = fname.split(".")[-1].lower()
|
|
|
|
if ext in pics:
|
|
|
|
name = urllib.quote(fname.encode("utf8"))
|
|
|
|
return '<img src="%s">' % name
|
|
|
|
else:
|
|
|
|
anki.sound.play(fname)
|
|
|
|
return '[sound:%s]' % fname
|
|
|
|
|
|
|
|
def urlToFile(self, url):
|
|
|
|
l = url.lower()
|
|
|
|
for suffix in pics+audio:
|
|
|
|
if l.endswith(suffix):
|
|
|
|
return self._retrieveURL(url)
|
2013-11-26 09:57:02 +01:00
|
|
|
# not a supported type
|
2013-07-11 10:21:16 +02:00
|
|
|
return
|
|
|
|
|
2013-07-18 13:32:41 +02:00
|
|
|
def isURL(self, s):
|
|
|
|
s = s.lower()
|
|
|
|
return (s.startswith("http://")
|
|
|
|
or s.startswith("https://")
|
2013-09-20 07:41:56 +02:00
|
|
|
or s.startswith("ftp://")
|
|
|
|
or s.startswith("file://"))
|
2013-07-18 13:32:41 +02:00
|
|
|
|
2013-07-11 10:21:16 +02:00
|
|
|
def _retrieveURL(self, url):
|
|
|
|
"Download file into media folder and return local filename or None."
|
2013-09-20 07:41:56 +02:00
|
|
|
# urllib doesn't understand percent-escaped utf8, but requires things like
|
|
|
|
# '#' to be escaped. we don't try to unquote the incoming URL, because
|
|
|
|
# we should only be receiving file:// urls from url mime, which is unquoted
|
2013-07-11 10:21:16 +02:00
|
|
|
if url.lower().startswith("file://"):
|
|
|
|
url = url.replace("%", "%25")
|
|
|
|
url = url.replace("#", "%23")
|
2013-09-20 07:41:56 +02:00
|
|
|
# fetch it into a temporary folder
|
2013-07-11 10:21:16 +02:00
|
|
|
self.mw.progress.start(
|
|
|
|
immediate=True, parent=self.parentWindow)
|
|
|
|
try:
|
|
|
|
req = urllib2.Request(url, None, {
|
|
|
|
'User-Agent': 'Mozilla/5.0 (compatible; Anki)'})
|
|
|
|
filecontents = urllib2.urlopen(req).read()
|
|
|
|
except urllib2.URLError, e:
|
|
|
|
showWarning(_("An error occurred while opening %s") % e)
|
|
|
|
return
|
|
|
|
finally:
|
|
|
|
self.mw.progress.finish()
|
|
|
|
path = unicode(urllib2.unquote(url.encode("utf8")), "utf8")
|
|
|
|
return self.mw.col.media.writeData(path, filecontents)
|
|
|
|
|
|
|
|
# HTML filtering
|
|
|
|
######################################################################
|
|
|
|
|
|
|
|
def _filterHTML(self, html, localize=False):
|
|
|
|
doc = BeautifulSoup(html)
|
|
|
|
# remove implicit regular font style from outermost element
|
|
|
|
if doc.span:
|
|
|
|
try:
|
|
|
|
attrs = doc.span['style'].split(";")
|
|
|
|
except (KeyError, TypeError):
|
|
|
|
attrs = []
|
|
|
|
if attrs:
|
|
|
|
new = []
|
|
|
|
for attr in attrs:
|
|
|
|
sattr = attr.strip()
|
|
|
|
if sattr and sattr not in ("font-style: normal", "font-weight: normal"):
|
|
|
|
new.append(sattr)
|
|
|
|
doc.span['style'] = ";".join(new)
|
|
|
|
# filter out implicit formatting from webkit
|
|
|
|
for tag in doc("span", "Apple-style-span"):
|
|
|
|
preserve = ""
|
|
|
|
for item in tag['style'].split(";"):
|
|
|
|
try:
|
|
|
|
k, v = item.split(":")
|
|
|
|
except ValueError:
|
|
|
|
continue
|
|
|
|
if k.strip() == "color" and not v.strip() == "rgb(0, 0, 0)":
|
|
|
|
preserve += "color:%s;" % v
|
|
|
|
if k.strip() in ("font-weight", "font-style"):
|
|
|
|
preserve += item + ";"
|
|
|
|
if preserve:
|
|
|
|
# preserve colour attribute, delete implicit class
|
|
|
|
tag['style'] = preserve
|
|
|
|
del tag['class']
|
|
|
|
else:
|
|
|
|
# strip completely
|
|
|
|
tag.replaceWithChildren()
|
|
|
|
for tag in doc("font", "Apple-style-span"):
|
|
|
|
# strip all but colour attr from implicit font tags
|
|
|
|
if 'color' in dict(tag.attrs):
|
|
|
|
for attr in tag.attrs:
|
|
|
|
if attr != "color":
|
|
|
|
del tag[attr]
|
|
|
|
# and apple class
|
|
|
|
del tag['class']
|
|
|
|
else:
|
|
|
|
# remove completely
|
|
|
|
tag.replaceWithChildren()
|
|
|
|
# now images
|
|
|
|
for tag in doc("img"):
|
|
|
|
# turn file:/// links into relative ones
|
|
|
|
try:
|
|
|
|
if tag['src'].lower().startswith("file://"):
|
|
|
|
tag['src'] = os.path.basename(tag['src'])
|
2013-07-18 13:32:41 +02:00
|
|
|
if localize and self.isURL(tag['src']):
|
2013-07-11 10:21:16 +02:00
|
|
|
# convert remote image links to local ones
|
|
|
|
fname = self.urlToFile(tag['src'])
|
|
|
|
if fname:
|
|
|
|
tag['src'] = fname
|
|
|
|
except KeyError:
|
|
|
|
# for some bizarre reason, mnemosyne removes src elements
|
|
|
|
# from missing media
|
|
|
|
pass
|
|
|
|
# strip all other attributes, including implicit max-width
|
|
|
|
for attr, val in tag.attrs:
|
|
|
|
if attr != "src":
|
|
|
|
del tag[attr]
|
|
|
|
# strip superfluous elements
|
|
|
|
for elem in "html", "head", "body", "meta":
|
|
|
|
for tag in doc(elem):
|
|
|
|
tag.replaceWithChildren()
|
|
|
|
html = unicode(doc)
|
|
|
|
return html
|
|
|
|
|
2012-12-21 08:51:59 +01:00
|
|
|
# Advanced menu
|
|
|
|
######################################################################
|
|
|
|
|
|
|
|
def onAdvanced(self):
|
|
|
|
m = QMenu(self.mw)
|
|
|
|
a = m.addAction(_("LaTeX"))
|
|
|
|
a.setShortcut(QKeySequence("Ctrl+T, T"))
|
|
|
|
a.connect(a, SIGNAL("triggered()"), self.insertLatex)
|
|
|
|
a = m.addAction(_("LaTeX equation"))
|
|
|
|
a.setShortcut(QKeySequence("Ctrl+T, E"))
|
|
|
|
a.connect(a, SIGNAL("triggered()"), self.insertLatexEqn)
|
|
|
|
a = m.addAction(_("LaTeX math env."))
|
|
|
|
a.setShortcut(QKeySequence("Ctrl+T, M"))
|
|
|
|
a.connect(a, SIGNAL("triggered()"), self.insertLatexMathEnv)
|
|
|
|
a = m.addAction(_("Edit HTML"))
|
|
|
|
a.setShortcut(QKeySequence("Ctrl+Shift+X"))
|
|
|
|
a.connect(a, SIGNAL("triggered()"), self.onHtmlEdit)
|
|
|
|
m.exec_(QCursor.pos())
|
|
|
|
|
|
|
|
# LaTeX
|
|
|
|
######################################################################
|
|
|
|
|
|
|
|
def insertLatex(self):
|
|
|
|
self.web.eval("wrap('[latex]', '[/latex]');")
|
|
|
|
|
|
|
|
def insertLatexEqn(self):
|
|
|
|
self.web.eval("wrap('[$]', '[/$]');")
|
|
|
|
|
|
|
|
def insertLatexMathEnv(self):
|
|
|
|
self.web.eval("wrap('[$$]', '[/$$]');")
|
|
|
|
|
|
|
|
# Keyboard layout
|
|
|
|
######################################################################
|
|
|
|
|
|
|
|
def setupKeyboard(self):
|
|
|
|
if isWin and self.mw.pm.profile['preserveKeyboard']:
|
|
|
|
a = ctypes.windll.user32.ActivateKeyboardLayout
|
|
|
|
a.restype = ctypes.c_void_p
|
|
|
|
a.argtypes = [ctypes.c_void_p, ctypes.c_uint]
|
|
|
|
g = ctypes.windll.user32.GetKeyboardLayout
|
|
|
|
g.restype = ctypes.c_void_p
|
|
|
|
g.argtypes = [ctypes.c_uint]
|
|
|
|
else:
|
|
|
|
a = g = None
|
|
|
|
self.activateKeyboard = a
|
|
|
|
self.getKeyboard = g
|
|
|
|
|
|
|
|
def updateKeyboard(self):
|
|
|
|
self.keyboardLayouts = {}
|
|
|
|
|
|
|
|
def saveKeyboard(self):
|
|
|
|
if not self.getKeyboard:
|
|
|
|
return
|
|
|
|
self.keyboardLayouts[self.currentField] = self.getKeyboard(0)
|
|
|
|
|
|
|
|
def restoreKeyboard(self):
|
|
|
|
if not self.getKeyboard:
|
|
|
|
return
|
|
|
|
if self.currentField in self.keyboardLayouts:
|
|
|
|
self.activateKeyboard(self.keyboardLayouts[self.currentField], 0)
|
|
|
|
|
|
|
|
# Pasting, drag & drop, and keyboard layouts
|
|
|
|
######################################################################
|
|
|
|
|
|
|
|
class EditorWebView(AnkiWebView):
|
|
|
|
|
|
|
|
def __init__(self, parent, editor):
|
2014-09-15 08:04:14 +02:00
|
|
|
AnkiWebView.__init__(self)
|
2012-12-21 08:51:59 +01:00
|
|
|
self.editor = editor
|
|
|
|
self.strip = self.editor.mw.pm.profile['stripHTML']
|
|
|
|
|
|
|
|
def keyPressEvent(self, evt):
|
|
|
|
if evt.matches(QKeySequence.Paste):
|
|
|
|
self.onPaste()
|
|
|
|
return evt.accept()
|
|
|
|
elif evt.matches(QKeySequence.Copy):
|
|
|
|
self.onCopy()
|
|
|
|
return evt.accept()
|
|
|
|
elif evt.matches(QKeySequence.Cut):
|
|
|
|
self.onCut()
|
|
|
|
return evt.accept()
|
|
|
|
QWebView.keyPressEvent(self, evt)
|
|
|
|
|
|
|
|
def onCut(self):
|
|
|
|
self.triggerPageAction(QWebPage.Cut)
|
|
|
|
self._flagAnkiText()
|
|
|
|
|
|
|
|
def onCopy(self):
|
|
|
|
self.triggerPageAction(QWebPage.Copy)
|
|
|
|
self._flagAnkiText()
|
|
|
|
|
|
|
|
def onPaste(self):
|
2013-07-18 12:56:47 +02:00
|
|
|
mime = self.mungeClip()
|
2012-12-21 08:51:59 +01:00
|
|
|
self.triggerPageAction(QWebPage.Paste)
|
2013-07-18 12:56:47 +02:00
|
|
|
self.restoreClip()
|
2012-12-21 08:51:59 +01:00
|
|
|
|
|
|
|
def mouseReleaseEvent(self, evt):
|
|
|
|
if not isMac and not isWin and evt.button() == Qt.MidButton:
|
|
|
|
# middle click on x11; munge the clipboard before standard
|
|
|
|
# handling
|
2013-07-18 12:56:47 +02:00
|
|
|
mime = self.mungeClip(mode=QClipboard.Selection)
|
2012-12-21 08:51:59 +01:00
|
|
|
AnkiWebView.mouseReleaseEvent(self, evt)
|
2013-07-18 12:56:47 +02:00
|
|
|
self.restoreClip(mode=QClipboard.Selection)
|
2012-12-21 08:51:59 +01:00
|
|
|
else:
|
|
|
|
AnkiWebView.mouseReleaseEvent(self, evt)
|
|
|
|
|
|
|
|
def focusInEvent(self, evt):
|
|
|
|
window = False
|
|
|
|
if evt.reason() in (Qt.ActiveWindowFocusReason, Qt.PopupFocusReason):
|
|
|
|
# editor area got focus again; need to tell js not to adjust cursor
|
|
|
|
self.eval("mouseDown++;")
|
|
|
|
window = True
|
|
|
|
AnkiWebView.focusInEvent(self, evt)
|
|
|
|
if evt.reason() == Qt.TabFocusReason:
|
|
|
|
self.eval("focusField(0);")
|
|
|
|
elif evt.reason() == Qt.BacktabFocusReason:
|
|
|
|
n = len(self.editor.note.fields) - 1
|
|
|
|
self.eval("focusField(%d);" % n)
|
|
|
|
elif window:
|
|
|
|
self.eval("mouseDown--;")
|
|
|
|
|
|
|
|
def dropEvent(self, evt):
|
|
|
|
oldmime = evt.mimeData()
|
|
|
|
# coming from this program?
|
|
|
|
if evt.source():
|
|
|
|
if oldmime.hasHtml():
|
|
|
|
mime = QMimeData()
|
2013-07-11 10:21:16 +02:00
|
|
|
mime.setHtml(self.editor._filterHTML(oldmime.html()))
|
2012-12-21 08:51:59 +01:00
|
|
|
else:
|
|
|
|
# old qt on linux won't give us html when dragging an image;
|
|
|
|
# in that case just do the default action (which is to ignore
|
|
|
|
# the drag)
|
|
|
|
return AnkiWebView.dropEvent(self, evt)
|
|
|
|
else:
|
|
|
|
mime = self._processMime(oldmime)
|
|
|
|
# create a new event with the new mime data and run it
|
|
|
|
new = QDropEvent(evt.pos(), evt.possibleActions(), mime,
|
|
|
|
evt.mouseButtons(), evt.keyboardModifiers())
|
|
|
|
evt.accept()
|
|
|
|
QWebView.dropEvent(self, new)
|
|
|
|
# tell the drop target to take focus so the drop contents are saved
|
|
|
|
self.eval("dropTarget.focus();")
|
|
|
|
self.setFocus()
|
|
|
|
|
2013-07-18 12:56:47 +02:00
|
|
|
def mungeClip(self, mode=QClipboard.Clipboard):
|
2012-12-21 08:51:59 +01:00
|
|
|
clip = self.editor.mw.app.clipboard()
|
|
|
|
mime = clip.mimeData(mode=mode)
|
|
|
|
self.saveClip(mode=mode)
|
|
|
|
mime = self._processMime(mime)
|
|
|
|
clip.setMimeData(mime, mode=mode)
|
2013-07-18 12:56:47 +02:00
|
|
|
return mime
|
2012-12-21 08:51:59 +01:00
|
|
|
|
2013-07-18 12:56:47 +02:00
|
|
|
def restoreClip(self, mode=QClipboard.Clipboard):
|
2012-12-21 08:51:59 +01:00
|
|
|
clip = self.editor.mw.app.clipboard()
|
2013-07-18 12:56:47 +02:00
|
|
|
clip.setMimeData(self.savedClip, mode=mode)
|
2012-12-21 08:51:59 +01:00
|
|
|
|
|
|
|
def saveClip(self, mode):
|
2013-07-18 12:56:47 +02:00
|
|
|
# we don't own the clipboard object, so we need to copy it or we'll crash
|
2012-12-21 08:51:59 +01:00
|
|
|
mime = self.editor.mw.app.clipboard().mimeData(mode=mode)
|
|
|
|
n = QMimeData()
|
|
|
|
if mime.hasText():
|
|
|
|
n.setText(mime.text())
|
|
|
|
if mime.hasHtml():
|
|
|
|
n.setHtml(mime.html())
|
|
|
|
if mime.hasUrls():
|
|
|
|
n.setUrls(mime.urls())
|
|
|
|
if mime.hasImage():
|
|
|
|
n.setImageData(mime.imageData())
|
2013-07-18 12:56:47 +02:00
|
|
|
self.savedClip = n
|
2012-12-21 08:51:59 +01:00
|
|
|
|
|
|
|
def _processMime(self, mime):
|
2013-11-27 11:24:41 +01:00
|
|
|
# print "html=%s image=%s urls=%s txt=%s" % (
|
|
|
|
# mime.hasHtml(), mime.hasImage(), mime.hasUrls(), mime.hasText())
|
|
|
|
# print "html", mime.html()
|
|
|
|
# print "urls", mime.urls()
|
|
|
|
# print "text", mime.text()
|
2013-07-11 10:21:16 +02:00
|
|
|
if mime.hasHtml():
|
|
|
|
return self._processHtml(mime)
|
|
|
|
elif mime.hasUrls():
|
|
|
|
return self._processUrls(mime)
|
2013-09-20 07:41:56 +02:00
|
|
|
elif mime.hasText():
|
|
|
|
return self._processText(mime)
|
2013-05-18 11:38:32 +02:00
|
|
|
elif mime.hasImage():
|
|
|
|
return self._processImage(mime)
|
2012-12-21 08:51:59 +01:00
|
|
|
else:
|
|
|
|
# nothing
|
|
|
|
return QMimeData()
|
|
|
|
|
2013-09-20 07:41:56 +02:00
|
|
|
# when user is dragging a file from a file manager on any platform, the
|
|
|
|
# url type should be set, and it is not URL-encoded. on a mac no text type
|
|
|
|
# is returned, and on windows the text type is not returned in cases like
|
|
|
|
# "foo's bar.jpg"
|
2012-12-21 08:51:59 +01:00
|
|
|
def _processUrls(self, mime):
|
|
|
|
url = mime.urls()[0].toString()
|
2013-05-09 08:34:16 +02:00
|
|
|
# chrome likes to give us the URL twice with a \n
|
|
|
|
url = url.splitlines()[0]
|
2014-01-14 06:59:16 +01:00
|
|
|
newmime = QMimeData()
|
2013-07-11 10:21:16 +02:00
|
|
|
link = self.editor.urlToLink(url)
|
2013-05-16 07:47:42 +02:00
|
|
|
if link:
|
2014-01-14 06:59:16 +01:00
|
|
|
newmime.setHtml(link)
|
|
|
|
elif mime.hasImage():
|
|
|
|
# if we couldn't convert the url to a link and there's an
|
|
|
|
# image on the clipboard (such as copy&paste from
|
|
|
|
# google images in safari), use that instead
|
|
|
|
return self._processImage(mime)
|
2013-11-26 09:57:02 +01:00
|
|
|
else:
|
2014-01-14 06:59:16 +01:00
|
|
|
newmime.setText(url)
|
|
|
|
return newmime
|
2012-12-21 08:51:59 +01:00
|
|
|
|
2013-09-20 07:41:56 +02:00
|
|
|
# if the user has used 'copy link location' in the browser, the clipboard
|
|
|
|
# will contain the URL as text, and no URLs or HTML. the URL will already
|
|
|
|
# be URL-encoded, and shouldn't be a file:// url unless they're browsing
|
|
|
|
# locally, which we don't support
|
2012-12-21 08:51:59 +01:00
|
|
|
def _processText(self, mime):
|
|
|
|
txt = unicode(mime.text())
|
|
|
|
html = None
|
|
|
|
# if the user is pasting an image or sound link, convert it to local
|
2013-07-18 13:32:41 +02:00
|
|
|
if self.editor.isURL(txt):
|
2012-12-21 08:51:59 +01:00
|
|
|
txt = txt.split("\r\n")[0]
|
2013-07-11 10:21:16 +02:00
|
|
|
html = self.editor.urlToLink(txt)
|
2012-12-21 08:51:59 +01:00
|
|
|
new = QMimeData()
|
|
|
|
if html:
|
|
|
|
new.setHtml(html)
|
|
|
|
else:
|
2013-07-17 03:19:09 +02:00
|
|
|
new.setText(txt)
|
2012-12-21 08:51:59 +01:00
|
|
|
return new
|
|
|
|
|
|
|
|
def _processHtml(self, mime):
|
|
|
|
html = mime.html()
|
2013-07-11 10:21:16 +02:00
|
|
|
newMime = QMimeData()
|
|
|
|
if self.strip and not html.startswith("<!--anki-->"):
|
|
|
|
# special case for google images: if after stripping there's no text
|
|
|
|
# and there are image links, we'll paste those as html instead
|
|
|
|
if not stripHTML(html).strip():
|
|
|
|
newHtml = ""
|
|
|
|
mid = self.editor.note.mid
|
|
|
|
for url in self.editor.mw.col.media.filesInStr(
|
|
|
|
mid, html, includeRemote=True):
|
|
|
|
newHtml += self.editor.urlToLink(url)
|
2013-07-11 10:43:11 +02:00
|
|
|
if not newHtml and mime.hasImage():
|
|
|
|
return self._processImage(mime)
|
2013-07-11 10:21:16 +02:00
|
|
|
newMime.setHtml(newHtml)
|
|
|
|
else:
|
|
|
|
# use .text() if available so newlines are preserved; otherwise strip
|
|
|
|
if mime.hasText():
|
|
|
|
return self._processText(mime)
|
|
|
|
else:
|
|
|
|
newMime.setText(stripHTML(mime.text()))
|
2012-12-21 08:51:59 +01:00
|
|
|
else:
|
2013-07-11 14:22:51 +02:00
|
|
|
if html.startswith("<!--anki-->"):
|
|
|
|
html = html[11:]
|
|
|
|
# no html stripping
|
2013-07-11 10:21:16 +02:00
|
|
|
html = self.editor._filterHTML(html, localize=True)
|
|
|
|
newMime.setHtml(html)
|
|
|
|
return newMime
|
2012-12-21 08:51:59 +01:00
|
|
|
|
|
|
|
def _processImage(self, mime):
|
|
|
|
im = QImage(mime.imageData())
|
|
|
|
uname = namedtmp("paste-%d" % im.cacheKey())
|
|
|
|
if self.editor.mw.pm.profile.get("pastePNG", False):
|
|
|
|
ext = ".png"
|
|
|
|
im.save(uname+ext, None, 50)
|
|
|
|
else:
|
|
|
|
ext = ".jpg"
|
|
|
|
im.save(uname+ext, None, 80)
|
|
|
|
# invalid image?
|
|
|
|
if not os.path.exists(uname+ext):
|
|
|
|
return QMimeData()
|
|
|
|
mime = QMimeData()
|
|
|
|
mime.setHtml(self.editor._addMedia(uname+ext))
|
|
|
|
return mime
|
|
|
|
|
|
|
|
def _flagAnkiText(self):
|
|
|
|
# add a comment in the clipboard html so we can tell text is copied
|
|
|
|
# from us and doesn't need to be stripped
|
|
|
|
clip = self.editor.mw.app.clipboard()
|
|
|
|
mime = clip.mimeData()
|
|
|
|
if not mime.hasHtml():
|
|
|
|
return
|
|
|
|
html = mime.html()
|
|
|
|
mime.setHtml("<!--anki-->" + mime.html())
|
|
|
|
|
|
|
|
def contextMenuEvent(self, evt):
|
|
|
|
m = QMenu(self)
|
|
|
|
a = m.addAction(_("Cut"))
|
2013-05-10 07:49:21 +02:00
|
|
|
a.connect(a, SIGNAL("triggered()"), self.onCut)
|
2012-12-21 08:51:59 +01:00
|
|
|
a = m.addAction(_("Copy"))
|
2013-05-10 07:49:21 +02:00
|
|
|
a.connect(a, SIGNAL("triggered()"), self.onCopy)
|
2012-12-21 08:51:59 +01:00
|
|
|
a = m.addAction(_("Paste"))
|
2013-05-10 07:49:21 +02:00
|
|
|
a.connect(a, SIGNAL("triggered()"), self.onPaste)
|
2013-07-16 09:42:50 +02:00
|
|
|
runHook("EditorWebView.contextMenuEvent", self, m)
|
2012-12-21 08:51:59 +01:00
|
|
|
m.popup(QCursor.pos())
|