Now you can add scripts from puzzle.py

This commit is contained in:
Neale Pickett 2020-02-22 17:25:46 -06:00 committed by John Donaldson
parent 333f728e2a
commit 8d8db58f73
1 changed files with 7 additions and 5 deletions

View File

@ -260,15 +260,12 @@ class Puzzle:
elif key == 'script':
stream = open(val, 'rb')
# Make sure this shows up in the header block of the HTML output.
self.files[val] = PuzzleFile(stream, val, visible=False)
self.scripts.append(val)
self.add_script_stream(stream, val)
elif key == "scripts" and isinstance(val, list):
for script in val:
stream = open(script, "rb")
self.files[script] = PuzzleFile(stream, script, visible=False)
self.scripts.append(script)
self.add_script_stream(stream, val)
elif key == "objective":
self.objective = val
@ -322,6 +319,11 @@ class Puzzle:
self.add_stream(stream, name, visible)
return stream
def add_script_stream(self, stream, name):
# Make sure this shows up in the header block of the HTML output.
self.files[name] = PuzzleFile(stream, name, visible=False)
self.scripts.append(name)
def add_stream(self, stream, name=None, visible=True):
if name is None:
name = self.random_hash()