two_button.py

# -*- mode: Python ; coding: utf-8 -*-
# Good / Again (Yes No) 2 buttons only
# License: GNU GPL, version 3 or later; http://www.gnu.org/copyleft/gpl.html
# No support. Use it AS IS on your own risk.

from aqt import mw
from anki import version
from . import semver
from aqt.utils import tooltip


rightLabel = "Pass"
wrongLabel = "Fail"

black = '#000'
green = '#080'

BUTTON_LABEL = ['<span style="color:' + black + ';">' + wrongLabel + '</span>',
                '<span style="color:' + green + ';">' + rightLabel + '</span>']

# default map (for v1 scheduler -- v2 always shows 4 buttons):
# 2 buttons:    1 (again) 2 (good) 3 (none) 4 (none)
# 3b:           1 (again) 2 (good) 3 (easy) 4 (none)
# 4b:           1 (again) 2 (hard) 3 (good) 4 (easy)
# (also 0 (None) in each row)

# remap:
# 2 buttons:    1 (again) 2 (good) 2 (good) 2 (good)
# 3b:           1 (again) 2 (good) 2 (good) 2 (good)
# 4b:           1 (again) 3 (good) 3 (good) 3 (good)


if semver.Version(version) >= semver.Version("2.1.33"):
    # mw.col.schedVer()
    from aqt import gui_hooks

    def two_button(button_tuple, reviewer, card):
        button_count = mw.col.sched.answerButtons(card)
        if button_count in [2, 3]:
            # for old scheduler
            button_tuple = ((1, BUTTON_LABEL[0]), (2, BUTTON_LABEL[1]))
        else:
            button_tuple = ((1, BUTTON_LABEL[0]), (3, BUTTON_LABEL[1]))
        return button_tuple

    def remap_answers(ease_tuple, reviewer, card):
        button_count = mw.col.sched.answerButtons(card)
        new_ease = 3
        if ease_tuple[1] <= 1:
            # failing
            new_ease = 1
        elif button_count < 4:
            # old scheduler and learning or relearning
            new_ease = 2
        return (ease_tuple[0], new_ease)

elif semver.Version(version) < semver.Version("2.1.33"):
    # use old style hooks for old versions
    from aqt.reviewer import Reviewer
    from anki.hooks import wrap
    from aqt.qt import *

    # Replace _answerButtonList method to reduce buttons to Pass/Fail always
    def answerButtonList(self):
        # https://git.io/JfykM
        abl = ((1, BUTTON_LABEL[0]),)
        cnt = self.mw.col.sched.answerButtons(self.card)
        if cnt == 2 or cnt == 3:
            return abl + ((2, BUTTON_LABEL[1]),)
        else:
            return abl + ((3, BUTTON_LABEL[1]),)

    def AEF_answerCard(self, ease):
        count = mw.col.sched.answerButtons(mw.reviewer.card)  # Get button count
        new_ease = 3  # default success, good = 3
        if ease <= 1:  # if failed, again = 1
            new_ease = 1
        elif count < 4:  # if 2-3 buttons, good = 2
            new_ease = 2
        ease = new_ease
        __originalAnswerCard(self, ease)

    __originalAnswerCard = Reviewer._answerCard
    Reviewer._answerCard = AEF_answerCard

    def myAnswerButtons(self, _old) -> str:
        times = []
        default = self._defaultEase()

        def but(i, label):
            if i == default:
                extra = 'id=defease'
            else:
                extra = ''
            due = self._buttonTime(i)
            outval = f'''<td align=center>{due}
                         <button {extra}
                         title="Shortcut key: {i}"
                         data-ease="{i}"
                         onclick='pycmd("ease{i}");'>{label}</button></td>'''
            return outval

        buf = "<center><table cellpading=0 cellspacing=0><tr>"
        for ease, lbl in answerButtonList(self):
            buf += but(ease, lbl)
        buf += '</tr></table>'
        script = """<script>$(function () { $('#defease').focus(); });
                    </script>"""
        return buf + script

    __originalAnswerButtons = Reviewer._answerButtons
    AEF_answerButtons = wrap(Reviewer._answerButtons, myAnswerButtons, 'around')
    Reviewer._answerButtons = AEF_answerButtons
        

def disable_two_button():
    if semver.Version(version) >= semver.Version("2.1.33"):
        gui_hooks.reviewer_will_answer_card.remove(remap_answers)
        gui_hooks.reviewer_will_init_answer_buttons.remove(two_button)
    elif semver.Version(version) < semver.Version("2.1.33"):
        Reviewer._answerCard = __originalAnswerCard
        Reviewer._answerButtons = __originalAnswerButtons


def enable_two_button():
    if semver.Version(version) >= semver.Version("2.1.33"):
        # prevent hook stacking
        disable_two_button()
        gui_hooks.reviewer_will_init_answer_buttons.append(two_button)
        gui_hooks.reviewer_will_answer_card.append(remap_answers)
    elif semver.Version(version) < semver.Version("2.1.33"):
        Reviewer._answerCard = AEF_answerCard
        Reviewer._answerButtons = AEF_answerButtons