sync_hook.py

from aqt.gui_hooks import sync_will_start, sync_did_finish
from aqt import QMessageBox, QTimer
from .reschedule import reschedule
from .configuration import Config
from .utils import *


def create_comparelog(local_rids: List[int]) -> None:
    local_rids.extend(
        [id for id in mw.col.db.list("SELECT id FROM revlog")]
    )


def auto_reschedule(local_rids: List[int]):
    if len(local_rids) == 0:
        return
    config = Config()
    config.load()
    if not config.auto_reschedule_after_sync:
        return

    local_rid_string = ",".join([str(local_rid) for local_rid in local_rids])

    # exclude entries where ivl == lastIvl: they indicate a dynamic deck without rescheduling
    remote_reviewed_cids = [cid for cid in mw.col.db.list(
            f"SELECT DISTINCT cid FROM revlog WHERE id NOT IN ({local_rid_string}) and ivl != lastIvl"
        )
    ]

    remote_reviewed_cid_string = ",".join([str(cid) for cid in remote_reviewed_cids])
    rescheduled_nids = [nid for nid in mw.col.db.list(
        f"""SELECT DISTINCT nid 
            FROM cards 
            WHERE id IN ({remote_reviewed_cid_string})
        """
        )
    ]

    filtered_nid_string = ",".join([str(nid) for nid in rescheduled_nids])
    fut = reschedule(None, recent=False, filter_flag=True, filtered_cids=set(remote_reviewed_cids), filtered_nid_string=filtered_nid_string)

    # wait for reschedule to finish

    parent_widget = mw.app.activeWindow() or mw
    msgBox = QMessageBox(parent_widget)
    msgBox.setStandardButtons(QMessageBox.StandardButton.NoButton)
    msgBox.setText(fut.result())
    timer = QTimer()
    timer.singleShot(1000, msgBox.accept)
    msgBox.exec()


def init_sync_hook():
    local_rids = []

    sync_will_start.append(lambda: create_comparelog(local_rids))
    sync_did_finish.append(lambda: auto_reschedule(local_rids))