mirror of
https://github.com/ankitects/anki.git
synced 2025-09-19 14:32:22 -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.
336 lines
11 KiB
Python
336 lines
11 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
|
|
from copy import deepcopy
|
|
from typing import Any
|
|
|
|
import aqt
|
|
from anki.errors import DeckRenameError
|
|
from anki.hooks import runHook
|
|
from anki.lang import _, ngettext
|
|
from anki.utils import fmtTimeSpan, ids2str
|
|
from aqt.qt import *
|
|
from aqt.sound import clearAudioQueue
|
|
from aqt.utils import askUser, getOnlyText, openHelp, openLink, shortcut, showWarning
|
|
|
|
|
|
class DeckBrowser:
|
|
_dueTree: Any
|
|
|
|
def __init__(self, mw):
|
|
self.mw = mw
|
|
self.web = mw.web
|
|
self.bottom = aqt.toolbar.BottomBar(mw, mw.bottomWeb)
|
|
self.scrollPos = QPoint(0, 0)
|
|
|
|
def show(self):
|
|
clearAudioQueue()
|
|
self.web.resetHandlers()
|
|
self.web.onBridgeCmd = self._linkHandler
|
|
self._renderPage()
|
|
|
|
def refresh(self):
|
|
self._renderPage()
|
|
|
|
# Event handlers
|
|
##########################################################################
|
|
|
|
def _linkHandler(self, url):
|
|
if ":" in url:
|
|
(cmd, arg) = url.split(":")
|
|
else:
|
|
cmd = url
|
|
if cmd == "open":
|
|
self._selDeck(arg)
|
|
elif cmd == "opts":
|
|
self._showOptions(arg)
|
|
elif cmd == "shared":
|
|
self._onShared()
|
|
elif cmd == "import":
|
|
self.mw.onImport()
|
|
elif cmd == "lots":
|
|
openHelp("using-decks-appropriately")
|
|
elif cmd == "hidelots":
|
|
self.mw.pm.profile["hideDeckLotsMsg"] = True
|
|
self.refresh()
|
|
elif cmd == "create":
|
|
deck = getOnlyText(_("Name for deck:"))
|
|
if deck:
|
|
self.mw.col.decks.id(deck)
|
|
self.refresh()
|
|
elif cmd == "drag":
|
|
draggedDeckDid, ontoDeckDid = arg.split(",")
|
|
self._dragDeckOnto(draggedDeckDid, ontoDeckDid)
|
|
elif cmd == "collapse":
|
|
self._collapse(arg)
|
|
return False
|
|
|
|
def _selDeck(self, did):
|
|
self.mw.col.decks.select(did)
|
|
self.mw.onOverview()
|
|
|
|
# HTML generation
|
|
##########################################################################
|
|
|
|
_body = """
|
|
<center>
|
|
<table cellspacing=0 cellpading=3>
|
|
%(tree)s
|
|
</table>
|
|
|
|
<br>
|
|
%(stats)s
|
|
%(countwarn)s
|
|
</center>
|
|
"""
|
|
|
|
def _renderPage(self, reuse=False):
|
|
if not reuse:
|
|
self._dueTree = self.mw.col.sched.deckDueTree()
|
|
self.__renderPage(None)
|
|
return
|
|
self.web.evalWithCallback("window.pageYOffset", self.__renderPage)
|
|
|
|
def __renderPage(self, offset):
|
|
tree = self._renderDeckTree(self._dueTree)
|
|
stats = self._renderStats()
|
|
self.web.stdHtml(
|
|
self._body % dict(tree=tree, stats=stats, countwarn=self._countWarn()),
|
|
css=["deckbrowser.css"],
|
|
js=["jquery.js", "jquery-ui.js", "deckbrowser.js"],
|
|
)
|
|
self.web.key = "deckBrowser"
|
|
self._drawButtons()
|
|
if offset is not None:
|
|
self._scrollToOffset(offset)
|
|
|
|
def _scrollToOffset(self, offset):
|
|
self.web.eval("$(function() { window.scrollTo(0, %d, 'instant'); });" % offset)
|
|
|
|
def _renderStats(self):
|
|
cards, thetime = self.mw.col.db.first(
|
|
"""
|
|
select count(), sum(time)/1000 from revlog
|
|
where id > ?""",
|
|
(self.mw.col.sched.dayCutoff - 86400) * 1000,
|
|
)
|
|
cards = cards or 0
|
|
thetime = thetime or 0
|
|
msgp1 = (
|
|
ngettext("<!--studied-->%d card", "<!--studied-->%d cards", cards) % cards
|
|
)
|
|
buf = _("Studied %(a)s %(b)s today.") % dict(
|
|
a=msgp1, b=fmtTimeSpan(thetime, unit=1, inTime=True)
|
|
)
|
|
return buf
|
|
|
|
def _countWarn(self):
|
|
if self.mw.col.decks.count() < 25 or self.mw.pm.profile.get("hideDeckLotsMsg"):
|
|
return ""
|
|
return "<br><div style='width:50%;border: 1px solid #000;padding:5px;'>" + (
|
|
_("You have a lot of decks. Please see %(a)s. %(b)s")
|
|
% dict(
|
|
a=(
|
|
"<a href=# onclick=\"return pycmd('lots')\">%s</a>" % _("this page")
|
|
),
|
|
b=(
|
|
"<br><small><a href=# onclick='return pycmd(\"hidelots\")'>("
|
|
"%s)</a></small>" % (_("hide")) + "</div>"
|
|
),
|
|
)
|
|
)
|
|
|
|
def _renderDeckTree(self, nodes, depth=0):
|
|
if not nodes:
|
|
return ""
|
|
if depth == 0:
|
|
buf = """
|
|
<tr><th colspan=5 align=left>%s</th><th class=count>%s</th>
|
|
<th class=count>%s</th><th class=optscol></th></tr>""" % (
|
|
_("Deck"),
|
|
_("Due"),
|
|
_("New"),
|
|
)
|
|
buf += self._topLevelDragRow()
|
|
else:
|
|
buf = ""
|
|
nameMap = self.mw.col.decks.nameMap()
|
|
for node in nodes:
|
|
buf += self._deckRow(node, depth, len(nodes), nameMap)
|
|
if depth == 0:
|
|
buf += self._topLevelDragRow()
|
|
return buf
|
|
|
|
def _deckRow(self, node, depth, cnt, nameMap):
|
|
name, did, due, lrn, new, children = node
|
|
deck = self.mw.col.decks.get(did)
|
|
if did == 1 and cnt > 1 and not children:
|
|
# if the default deck is empty, hide it
|
|
if not self.mw.col.db.scalar("select 1 from cards where did = 1"):
|
|
return ""
|
|
# parent toggled for collapsing
|
|
for parent in self.mw.col.decks.parents(did, nameMap):
|
|
if parent["collapsed"]:
|
|
buff = ""
|
|
return buff
|
|
prefix = "-"
|
|
if self.mw.col.decks.get(did)["collapsed"]:
|
|
prefix = "+"
|
|
due += lrn
|
|
|
|
def indent():
|
|
return " " * 6 * depth
|
|
|
|
if did == self.mw.col.conf["curDeck"]:
|
|
klass = "deck current"
|
|
else:
|
|
klass = "deck"
|
|
buf = "<tr class='%s' id='%d'>" % (klass, did)
|
|
# deck link
|
|
if children:
|
|
collapse = (
|
|
"<a class=collapse href=# onclick='return pycmd(\"collapse:%d\")'>%s</a>"
|
|
% (did, prefix)
|
|
)
|
|
else:
|
|
collapse = "<span class=collapse></span>"
|
|
if deck["dyn"]:
|
|
extraclass = "filtered"
|
|
else:
|
|
extraclass = ""
|
|
buf += """
|
|
|
|
<td class=decktd colspan=5>%s%s<a class="deck %s"
|
|
href=# onclick="return pycmd('open:%d')">%s</a></td>""" % (
|
|
indent(),
|
|
collapse,
|
|
extraclass,
|
|
did,
|
|
name,
|
|
)
|
|
# due counts
|
|
def nonzeroColour(cnt, colour):
|
|
if not cnt:
|
|
colour = "#e0e0e0"
|
|
if cnt >= 1000:
|
|
cnt = "1000+"
|
|
return "<font color='%s'>%s</font>" % (colour, cnt)
|
|
|
|
buf += "<td align=right>%s</td><td align=right>%s</td>" % (
|
|
nonzeroColour(due, "#007700"),
|
|
nonzeroColour(new, "#000099"),
|
|
)
|
|
# options
|
|
buf += (
|
|
"<td align=center class=opts><a onclick='return pycmd(\"opts:%d\");'>"
|
|
"<img src='/_anki/imgs/gears.svg' class=gears></a></td></tr>" % did
|
|
)
|
|
# children
|
|
buf += self._renderDeckTree(children, depth + 1)
|
|
return buf
|
|
|
|
def _topLevelDragRow(self):
|
|
return "<tr class='top-level-drag-row'><td colspan='6'> </td></tr>"
|
|
|
|
# Options
|
|
##########################################################################
|
|
|
|
def _showOptions(self, did):
|
|
m = QMenu(self.mw)
|
|
a = m.addAction(_("Rename"))
|
|
a.triggered.connect(lambda b, did=did: self._rename(did))
|
|
a = m.addAction(_("Options"))
|
|
a.triggered.connect(lambda b, did=did: self._options(did))
|
|
a = m.addAction(_("Export"))
|
|
a.triggered.connect(lambda b, did=did: self._export(did))
|
|
a = m.addAction(_("Delete"))
|
|
a.triggered.connect(lambda b, did=did: self._delete(did))
|
|
runHook("showDeckOptions", m, did)
|
|
m.exec_(QCursor.pos())
|
|
|
|
def _export(self, did):
|
|
self.mw.onExport(did=did)
|
|
|
|
def _rename(self, did):
|
|
self.mw.checkpoint(_("Rename Deck"))
|
|
deck = self.mw.col.decks.get(did)
|
|
oldName = deck["name"]
|
|
newName = getOnlyText(_("New deck name:"), default=oldName)
|
|
newName = newName.replace('"', "")
|
|
if not newName or newName == oldName:
|
|
return
|
|
try:
|
|
self.mw.col.decks.rename(deck, newName)
|
|
except DeckRenameError as e:
|
|
return showWarning(e.description)
|
|
self.show()
|
|
|
|
def _options(self, did):
|
|
# select the deck first, because the dyn deck conf assumes the deck
|
|
# we're editing is the current one
|
|
self.mw.col.decks.select(did)
|
|
self.mw.onDeckConf()
|
|
|
|
def _collapse(self, did):
|
|
self.mw.col.decks.collapse(did)
|
|
self._renderPage(reuse=True)
|
|
|
|
def _dragDeckOnto(self, draggedDeckDid, ontoDeckDid):
|
|
try:
|
|
self.mw.col.decks.renameForDragAndDrop(draggedDeckDid, ontoDeckDid)
|
|
except DeckRenameError as e:
|
|
return showWarning(e.description)
|
|
|
|
self.show()
|
|
|
|
def _delete(self, did):
|
|
if str(did) == "1":
|
|
return showWarning(_("The default deck can't be deleted."))
|
|
self.mw.checkpoint(_("Delete Deck"))
|
|
deck = self.mw.col.decks.get(did)
|
|
if not deck["dyn"]:
|
|
dids = [did] + [r[1] for r in self.mw.col.decks.children(did)]
|
|
cnt = self.mw.col.db.scalar(
|
|
"select count() from cards where did in {0} or "
|
|
"odid in {0}".format(ids2str(dids))
|
|
)
|
|
if cnt:
|
|
extra = ngettext(" It has %d card.", " It has %d cards.", cnt) % cnt
|
|
else:
|
|
extra = None
|
|
if (
|
|
deck["dyn"]
|
|
or not extra
|
|
or askUser(
|
|
(_("Are you sure you wish to delete %s?") % deck["name"]) + extra
|
|
)
|
|
):
|
|
self.mw.progress.start(immediate=True)
|
|
self.mw.col.decks.rem(did, True)
|
|
self.mw.progress.finish()
|
|
self.show()
|
|
|
|
# Top buttons
|
|
######################################################################
|
|
|
|
drawLinks = [
|
|
["", "shared", _("Get Shared")],
|
|
["", "create", _("Create Deck")],
|
|
["Ctrl+I", "import", _("Import File")], # Ctrl+I works from menu
|
|
]
|
|
|
|
def _drawButtons(self):
|
|
buf = ""
|
|
drawLinks = deepcopy(self.drawLinks)
|
|
for b in drawLinks:
|
|
if b[0]:
|
|
b[0] = _("Shortcut key: %s") % shortcut(b[0])
|
|
buf += """
|
|
<button title='%s' onclick='pycmd(\"%s\");'>%s</button>""" % tuple(
|
|
b
|
|
)
|
|
self.bottom.draw(buf)
|
|
self.bottom.web.onBridgeCmd = self._linkHandler
|
|
|
|
def _onShared(self):
|
|
openLink(aqt.appShared + "decks/")
|