Anki/aqt/overview.py

# -*- coding: utf-8 -*-
# Copyright: Damien Elmes <anki@ichi2.net>
# License: GNU GPL, version 3 or later; http://www.gnu.org/copyleft/gpl.html

import simplejson
from PyQt4.QtCore import *
from PyQt4.QtGui import *
from anki.consts import NEW_CARDS_RANDOM

class Overview(object):
    "Deck overview."

    def __init__(self, mw):
        self.mw = mw
        self.web = mw.web

    def show(self):
        self._setupToolbar()
        self.mw.setKeyHandler(self._keyHandler)
        self.web.setLinkHandler(self._linkHandler)
        self._renderPage()

    # Handlers
    ############################################################

    def _keyHandler(self, evt):
        txt = evt.text()
        if txt == "1":
            self._linkHandler("studysel")
        elif txt == "2":
            self._linkHandler("studyall")
        elif txt == "3":
            self._linkHandler("cramsel")
        elif txt == "4":
            self._linkHandler("cramall")
        elif txt == "o":
            self._linkHandler("opts")
        elif txt == "d":
            self._linkHandler("list")
        else:
            evt.ignore()
            return
        evt.accept()

    def _linkHandler(self, url):
        print "link", url
        if url == "studysel":
            self.mw.deck.sched.useGroups = True
            self.mw.deck.reset()
            self.mw.moveToState("review")
        elif url == "opts":
            pass
        elif url == "list":
            self.mw.close()

    # HTML
    ############################################################

    def _renderPage(self):
        css = self.mw.sharedCSS + self._overviewCSS
        fc = self._ovForecast()
        tbl = self._overviewTable()
        but = self.mw.button
        buts = (but("list", _("Deck List"), "d") +
                but("refr", _("Refresh"), "r") +
                but("opts", _("Study Options"), "o"))
        self.web.stdHtml(self._overviewBody % dict(
            title=_("Overview"),
            table=tbl,
            fcsub=_("Reviews over next two weeks"),
            fcdata=fc,
            buts=buts,
            opts=self._ovOpts(),
            ), css)

    _overviewBody = """
<center>
<h1>%(title)s</h1>
%(buts)s
<p>
%(table)s
<p>
<div id="placeholder" style="width:350px; height:100px;"></div>
<span class=sub>%(fcsub)s</span>
<p>
%(opts)s
</center>

<script id="source" language="javascript" type="text/javascript">
$(function () {
    var d = %(fcdata)s;
    if (d) {
    $.plot($("#placeholder"), [
    { data: d, bars: { show: true, barWidth: 0.8 }, color: "#0c0" }
    ], {
    xaxis: { ticks: [[0.4, "Today"]] }
    });
    } else {
    $("#placeholder").hide();
    $(".sub").hide();
    }
});
</script>
"""

    _overviewCSS = """
.due { text-align: right; }
.new { text-align: right; }
.sub { font-size: 80%; color: #555; }
"""

    def _overviewTable(self):
        counts = self._ovCounts()
        but = self.mw.button
        buf = "<table cellspacing=0 cellpadding=3 width=400>"
        buf += "<tr><th></th><th align=right>%s</th>" % _("Due")
        buf += "<th align=right>%s</th><th></th></tr>" % _("New")
        line = "<tr><td><b>%s</b></td><td class=due>%s</td>"
        line += "<td class=new>%s</td><td align=right>%s</td></tr>"
        buf += line % (
            "<a href=chgrp>%s</a>" % _("Selected Groups"),
            counts[0], counts[1],
            but("studysel", _("Study"), "1", "gbut") +
            but("cramsel", _("Cram"), "3"))
        buf += line % (
            _("Whole Deck"),
            counts[2], counts[3],
            but("studyall", _("Study"), "2", "gbut") +
            but("cramall", _("Cram"), "4"))
        buf += "</table>"
        return buf

    def _ovOpts(self):
        return ""
#         if self.mw.deck.qconf['newCardOrder'] == NEW_CARDS_RANDOM:
#             ord = _("random")
#         else:
#             ord = _("order added")
#         buf = """
# <table width=400>
# <tr><td><b>%s</b></td><td align=center>%s</td></tr>
# <tr><td><b>%s</b></td><td align=center>%s</td></tr>
# </table>""" % (
#     _("New cards per day"), self.mw.deck.qconf['newPerDay'],
#     _("New card order"), ord)
#         return buf

    # Data
    ##########################################################################

    def _ovCounts(self):
        oldNew = self.mw.deck.qconf['newGroups']
        oldRev = self.mw.deck.qconf['revGroups']
        # we have the limited count already
        selcnt = self.mw.deck.sched.selCounts()
        allcnt = self.mw.deck.sched.allCounts()
        return [
            selcnt[1] + selcnt[2],
            selcnt[0],
            allcnt[1] + allcnt[2],
            allcnt[0],
        ]

    def _ovForecast(self):
        fc = self.mw.deck.sched.dueForecast(14)
        if not sum(fc):
            return "''"
        return simplejson.dumps(tuple(enumerate(fc)))

    # Toolbar
    ##########################################################################

    def _setupToolbar(self):
        if not self.mw.config['showToolbar']:
            return
        self.mw.form.toolBar.show()