__init__.py

# Injecting js into stats page
from anki.hooks import wrap

from aqt.stats import NewDeckStats
import os.path
from aqt import mw

addon_dir = os.path.dirname(__file__)

def new_refresh(self: NewDeckStats):
    with open(f"{addon_dir}/stats.min.js") as f: # Putting this inside the function allows you to rebuild the page without restarting anki
        innerJs = f.read()
    with open(f"{addon_dir}/stats.min.css") as f:
        innerCss = f.read()

    config = mw.addonManager.getConfig(__name__)
    other = {
        "rollover": mw.col.get_preferences().scheduling.rollover,
        "learn_ahead_secs": mw.col.get_preferences().scheduling.learn_ahead_secs,
        "deck_configs": {conf["id"]: conf for conf in mw.col.decks.all_config()},
        "deck_config_ids": {deck["id"]: deck.get("conf", None) for deck in mw.col.decks.all()},
        "days_elapsed": mw.col.sched.today
    }
    setVars = (
        f"const css = `{innerCss}`;" 
        f"const SSEconfig = {json.dumps(config)};"
        f"const SSEother = {json.dumps(other)};"
    )
    self.form.web.eval(setVars + innerJs)

NewDeckStats.refresh = wrap(NewDeckStats.refresh, new_refresh, "after")

# Search endpoint
from flask import request, Response
import json

from aqt.mediasrv import post_handlers
from anki.utils import int_time

def card_search() -> bytes:
    search = request.data
    try:
        return Response(str(list(mw.col.find_cards(search))))
    except: 
        return None

post_handlers["cardSearch"] = card_search

CARD_COLUMNS = ["id","nid","did","ord","mod","usn","type","queue","due","ivl","factor","reps","lapses","left","odue","odid","flags","data"]

def card_data() -> bytes:
    cards = request.data.strip(b"[]").decode()
    cardData = mw.col.db.all(f"SELECT * FROM cards WHERE id IN ({cards})")
    cardData = [{k: v for k, v in zip(CARD_COLUMNS, a)} for a in cardData]
    return Response(json.dumps(cardData))

post_handlers["cardData"] = card_data

REVLOG_COLUMNS = ["id", "cid", "ease", "ivl", "lastIvl", "time", "factor", "type"]
DAY_SECONDS = 60 * 60 * 24

def revlogs() -> bytes:
    req = json.loads(request.data)
    cards = ','.join(str(cid) for cid in req["cids"])
    day_range = req["day_range"]

    if day_range > 0:
        rollover = mw.col.get_preferences().scheduling.rollover
        today = (int_time() - (rollover * 60 * 60)) / DAY_SECONDS
        lower_limit = (today - day_range) * DAY_SECONDS * 1000 # +4 ?!?!?
    else:
        lower_limit = 0

    revlogs = mw.col.db.all(f"""
    SELECT 
        {','.join(REVLOG_COLUMNS)} 
    FROM revlog 
    WHERE
        cid IN ({cards}) 
        AND revlog.id > ({lower_limit}) 
    ORDER BY revlog.id
    """)
    revlogs = [{k.replace("revlog.", ""): v for k, v in zip(REVLOG_COLUMNS, a)} for a in revlogs]
    return Response(json.dumps(revlogs))


post_handlers["revlogs"] = revlogs