rule.py

from aqt import mw

from .action import applyActions, reverseAction
from .config import getUserOption
from .trigger import checkTrigger, reverseTrigger

"""True when applying holds. This ensure that a flush triggered by application of the rules does not trigger other rule application"""
currentlyApplying = False


def applyRuleToNote(note, rule):
    """Whether there was a change"""
    global currentlyApplying
    if currentlyApplying:
        return
    if rule is None:
        return False
    currentlyApplying = True
    if not checkTrigger(note, rule["trigger"]):
        currentlyApplying = False
        return False
    ret = applyActions(note, rule["action"])
    currentlyApplying = False
    return ret


def applyRulesToNote(note, rules):
    """Whether there was a change"""
    someChange = False
    for rule in rules:
        someChange = applyRuleToNote(note, rule) or someChange
    return someChange


def reverse(rule):
    return {**rule,
            "trigger": reverseTrigger(rule["trigger"]),
            "action": reverseAction(rule["action"]),
            }


def updateNote(note):
    model = note.note_type()
    modelName = model["name"]
    rules = getUserOption("rules").get(modelName)
    if rules is None:
        return 0, {modelName}
    iter = 0
    while applyRulesToNote(note, rules):
        iter += 1
        if iter >= 10:
            break
    return iter, set()


def updateNid(nid):
    note = mw.col.getNote(nid)
    return updateNote(note)


def updateAll():
    nbChanges = 0
    missings = set()
    for nid in mw.col.findNotes(""):
        nbChangeNid, missingNid = updateNid(nid)
        nbChanges += nbChangeNid
        missings |= missingNid
    return nbChanges, missings