Current File : //lib/python3.6/site-packages/tracer/resources/rules.py
#-*- coding: utf-8 -*-
# rules.py
# Manager for rules file
#
# Copyright (C) 2016 Jakub Kadlcik
#
# This copyrighted material is made available to anyone wishing to use,
# modify, copy, or redistribute it subject to the terms and conditions of
# the GNU General Public License v.2, or (at your option) any later version.
# This program is distributed in the hope that it will be useful, but WITHOUT
# ANY WARRANTY expressed or implied, including the implied warranties of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General
# Public License for more details. You should have received a copy of the
# GNU General Public License along with this program; if not, write to the
# Free Software Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
# 02110-1301, USA.
#
from __future__ import absolute_import
from xml.dom import minidom
from xml.parsers.expat import ExpatError
from tracer.paths import DATA_DIR, USER_CONFIG_DIRS
from tracer.resources.exceptions import PathNotFound, TracerError
from os.path import dirname
class Rules(object):
DEFINITIONS = map(lambda x: x + "/rules.xml", [DATA_DIR] + USER_CONFIG_DIRS)
ACTIONS = {
"CALL-PARENT" : "call-parent",
"RETURN" : "return",
}
_DEFAULT_ACTION = ACTIONS["CALL-PARENT"]
_rules = None
@staticmethod
def find(app_name):
if not Rules._rules:
Rules._load_definitions()
for rule in Rules._rules:
if rule.name == app_name:
return rule
@staticmethod
def all():
if not Rules._rules:
Rules._load_definitions()
return Rules._rules
@staticmethod
def _load_definitions():
Rules._rules = []
for file in Rules.DEFINITIONS:
try: Rules._load(file);
except PathNotFound as ex:
if not dirname(file) in USER_CONFIG_DIRS:
raise ex
@staticmethod
def _load(file):
try:
with open(file, "r") as f:
xmldoc = minidom.parseString(f.read())
except IOError:
raise PathNotFound('DATA_DIR')
except ExpatError as ex:
msg = "Unable to parse {0}\nHint: {1}".format(file, ex)
raise TracerError(msg)
for rules in xmldoc.getElementsByTagName("rules"):
for rule in rules.getElementsByTagName("rule"):
attrs = dict(rule.attributes.items())
r = Rule(attrs)
if r in Rules._rules:
i = Rules._rules.index(r)
Rules._rules[i].update(r)
else:
r.setdefault('action', Rules._DEFAULT_ACTION)
Rules._rules.append(r)
class Rule(object):
"""
Represent the rule defined in `rules.xml`
Attributes
----------
name : str
action : str
See `Rules.ACTIONS` for possible values
"""
_attributes = None
def __init__(self, attributes_dict):
self._attributes = attributes_dict
def __eq__(self, other):
return isinstance(other, Rule) and self.name == other.name
def __ne__(self, other):
return not self.__eq__(other)
def __getattr__(self, item):
return self._attributes[item]
def __len__(self):
return len(self._attributes)
def __contains__(self, item):
return item in self._attributes
def __str__(self):
return "<Rule: " + self._attributes["name"] + ">"
def __repr__(self):
return self.__str__()
def setdefault(self, key, value):
self._attributes.setdefault(key, value)
def update(self, values):
if isinstance(values, Rule):
values = values._attributes
self._attributes.update(values)
Mini Shell