anki/qt/aqt/stats.py

208 lines
6.6 KiB
Python
Raw Normal View History

2019-02-05 04:59:03 +01:00
# Copyright: Ankitects Pty Ltd and contributors
# License: GNU AGPL, version 3 or later; http://www.gnu.org/licenses/agpl.html
from __future__ import annotations
2019-12-20 10:19:03 +01:00
import time
2021-02-02 14:30:53 +01:00
from typing import Any
2019-12-20 10:19:03 +01:00
import aqt
import aqt.forms
import aqt.main
from aqt import gui_hooks
2019-12-20 10:19:03 +01:00
from aqt.qt import *
2020-06-30 08:23:46 +02:00
from aqt.theme import theme_manager
2019-12-23 01:34:10 +01:00
from aqt.utils import (
addCloseShortcut,
disable_help_button,
2019-12-23 01:34:10 +01:00
getSaveFile,
maybeHideClose,
restoreGeom,
saveGeom,
tooltip,
tr,
2019-12-23 01:34:10 +01:00
)
class NewDeckStats(QDialog):
"""New deck stats."""
2021-02-01 14:28:21 +01:00
def __init__(self, mw: aqt.main.AnkiQt) -> None:
QDialog.__init__(self, mw, Qt.WindowType.Window)
mw.garbage_collect_on_dialog_finish(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)
disable_help_button(self)
f = self.form
f.setupUi(self)
f.groupBox.setVisible(False)
f.groupBox_2.setVisible(False)
restoreGeom(self, self.name)
b = f.buttonBox.addButton(
tr.statistics_save_pdf(), QDialogButtonBox.ButtonRole.ActionRole
)
qconnect(b.clicked, self.saveImage)
b.setAutoDefault(False)
maybeHideClose(self.form.buttonBox)
addCloseShortcut(self)
gui_hooks.stats_dialog_will_show(self)
self.form.web.hide_while_preserving_layout()
self.show()
self.refresh()
2021-01-05 22:04:31 +01:00
self.form.web.set_bridge_command(self._on_bridge_cmd, self)
self.activateWindow()
def reject(self) -> None:
2021-11-29 03:31:37 +01:00
self.form.web.cleanup()
self.form.web = None
saveGeom(self, self.name)
aqt.dialogs.markClosed("NewDeckStats")
QDialog.reject(self)
2021-02-02 14:30:53 +01:00
def closeWithCallback(self, callback: Callable[[], None]) -> None:
self.reject()
callback()
2021-02-01 14:28:21 +01:00
def _imagePath(self) -> str:
2019-12-23 01:34:10 +01:00
name = time.strftime("-%Y-%m-%d@%H-%M-%S.pdf", time.localtime(time.time()))
2021-03-26 04:48:26 +01:00
name = f"anki-{tr.statistics_stats()}{name}"
2019-12-23 01:34:10 +01:00
file = getSaveFile(
self,
2021-03-26 04:48:26 +01:00
title=tr.statistics_save_pdf(),
2019-12-23 01:34:10 +01:00
dir_description="stats",
key="stats",
ext=".pdf",
fname=name,
)
2017-09-10 07:50:34 +02:00
return file
2017-01-13 08:36:26 +01:00
2021-02-01 14:28:21 +01:00
def saveImage(self) -> None:
2017-01-13 08:36:26 +01:00
path = self._imagePath()
2017-09-10 07:50:34 +02:00
if not path:
return
# When scrolled down in dark mode, the top of the page in the
# final PDF will have a white background, making the text and graphs
# unreadable. A simple fix for now is to scroll to the top of the
# page first.
def after_scroll(arg: Any) -> None:
self.form.web.page().printToPdf(path)
tooltip(tr.statistics_saved())
self.form.web.evalWithCallback("window.scrollTo(0, 0);", after_scroll)
2021-02-02 14:30:53 +01:00
# legacy add-ons
def changePeriod(self, n: Any) -> None:
pass
2021-02-02 14:30:53 +01:00
def changeScope(self, type: Any) -> None:
pass
2021-01-05 22:04:31 +01:00
def _on_bridge_cmd(self, cmd: str) -> bool:
if cmd.startswith("browserSearch"):
2021-01-07 01:09:55 +01:00
_, query = cmd.split(":", 1)
2021-01-05 22:04:31 +01:00
browser = aqt.dialogs.open("Browser", self.mw)
browser.search_for(query)
return False
def refresh(self) -> None:
self.form.web.load_ts_page("graphs")
class DeckStats(QDialog):
"""Legacy deck stats, used by some add-ons."""
def __init__(self, mw: aqt.main.AnkiQt) -> None:
QDialog.__init__(self, mw, Qt.WindowType.Window)
mw.garbage_collect_on_dialog_finish(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)
disable_help_button(self)
f = self.form
if theme_manager.night_mode and not theme_manager.macos_dark_mode():
# the grouping box renders incorrectly in the fusion theme. 5.9+
# 5.13 behave differently to 5.14, but it looks bad in either case,
# and adjusting the top margin makes the 'save PDF' button show in
# the wrong place, so for now we just disable the border instead
self.setStyleSheet("QGroupBox { border: 0; }")
f.setupUi(self)
restoreGeom(self, self.name)
b = f.buttonBox.addButton(
tr.statistics_save_pdf(), QDialogButtonBox.ButtonRole.ActionRole
)
qconnect(b.clicked, self.saveImage)
b.setAutoDefault(False)
qconnect(f.groups.clicked, lambda: self.changeScope("deck"))
f.groups.setShortcut("g")
qconnect(f.all.clicked, lambda: self.changeScope("collection"))
qconnect(f.month.clicked, lambda: self.changePeriod(0))
qconnect(f.year.clicked, lambda: self.changePeriod(1))
qconnect(f.life.clicked, lambda: self.changePeriod(2))
maybeHideClose(self.form.buttonBox)
addCloseShortcut(self)
gui_hooks.stats_dialog_old_will_show(self)
self.show()
self.refresh()
self.activateWindow()
def reject(self) -> None:
2021-11-29 03:31:37 +01:00
self.form.web.cleanup()
self.form.web = None
saveGeom(self, self.name)
aqt.dialogs.markClosed("DeckStats")
QDialog.reject(self)
2021-02-02 14:30:53 +01:00
def closeWithCallback(self, callback: Callable[[], None]) -> None:
self.reject()
callback()
2021-02-01 14:28:21 +01:00
def _imagePath(self) -> str:
name = time.strftime("-%Y-%m-%d@%H-%M-%S.pdf", time.localtime(time.time()))
2021-03-26 04:48:26 +01:00
name = f"anki-{tr.statistics_stats()}{name}"
file = getSaveFile(
self,
2021-03-26 04:48:26 +01:00
title=tr.statistics_save_pdf(),
dir_description="stats",
key="stats",
ext=".pdf",
fname=name,
)
return file
2021-02-01 14:28:21 +01:00
def saveImage(self) -> None:
path = self._imagePath()
if not path:
return
self.form.web.page().printToPdf(path)
2021-03-26 04:48:26 +01:00
tooltip(tr.statistics_saved())
def changePeriod(self, n: int) -> None:
self.period = n
self.refresh()
def changeScope(self, type: str) -> None:
self.wholeCollection = type == "collection"
self.refresh()
def refresh(self) -> None:
self.mw.progress.start(parent=self)
stats = self.mw.col.stats()
stats.wholeCollection = self.wholeCollection
self.report = stats.report(type=self.period)
self.form.web.set_title("deck stats")
self.form.web.stdHtml(
f"<html><body>{self.report}</body></html>",
js=["js/vendor/jquery.min.js", "js/vendor/plot.js"],
context=self,
)
self.mw.progress.finish()