def importSource (self, event = None):
"""Asks the user to choose a file to import source from, then imports into the current story."""
dialog = wx.FileDialog(self, 'Import Source Code', os.getcwd(), '', \
'Text Files (*.txt)|*.txt|Twee Source Code (*.tw)|*.tw', wx.OPEN | wx.FD_CHANGE_DIR)
if dialog.ShowModal() == wx.ID_OK:
try:
# have a TiddlyWiki object parse it for us
source = open(dialog.GetPath(), 'r')
tw = TiddlyWiki()
tw.addTwee(source.read())
source.close()
# add passages for each of the tiddlers the TiddlyWiki saw
if len(tw.tiddlers):
for t in tw.tiddlers:
tiddler = tw.tiddlers[t]
new = self.storyPanel.newWidget(title = tiddler.title, text = tiddler.text, quietly = True)
new.tags = tiddler.tags
self.setDirty(True, 'Import')
else:
dialog = wx.MessageDialog(self, 'No passages were found in this file. Make sure ' + \
'this is a Twee source file.', 'No Passages Found', \
wx.ICON_INFO | wx.OK)
dialog.ShowModal()
except:
self.app.displayError('importing your source code')
def main(argv):
parser = argparse.ArgumentParser(
description="Convert twee source code into SAM source code")
parser.add_argument("-a", "--author", default="twee")
parser.add_argument("-m", "--merge", default="")
parser.add_argument("-p", "--plugins", nargs="*", default=[])
parser.add_argument("-r", "--rss", default="")
parser.add_argument("-t", "--target", default="jonah")
parser.add_argument("sources")
parser.add_argument("destination")
opts = parser.parse_args()
# construct a TW object
tw = TiddlyWiki(opts.author)
# read in a file to be merged
if opts.merge:
with io.open(opts.merge, encoding="utf-8-sig") as f:
tw.addHtml(f.read())
# read source files
sources = glob.glob(opts.sources)
if not sources:
logging.error('twee2sam: no source files specified\n')
sys.exit(2)
for source in sources:
with io.open(source, encoding="utf-8-sig") as f:
tw.addTwee(f.read())
src_dir = os.path.dirname(sources[0])
#
# Parse the file
#
twp = TwParser(tw)
#
# Number the passages
#
passage_indexes = {}
def process_passage_index(passage):
global next_seq
if not passage.title in passage_indexes:
passage_indexes[passage.title] = process_passage_index.next_seq
process_passage_index.next_seq += 1
process_passage_index.next_seq = 0
# 'Start' _must_ be the first script
if not 'Start' in twp.passages:
logging.error('twee2sam: "Start" passage not found.\n')
sys.exit(2)
process_passage_index(twp.passages['Start'])
for passage in twp.passages.values():
process_passage_index(passage)
#
# Generate the file list
#
passage_order = [
psg for psg, idx in sorted(passage_indexes.items(), key=itemgetter(1))
]
def name_to_identifier(s):
return re.sub(r'[^0-9A-Za-z]', '_', s)
def script_name(s):
return name_to_identifier(s) + '.twsam'
if not os.path.exists(opts.destination):
os.makedirs(opts.destination)
with io.open(os.path.join(opts.destination, 'Script.list.txt'),
'w',
encoding="utf-8") as f_list:
for passage_name in passage_order:
passage = twp.passages[passage_name]
f_list.write(u"%s" % script_name(passage.title))
f_list.write(u'\n')
#
# Generate SAM scripts
#
# A is used as a temp var for menu selection
# B is used as a temp var for menu selection
# C and above are available
variables = VariableFactory(2)
image_list = []
music_list = []
for passage in twp.passages.values():
with io.open(os.path.join(opts.destination,
script_name(passage.title)),
'w',
encoding="utf-8") as script:
def check_print():
if check_print.pending:
script.write(u'!\n')
check_print.in_buffer = 0
check_print.pending = False
check_print.pending = False
check_print.in_buffer = 0
def warning(msg):
logging.warning("Warning on \'{0}\': {1}".format(
passage.title, msg))
def out_string(msg):
MAX_LEN = 512
# go through the string and replace characters
msg = ''.join(
map(
lambda x: {
'"': "'",
'[': '{',
']': '}'
}[x] if x in ('"', '[', '{') else x, msg))
msg_len = len(msg)
# Checks for buffer overflow
if check_print.in_buffer + msg_len > MAX_LEN - 1:
warning(
"The text exceeds the maximum buffer size; try to intersperse the text with some <<pause>> macros"
)
remaining = max(0, MAX_LEN - 1 - check_print.in_buffer)
msg = msg[:remaining]
script.write(u'"{0}"'.format(msg))
script.write(u'\n')
check_print.in_buffer += len(msg)
def out_set(cmd):
out_expr(cmd.expr)
script.write(u' ')
target = variables.set_var(cmd.target)
script.write(u"%s\n" % target)
def out_if(cmd):
out_expr(cmd.expr)
script.write(u'[\n')
process_command_list(cmd.children, True)
script.write(u' 0]\n')
def out_print(cmd):
# print a numeric qvariable
out_expr(cmd.expr)
script.write(u'"\#"')
def out_expr(expr):
def var_locator(name):
return variables.get_var(name).replace(':', '')
generated = twexpression.to_sam(expr, var_locator=var_locator)
script.write(u"%s" % generated)
def out_call(cmd):
call_target = None
for k in passage_indexes.keys():
if cmd.target == k:
call_target = passage_indexes[k]
if call_target:
script.write(u"%s" % call_target)
script.write(u'c\n')
def out_jump(cmd):
call_target = None
for k in passage_indexes.keys():
if cmd.target == k:
call_target = passage_indexes[k]
if call_target:
script.write(u"%s" % call_target)
script.write(u'j\n')
# Outputs all the text
links = []
def register_link(cmd, is_conditional):
temp_var = variables.new_temp_var() if is_conditional else None
links.append((cmd, temp_var))
if temp_var:
script.write(u'1%s' % variables.set_var(temp_var))
def process_command_list(commands, is_conditional=False):
for cmd in commands:
if cmd.kind == 'text':
text = cmd.text.strip()
if text:
out_string(cmd.text)
check_print.pending = True
elif cmd.kind == 'print':
out_print(cmd)
elif cmd.kind == 'image':
check_print()
if not cmd.path in image_list:
image_list.append(cmd.path)
script.write(u'{0}i\n'.format(
image_list.index(cmd.path)))
elif cmd.kind == 'link':
register_link(cmd, is_conditional)
out_string(cmd.actual_label())
elif cmd.kind == 'list':
for lcmd in cmd.children:
if lcmd.kind == 'link':
register_link(lcmd, is_conditional)
elif cmd.kind == 'pause':
check_print.pending = True
check_print()
elif cmd.kind == 'set':
out_set(cmd)
elif cmd.kind == 'if':
out_if(cmd)
elif cmd.kind == 'call':
out_call(cmd)
elif cmd.kind == 'jump':
out_jump(cmd)
elif cmd.kind == 'return':
script.write(u'$\n')
elif cmd.kind == 'music':
if not cmd.path in music_list:
music_list.append(cmd.path)
script.write(u'{0}m\n'.format(
music_list.index(cmd.path)))
elif cmd.kind == 'display':
try:
target = twp.passages[cmd.target]
except KeyError:
logging.error(
"Display macro target passage {0} not found!".
format(cmd.target),
file=sys.stderr)
return
process_command_list(target.commands)
process_command_list(passage.commands)
check_print()
# Builds the menu from the links
if links:
# Outputs the options separated by line breaks, max 28 chars per line
for link, temp_var in links:
if temp_var:
script.write(u'{0}['.format(
variables.get_var(temp_var)))
out_string(link.actual_label()[:28] + '\n')
if temp_var:
script.write(u'0]\n')
script.write(u'?A.\n')
check_print.in_buffer = 0
# Outputs the menu destinations
script.write(u'0B.\n')
for link, temp_var in links:
if temp_var:
script.write(u'{0}['.format(
variables.get_var(temp_var)))
if not link.target in passage_indexes:
# TODO: Create a better exception
raise BaseException(
'Link points to a nonexisting passage: "{0}"'.
format(link.target))
script.write(u'A:B:=[{0}j]'.format(
passage_indexes[link.target]))
script.write(u'B:1+B.\n')
if temp_var:
script.write(u'0]\n')
else:
# No links? Generates an infinite loop.
script.write(u'1[1]\n')
#
# Function to copy the files on a list and generate a list file
#
def copy_and_build_list(list_file_name,
file_list,
item_extension,
item_suffix='',
empty_item='blank'):
with io.open(os.path.join(opts.destination, list_file_name),
'w',
encoding="utf-8") as list_file:
for file_path in file_list:
item_name = name_to_identifier(
os.path.splitext(os.path.basename(file_path))[0])
list_file.write("%s%s\n" % (item_name, item_suffix))
shutil.copyfile(
os.path.join(src_dir, file_path),
os.path.join(opts.destination,
'%s.%s' % (item_name, item_extension)))
if not file_list:
list_file.write(u"%s%s\n" % (empty_item, item_suffix))
#
# Copy images and builds the image list
#
copy_and_build_list('Images.txt', image_list, 'png')
#
# Copy music and builds the music list
#
copy_and_build_list('Music.list.txt', music_list, 'epsgmod', '.epsgmod',
'empty')
def main (argv):
parser = argparse.ArgumentParser(description="Convert twee source code into SAM source code")
parser.add_argument("-a", "--author", default="twee")
parser.add_argument("-m", "--merge", default="")
parser.add_argument("-p", "--plugins", nargs="*", default=[])
parser.add_argument("-r", "--rss", default="")
parser.add_argument("-t", "--target", default="jonah")
parser.add_argument("sources")
parser.add_argument("destination")
opts = parser.parse_args()
# construct a TW object
tw = TiddlyWiki(opts.author)
# read in a file to be merged
if opts.merge:
with io.open(opts.merge, encoding="utf-8-sig") as f:
tw.addHtml(f.read())
# read source files
sources = glob.glob(opts.sources)
if not sources:
logging.error('twee2sam: no source files specified\n')
sys.exit(2)
for source in sources:
with io.open(source, encoding="utf-8-sig") as f:
tw.addTwee(f.read())
src_dir = os.path.dirname(sources[0])
#
# Parse the file
#
twp = TwParser(tw)
#
# Number the passages
#
passage_indexes = {}
def process_passage_index(passage):
global next_seq
if not passage.title in passage_indexes:
passage_indexes[passage.title] = process_passage_index.next_seq
process_passage_index.next_seq += 1
process_passage_index.next_seq = 0
# 'Start' _must_ be the first script
if not 'Start' in twp.passages:
logging.error('twee2sam: "Start" passage not found.\n')
sys.exit(2)
process_passage_index(twp.passages['Start'])
for passage in twp.passages.values():
process_passage_index(passage)
#
# Generate the file list
#
passage_order = [psg for psg, idx in sorted(passage_indexes.items(), key=itemgetter(1))]
def name_to_identifier(s):
return re.sub(r'[^0-9A-Za-z]', '_', s)
def script_name(s):
return name_to_identifier(s) + '.twsam'
if not os.path.exists(opts.destination):
os.makedirs(opts.destination)
with io.open(os.path.join(opts.destination, 'Script.list.txt'), 'w', encoding="utf-8") as f_list:
for passage_name in passage_order:
passage = twp.passages[passage_name]
f_list.write(u"%s" % script_name(passage.title))
f_list.write(u'\n')
#
# Generate SAM scripts
#
# A is used as a temp var for menu selection
# B is used as a temp var for menu selection
# C and above are available
variables = VariableFactory(2)
image_list = []
music_list = []
for passage in twp.passages.values():
with io.open(os.path.join(opts.destination, script_name(passage.title)), 'w', encoding="utf-8") as script:
def check_print():
if check_print.pending:
script.write(u'!\n')
check_print.in_buffer = 0
check_print.pending = False
check_print.pending = False
check_print.in_buffer = 0
def warning(msg):
logging.warning("Warning on \'{0}\': {1}".format(passage.title, msg))
def out_string(msg):
MAX_LEN = 512
# go through the string and replace characters
msg = ''.join(map(lambda x: {'"': "'", '[': '{', ']':'}'}[x] if x in ('"','[','{') else x, msg))
msg_len = len(msg)
# Checks for buffer overflow
if check_print.in_buffer + msg_len > MAX_LEN - 1:
warning("The text exceeds the maximum buffer size; try to intersperse the text with some <<pause>> macros")
remaining = max(0, MAX_LEN - 1 - check_print.in_buffer)
msg = msg[:remaining]
script.write(u'"{0}"'.format(msg))
script.write(u'\n')
check_print.in_buffer += len(msg)
def out_set(cmd):
out_expr(cmd.expr)
script.write(u' ')
target = variables.set_var(cmd.target)
script.write(u"%s\n" % target)
def out_if(cmd):
out_expr(cmd.expr)
script.write(u'[\n')
process_command_list(cmd.children, True)
script.write(u' 0]\n')
def out_print(cmd):
# print a numeric qvariable
out_expr(cmd.expr)
script.write(u'"\#"')
def out_expr(expr):
def var_locator(name):
return variables.get_var(name).replace(':', '')
generated = twexpression.to_sam(expr, var_locator = var_locator)
script.write(u"%s" % generated)
def out_call(cmd):
call_target = None
for k in passage_indexes.keys():
if cmd.target == k:
call_target = passage_indexes[k]
if call_target:
script.write(u"%s" % call_target)
script.write(u'c\n')
# Outputs all the text
links = []
def register_link(cmd, is_conditional):
temp_var = variables.new_temp_var() if is_conditional else None
links.append((cmd, temp_var))
if temp_var:
script.write(u'1%s' % variables.set_var(temp_var))
def process_command_list(commands, is_conditional=False):
for cmd in commands:
if cmd.kind == 'text':
text = cmd.text.strip()
if text:
out_string(text)
check_print.pending = True
elif cmd.kind == 'print':
out_print(cmd)
elif cmd.kind == 'image':
check_print()
if not cmd.path in image_list:
image_list.append(cmd.path)
script.write(u'{0}i\n'.format(image_list.index(cmd.path)))
elif cmd.kind == 'link':
register_link(cmd, is_conditional)
out_string(cmd.actual_label())
elif cmd.kind == 'list':
for lcmd in cmd.children:
if lcmd.kind == 'link':
register_link(lcmd, is_conditional)
elif cmd.kind == 'pause':
check_print.pending = True
check_print()
elif cmd.kind == 'set':
out_set(cmd)
elif cmd.kind == 'if':
out_if(cmd)
elif cmd.kind == 'call':
out_call(cmd)
elif cmd.kind == 'return':
script.write(u'$\n')
elif cmd.kind == 'music':
if not cmd.path in music_list:
music_list.append(cmd.path)
script.write(u'{0}m\n'.format(music_list.index(cmd.path)))
elif cmd.kind == 'display':
try:
target = twp.passages[cmd.target]
except KeyError:
logging.error("Display macro target passage {0} not found!".format(cmd.target), file=sys.stderr)
return
process_command_list(target.commands)
process_command_list(passage.commands)
check_print()
# Builds the menu from the links
if links:
# Outputs the options separated by line breaks, max 28 chars per line
for link, temp_var in links:
if temp_var:
script.write(u'{0}['.format(variables.get_var(temp_var)))
out_string(link.actual_label()[:28] + '\n')
if temp_var:
script.write(u'0]\n')
script.write(u'?A.\n')
check_print.in_buffer = 0
# Outputs the menu destinations
script.write(u'0B.\n');
for link, temp_var in links:
if temp_var:
script.write(u'{0}['.format(variables.get_var(temp_var)))
if not link.target in passage_indexes:
# TODO: Create a better exception
raise BaseException('Link points to a nonexisting passage: "{0}"'.format(link.target))
script.write(u'A:B:=[{0}j]'.format(passage_indexes[link.target]))
script.write(u'B:1+B.\n')
if temp_var:
script.write(u'0]\n')
else:
# No links? Generates an infinite loop.
script.write(u'1[1]\n')
#
# Function to copy the files on a list and generate a list file
#
def copy_and_build_list(list_file_name, file_list, item_extension, item_suffix = '', empty_item = 'blank'):
with io.open(os.path.join(opts.destination, list_file_name), 'w', encoding="utf-8") as list_file:
for file_path in file_list:
item_name = name_to_identifier(os.path.splitext(os.path.basename(file_path))[0])
list_file.write("%s%s\n" % (item_name, item_suffix))
shutil.copyfile(os.path.join(src_dir, file_path), os.path.join(opts.destination, '%s.%s' % (item_name, item_extension)))
if not file_list:
list_file.write(u"%s%s\n" % (empty_item, item_suffix))
#
# Copy images and builds the image list
#
copy_and_build_list('Images.txt', image_list, 'png')
#
# Copy music and builds the music list
#
copy_and_build_list('Music.list.txt', music_list, 'epsgmod', '.epsgmod', 'empty')
def main (argv):
# defaults
author = 'twee'
target = 'jonah'
merge = rss_output = ''
plugins = []
# read command line switches
try:
opts, args = getopt.getopt(argv, 'a:m:p:r:t:', ['author=', 'merge=', 'plugins=', 'rss=', 'target='])
except getopt.GetoptError:
usage()
sys.exit(2)
for opt, arg in opts:
if (opt in ('-a', '--author')):
author = arg
elif (opt in ('-m', '--merge')):
merge = arg
elif (opt in ('-p', '--plugins')):
plugins = arg.split(',')
elif (opt in ('-r', '--rss')):
rss_output = arg
elif (opt in ('-t', '--target')):
target = arg
# construct a TW object
tw = TiddlyWiki(author)
# read in a file to be merged
if merge != '':
file = open(merge)
tw.addHtml(file.read())
file.close()
# read source files
sources = []
for arg in args:
for file in glob.glob(arg):
sources.append(file)
if len(sources) == 0:
print 'twee: no source files specified\n'
sys.exit(2)
for source in sources:
file = open(source)
tw.addTwee(file.read())
file.close()
# generate RSS if requested
if rss_output != '':
rss_file = open(rss_output, 'w')
tw.toRss().write_xml(rss_file)
rss_file.close()
# output the target header
# if (target != 'none') and (target != 'plugin'):
# file = open(scriptPath + os.sep + 'targets' + os.sep + target \
# + os.sep + 'header.html')
# print(file.read())
# file.close()
# the tiddlers
print TwParser(tw)
