import re
import model
def parse(file):
destination = None
messages = []
condition = None
master_condition = None
for line in file:
match = re.search(r'messages -> ([A-Za-z_0-9]+) {', line)
if match:
if condition:
master_condition = condition
condition = None
destination = match.group(1)
continue
if line.startswith('#'):
if line.startswith('#if '):
condition = line.rstrip()[4:]
elif line.startswith('#endif'):
condition = None
continue
match = re.search(r'([A-Za-z_0-9]+)\((.*?)\)(?:(?:\s+->\s+)\((.*?)\))?(?:\s+(.*))?', line)
if match:
name, parameters_string, reply_parameters_string, attributes_string = match.groups()
if parameters_string:
parameters = parse_parameters_string(parameters_string)
for parameter in parameters:
parameter.condition = condition
else:
parameters = []
attributes = parse_attributes_string(attributes_string)
if reply_parameters_string:
reply_parameters = parse_parameters_string(reply_parameters_string)
for reply_parameter in reply_parameters:
reply_parameter.condition = condition
elif reply_parameters_string == '':
reply_parameters = []
else:
reply_parameters = None
messages.append(model.Message(name, parameters, reply_parameters, attributes, condition))
return model.MessageReceiver(destination, messages, master_condition)
def parse_attributes_string(attributes_string):
if not attributes_string:
return None
return attributes_string.split()
def parse_parameters_string(parameters_string):
parameters = []
for parameter_string in parameters_string.split(', '):
match = re.search(r'\s*(?:\[(?P<attributes>.*?)\]\s+)?(?P<type_and_name>.*)', parameter_string)
attributes_string, type_and_name_string = match.group('attributes', 'type_and_name')
parameter_type, parameter_name = type_and_name_string.rsplit(' ', 1)
parameters.append(model.Parameter(type=parameter_type, name=parameter_name, attributes=parse_attributes_string(attributes_string)))
return parameters