anki/aqt/stats.py

80 lines
2.6 KiB
Python
Raw Normal View History

# Copyright: Damien Elmes <anki@ichi2.net>
# -*- coding: utf-8 -*-
# License: GNU AGPL, version 3 or later; http://www.gnu.org/licenses/agpl.html
from aqt.qt import *
import os, time
2017-01-13 08:36:26 +01:00
from aqt.utils import saveGeom, restoreGeom, maybeHideClose, addCloseShortcut, \
tooltip
import aqt
# Deck Stats
######################################################################
class DeckStats(QDialog):
def __init__(self, mw):
QDialog.__init__(self, mw, Qt.Window)
mw.setupDialogGC(self)
self.mw = mw
self.name = "deckStats"
self.period = 0
self.form = aqt.forms.stats.Ui_Dialog()
self.oldPos = None
self.wholeCollection = False
self.setMinimumWidth(700)
f = self.form
f.setupUi(self)
restoreGeom(self, self.name)
2017-01-13 08:36:26 +01:00
b = f.buttonBox.addButton(_("Save PDF"),
QDialogButtonBox.ActionRole)
2017-01-13 08:36:26 +01:00
b.clicked.connect(self.saveImage)
b.setAutoDefault(False)
f.groups.clicked.connect(lambda: self.changeScope("deck"))
f.groups.setShortcut("g")
f.all.clicked.connect(lambda: self.changeScope("collection"))
f.month.clicked.connect(lambda: self.changePeriod(0))
f.year.clicked.connect(lambda: self.changePeriod(1))
f.life.clicked.connect(lambda: self.changePeriod(2))
maybeHideClose(self.form.buttonBox)
addCloseShortcut(self)
self.refresh()
self.show()
self.activateWindow()
def reject(self):
saveGeom(self, self.name)
QDialog.reject(self)
2017-01-13 08:36:26 +01:00
def _imagePath(self):
name = time.strftime("-%Y-%m-%d@%H-%M-%S.pdf",
time.localtime(time.time()))
name = "anki-"+_("stats")+name
desktopPath = QStandardPaths.writableLocation(
QStandardPaths.DesktopLocation)
if not os.path.exists(desktopPath):
os.mkdir(desktopPath)
path = os.path.join(desktopPath, name)
2017-01-13 08:36:26 +01:00
return path
def saveImage(self):
path = self._imagePath()
self.form.web.page().printToPdf(path)
tooltip(_("A PDF file was saved to your desktop."))
def changePeriod(self, n):
self.period = n
self.refresh()
def changeScope(self, type):
self.wholeCollection = type == "collection"
self.refresh()
def refresh(self):
self.mw.progress.start(immediate=True)
stats = self.mw.col.stats()
stats.wholeCollection = self.wholeCollection
self.report = stats.report(type=self.period)
self.form.web.stdHtml("<html><body>"+self.report+"</body></html>")
self.mw.progress.finish()