File size: 2,397 Bytes
a695f69 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 | """
ShufflerController — mirrors the JS makeShuffler() no-repeat logic.
The seen-list for each named pool is kept inside a dict that lives in
gr.BrowserState (localStorage). The state dict shape:
{
"jam": {"seen": ["...", ...]},
"tt_easy": {"seen": [...]},
"tt_medium": {"seen": [...]},
"tt_hard": {"seen": [...]},
"imp_all": {"seen": [...]},
"imp_0": {"seen": [...]}, # per-category index
...
"iv_all": {"seen": [...]},
"iv_0": {"seen": [...]},
...
}
"""
import json
import random
from pydantic import BaseModel
from models.session_models import ShufflerState
def _serialise(item) -> str:
if isinstance(item, str):
return item
if isinstance(item, BaseModel):
return json.dumps(item.model_dump(), ensure_ascii=False, sort_keys=True)
return json.dumps(item, ensure_ascii=False)
def next_item(pool: list, key: str, raw_state: dict) -> tuple:
"""
Pick the next unseen item from pool.
Returns (item, updated_raw_state).
"""
if not pool:
return None, raw_state
state_dict = raw_state.get(key, {}) if isinstance(raw_state, dict) else {}
shuffler = ShufflerState.model_validate(state_dict)
seen_set = set(shuffler.seen)
unseen = [i for i, item in enumerate(pool) if _serialise(item) not in seen_set]
if not unseen:
# Full cycle complete — reset
shuffler.seen = []
seen_set = set()
unseen = list(range(len(pool)))
idx = random.choice(unseen)
picked = pool[idx]
shuffler.seen.append(_serialise(picked))
updated = dict(raw_state) if isinstance(raw_state, dict) else {}
updated[key] = shuffler.model_dump()
return picked, updated
def reset_key(key: str, raw_state: dict) -> dict:
"""Reset the seen list for a single key."""
updated = dict(raw_state) if isinstance(raw_state, dict) else {}
updated[key] = ShufflerState().model_dump()
return updated
def imp_pool_key(cat: str, categories: list) -> str:
if cat == "All":
return "imp_all"
idx = next((i for i, c in enumerate(categories) if c.category == cat), 0)
return f"imp_{idx}"
def iv_pool_key(cat: str, categories: list) -> str:
if cat == "All":
return "iv_all"
idx = next((i for i, c in enumerate(categories) if c.category == cat), 0)
return f"iv_{idx}"
|