-import unittest
from plomrogue.errors import ArgError
from plomrogue.mapping import YX
def __init__(self, game=None):
self.game = game
+ self.string_options = {}
def tokenize(self, msg):
"""Parse msg string into tokens.
x = get_axis_position_from_argument('X', tokens[1])
return YX(y, x)
- def parse(self, msg):
+ def parse(self, msg, replace_newline=True):
"""Parse msg as call to function, return function with args tuple.
Respects function signature defined in function's .argtypes attribute.
+
+ Refuses messages with any but a small list of acceptable characters.
+
"""
+ import string
+ if replace_newline:
+ msg = msg.replace('\n', ' ') # Inserted by some tablet keyboards.
+ legal_chars = string.digits + string.ascii_letters +\
+ string.punctuation + ' ' + 'ÄäÖöÜüߧ' + 'éèáàô' + '–…'
+ for c in msg:
+ if not c in legal_chars:
+ raise ArgError('Command/message contains illegal character(s), '
+ 'may only contain ones of: %s' % legal_chars)
tokens = self.tokenize(msg)
if len(tokens) == 0:
return None, ()
tmpl_tokens = signature.split()
if len(tmpl_tokens) != len(args_tokens):
raise ArgError('Number of arguments (' + str(len(args_tokens)) +
- ') not expected number (' + str(len(tmpl_tokens))
- + ').')
+ ') not expected number (' + str(len(tmpl_tokens)) +
+ ').')
args = []
string_string = 'string'
for i in range(len(tmpl_tokens)):
if not arg.isdigit() or int(arg) < 1:
raise ArgError('Argument must be positive integer.')
args += [int(arg)]
+ elif tmpl == 'int':
+ try:
+ args += [int(arg)]
+ except ValueError:
+ raise ArgError('Argument must be integer.')
+ elif tmpl == 'bool':
+ if not arg.isdigit() or int(arg) not in (0, 1):
+ raise ArgError('Argument must be 0 or 1.')
+ args += [bool(int(arg))]
elif tmpl == 'char':
try:
ord(arg)
elif tmpl == string_string:
args += [arg]
elif tmpl[:len(string_string) + 1] == string_string + ':':
- if not hasattr(self.game, 'get_string_options'):
- raise ArgError('No string option directory.')
string_option_type = tmpl[len(string_string) + 1:]
- options = self.game.get_string_options(string_option_type)
- if options is None:
- raise ArgError('Unknown string option type.')
+ if not string_option_type in self.string_options.keys():
+ raise ArgError('Unknown string option type: %s' % string_option_type)
+ options = self.string_options[string_option_type]
if arg not in options:
msg = 'Argument #%s must be one of: %s' % (i + 1, options)
raise ArgError(msg)