gamebookformat/sections.py

import random
import sys

class Section:
    def __init__(self, name, text):
        self.name = name
        self.text = text
        self.tags = set()

    def set_tags(self, tags):
        self.tags = set(tags)

    def __repr__(self):
        return "Section(%s, %s, %s)" % (repr(self.name), repr(self.text),
                                        repr(self.tags))

    def format(self, references):
        return self.text % references

class ShuffledSections:
    def __init__(self, as_list, from_nr, to_nr, from_name):
        self.as_list = as_list
        self.from_nr = from_nr
        self.to_nr = to_nr
        self.from_name = from_name
        self.name_to_nr = {}
        for n in from_name:
            self.name_to_nr[n] = to_nr[from_name[n]]

class Book:
    def __init__(self):
        self.sections = []
        self.from_name = {}
        self.nr_sections = {}
        self.max = 0

    def add(self, section):
        if section.name in self.from_name:
            raise Exception('Duplicate section names (%s) not allowed.' %
                            section.name)
        self.sections.append(section)
        self.from_name[section.name] = section
        if len(self.sections) > self.max:
            self.max = len(self.sections)

    def force_section_nr(self, name, nr):
        self.nr_sections[nr] = name
        if nr > self.max:
            self.max = nr

    def shuffle(self):
        as_list = [None]
        from_nr = {}
        to_nr = {}
        shuffled = self.sections[:]
        for p in self.nr_sections.values():
            shuffled.remove(self.from_name[p])
        random.shuffle(shuffled)
        for nr in range(1, self.max + 1):
            if self.nr_sections.has_key(nr):
                section = self.from_name[self.nr_sections[nr]]
            elif len(shuffled):
                section = shuffled.pop()
            else:
                section = None
            as_list.append(section)
            from_nr[nr] = section
            if section:
                to_nr[section] = nr
        return ShuffledSections(as_list, from_nr, to_nr, self.from_name.copy())