aly / bandersnatch.py

#!/usr/bin/env python3 from random import choice import json from sys import stderr with open("bandersnatch.json") as f: bandersnatch = json.load(f) with open("segmentmap.json") as f: smap = json.load(f) initial_state = { "p_sp": True, "p_tt": True, "p_8a": False, "p_td": True, "p_cs": False, "p_w1": False, "p_2b": False, "p_3j": False, "p_pt": False, "p_cd": False, "p_cj": False, "p_sj": False, "p_sj2": False, "p_tud": False, "p_lsd": False, "p_vh": False, "p_3l": False, "p_3s": False, "p_3z": False, "p_ps": "n", "p_wb": False, "p_kd": False, "p_bo": False, "p_5v": False, "p_pc": "n", "p_sc": False, "p_ty": False, "p_cm": False, "p_pr": False, "p_3ad": False, "p_s3af": False, "p_nf": False, "p_np": False, "p_ne": False, "p_pp": False, "p_tp": False, "p_bup": False, "p_be": False, "p_pe": False, "p_pae": False, "p_te": False, "p_snt": False, "p_8j": False, "p_8d": False, "p_8m": False, "p_8q": False, "p_8s": False, "p_8v": False, "p_vs": "n", "p_scs": False, "p_3ab": False, "p_3ac": False, "p_3aj": False, "p_3ah": False, "p_3ak": False, "p_3al": False, "p_3af": False, "p_5h": False, "p_5ac": False, "p_5ag": False, "p_5ad": False, "p_6c": False, "length": 0 } state = dict(initial_state) info = bandersnatch["videos"]["80988062"]["interactiveVideoMoments"]["value"] moments = info["momentsBySegment"] preconditions = info["preconditions"] segmentGroups = info["segmentGroups"] thumbnails = info["choicePointNavigatorMetadata"]["choicePointsMetadata"]["choicePoints"] segmentMap = {} for segmentList in moments.values(): for _segment in segmentList: for _choice in _segment.get("choices", []): if "segmentId" in _choice and "text" in _choice: segmentMap[_choice["segmentId"]] = _choice["text"].title() def msToTS(ms): s,ms = divmod(ms,1000) m,s = divmod(s,60) h,m = divmod(m,60) return "{:02d}:{:02d}:{:02d}.{:03d}".format(h,m,s,ms) def conditionHandler(cond): global state if not cond: return True if cond[0] == "persistentState": return state[cond[1]] if cond[0] == "not": return not all(conditionHandler(c) for c in cond[1:]) if cond[0] == "and": return all(conditionHandler(c) for c in cond[1:]) if cond[0] == "eql": return conditionHandler(cond[1]) == cond[2] if cond[0] == "or": return any(conditionHandler(c) for c in cond[1:]) def groupHandler(group, segment=None): out = [] if segment: group.append(segment) for item in group: if type(item) is str and conditionHandler( preconditions.get(item,[]) ): out.append(item) if type(item) is dict: if "segmentGroup" in item: out += groupHandler(segmentGroups[item["segmentGroup"]]) if "precondition" in item: if conditionHandler( preconditions.get(item["precondition"],[]) ): out.append(item["segment"]) # print("out="+repr(out),file=stderr) return out def followTheStory(segment): global state global history possibilities = [] if segment in moments: m = moments[segment] for moment in m: if moment["type"] == "notification:playbackImpression": state.update( moment.get("impressionData",{}).get("data", {}).get("persistent", {}) ) if moment["type"] == "scene:cs_bs": for option in moment["choices"]: state.update( option.get("impressionData",{}).get("data", {}).get("persistent", {}) ) if "segmentId" in option: p = groupHandler([option["segmentId"]]) elif "sg" in option: p = groupHandler(segmentGroups[option["sg"]]) elif moment["trackingInfo"]["optionType"] == "fakeOption": continue else: raise Exception(option["id"]) possibilities += p if moment["type"] == "notification:action": possibilities.append(segment) if segment in segmentGroups: possibilities += groupHandler(segmentGroups[segment]) # print("poss="+repr(possibilities),file=stderr) if not possibilities: # raise Exception("hoi") possibilities += groupHandler(segmentGroups["respawnOptions"]) return choice(possibilities) def get_segment_info(segment): _ = thumbnails.get(segment, {}) return {"id": segment, "url": _["image"]["styles"]["backgroundImage"][4:-1] if "image" in _ else "", "caption": _.get("description", "No caption"), "chose": segmentMap.get(segment, "No caption ({})".format(segment))} def bandersnatch(): global state concat, options = [], [] state = dict(initial_state) current_segment = "1A" while True: state["length"] += smap[current_segment]["endTimeMs"] - smap[current_segment]["startTimeMs"] if current_segment[:3].lower() == "0cr": options.append(get_segment_info(current_segment)) concat.append(current_segment) options.append(get_segment_info("IDNT")) concat.append('IDNT') state["length"] += 10 break options.append(get_segment_info(current_segment)) concat.append(current_segment) current_segment = followTheStory(current_segment) if current_segment is None: break return concat, options, msToTS(state["length"]) if __name__ == "__main__": print(bandersnatch(),file=stderr)

#!/usr/bin/env python from flask import Flask, render_template_string, send_file from bandersnatch import bandersnatch TEMPLATE = """ <!doctype html> <html> <head> <style> body { background-color: black; color: white; font-family: sans-serif } label { background-color: black; } .segment { width:472px; height:266px; } div.segment { margin:30px } video.segment { } </style> <title>Bandersnatch</title> </head> <body> <h3>{{ length }}</h3> <textarea>{% for segment in concat %} file '{{ segment }}.mkv' {% endfor %}</textarea> <div style='display: flex; flex-wrap: wrap;'> {% for option in options %} <div class='segment' id='{{ option.id }}' style='background-image: url({{ option.url }})'> <video class='segment' id='{{ option.id }}-vid' controls {% if option.url %}poster='{{ option.url }}' preload='none'{% else %}preload='metadata'{% endif %}> <source src='https://s3.home.b303.me/bandersnatch/{{ option.id }}.mkv{% if not option.url %}#t=0.5{% endif %}' type='video/mp4'> </video> <label for='{{ option.id }}'>{% if option.chose != "No caption (1A)" %}<span class='chose'>Chose {{ option.chose }}</span> &gt;{% endif %} <span class='caption'>{{ option.caption }}</span></label> </div> {% endfor %} </div> </body> </html> """ app = Flask(__name__) @app.route("/<segment>.mkv") def mkv(segment): return send_file("{}.mkv".format(segment)) @app.route("/") def index(): concat, options, length = bandersnatch() return render_template_string(TEMPLATE, concat=concat, options=options, length=length) if __name__ == "__main__": app.run(debug=True, host="0.0.0.0", port=24572)