Exemplo n.º 1
0
Arquivo: loadXML.py Projeto: AKJ/sage
def load_triggers_from_xml(file):
	xmlfile = parse(file)
	group = xmlfile.getroot()

	new_group = {}

	# Ensure these triggers have a group
	if group.tag != 'group':
		print "load_triggers_from_xml: File %s not contained in a <group>" % file
		return False

	# Gracefully handle group name being omitted
	try:
		new_group['name'] = group.attrib['name']
	except KeyError:
		print "load_triggers_from_xml: Group name not defined in %s" % file
		return False

	# Optionally apply group enabled/disabled
	try:
		new_group['enabled'] = group.attrib['enabled']
		new_group['enabled'] = str_to_bool(new_group['enabled'])
	except KeyError:
		new_group['enabled'] = False

	core.add_trigger_group(new_group['name'], new_group['enabled'])

	# Load triggers for group
	for trigger in group:
		new_trigger = {
			'name': None,
			'type': None,
			'pattern': None,
			'enabled': True,
			'function': [],
			'pfunction': [],
			'pdisable_trigger': [],
			'affliction': [],
			'cure': [],
			'bcure': [],
			'symptom': [],
			'gag': [],
			'enable_group': [],
			'disable_group': [],
			'enable_trigger': [],
			'disable_trigger': [],
			'ondef': [],
			'undef': [],
			'send': [],
			'deflist': None
		}

		for node in trigger:
			if node.tag == 'name':
				new_trigger['name'] = node.text

			elif node.tag == 'type':
				new_trigger['type'] = node.text

			elif node.tag == 'pattern':
				new_trigger['pattern'] = node.text

			elif node.tag == 'enabled':
				new_trigger['enabled'] = str_to_bool(node.text)

			elif node.tag == 'function':
				new_trigger['function'].append(node.text)

			elif node.tag == 'pfunction':
				new_trigger['pfunction'].append(node.text)

			elif node.tag == 'pdisable_trigger':
				new_trigger['pdisable_trigger'].append(node.text)

			elif node.tag == 'affliction':
				new_trigger['affliction'].append(node.text)

			elif node.tag == 'symptom':
				new_trigger['symptom'].append(node.text)

			elif node.tag == 'cure':
				new_trigger['cure'].append(node.text)

			elif node.tag == 'bcure':
				new_trigger['bcure'].append(node.text)

			elif node.tag == 'gag':
				new_trigger['gag'].append(node.text)

			elif node.tag == 'send':
				new_trigger['send'].append(node.text)

			elif node.tag == 'enable_group':
				new_trigger['enable_group'].append(node.text)

			elif node.tag == 'disable_group':
				new_trigger['disable_group'].append(node.text)

			elif node.tag == 'enable_trigger':
				new_trigger['enable_trigger'].append(node.text)

			elif node.tag == 'disable_trigger':
				new_trigger['disable_trigger'].append(node.text)

			elif node.tag == 'undef':
				new_trigger['undef'].append(node.text)

			elif node.tag == 'ondef':
				new_trigger['ondef'].append(node.text)

			elif node.tag == 'deflist':
				new_trigger['deflist'] = node.text

		# Validate <trigger>
		keys = new_trigger.keys()

		# There has to be a better way.....
		if 'name' not in keys or 'type' not in keys or 'pattern' not in keys:
			print "load_triggers_from_xml: Required trigger element missing in %s" % file
			return False

		# Validate trigger types
		if new_trigger['type'] not in ('exact', 'substring', 'regex', 'startswith', 'endswith'):
			print "load_triggers_from_xml: Invalid trigger type in %s" % file
			return False


		# Figure out enabled
		try:
			if new_trigger['enabled'] is False:
				pass
		except KeyError:
			new_trigger['enabled'] = False

		functions = []

		if len(new_trigger['affliction']) > 0:
			for affliction in new_trigger['affliction']:
				functions.append('affliction ' + affliction)

		if len(new_trigger['symptom']) > 0:
			for symptom in new_trigger['symptom']:
				functions.append('symptom ' + symptom)

		if len(new_trigger['cure']) > 0:
			for cure in new_trigger['cure']:
				functions.append('cure ' + cure)

		if len(new_trigger['bcure']) > 0:
			for bcure in new_trigger['bcure']:
				functions.append('bcure ' + bcure)

		if len(new_trigger['gag']) > 0:
			for gag in new_trigger['gag']:
				functions.append('gag ' + gag)

		if len(new_trigger['send']) > 0:
			for send in new_trigger['send']:
				functions.append('send ' + send)

		if len(new_trigger['enable_group']) > 0:
			for group in new_trigger['enable_group']:
				functions.append('enable_group ' + group)

		if len(new_trigger['disable_group']) > 0:
			for group in new_trigger['disable_group']:
				functions.append('disable_group ' + group)

		if len(new_trigger['enable_trigger']) > 0:
			for trigger in new_trigger['enable_trigger']:
				functions.append('enable_trigger ' + trigger)

		if len(new_trigger['disable_trigger']) > 0:
			for trigger in new_trigger['disable_trigger']:
				functions.append('disable_trigger ' + trigger)

		if len(new_trigger['ondef']) > 0:
			for defence in new_trigger['ondef']:
				functions.append('ondef ' + defence)

		if len(new_trigger['undef']) > 0:
			for defence in new_trigger['undef']:
				functions.append('undef ' + defence)

		if new_trigger['deflist'] is not None:
			functions.append('deflist ' + new_trigger['deflist'])

		if len(new_trigger['function']) > 0:
			for function in new_trigger['function']:
				functions.append(function)

		if len(new_trigger['pfunction']) > 0:
			for function in new_trigger['pfunction']:
				functions.append('pfunction ' + function)

		if len(new_trigger['pdisable_trigger']) > 0:
			for trigger in new_trigger['pdisable_trigger']:
				functions.append('pdisable_trigger ' + trigger)

		if len(functions) == 0:
			print "load_triggers_from_xml: No function defined for '%s' in %s" % new_trigger['name'], file
			#print "load_triggers_from_xml: No function defined for generic trigger in %s" % file

		core.add_trigger(
			new_trigger['name'],
			new_group['name'],
			new_trigger['type'],
			new_trigger['pattern'],
			functions,
			new_trigger['enabled'],
		)

	# Done!
	return new_group['name']