Skip to content
Snippets Groups Projects
Commit 12d72047 authored by Jan Mach's avatar Jan Mach
Browse files

Refactoring: Moved the RuleTreeTraverser and PrintingTreeTraverser from... 

Refactoring: Moved the RuleTreeTraverser and PrintingTreeTraverser from pynspect.rules into separate module called pynspect.traversers.
parent 4bf47a43
No related branches found
No related tags found
No related merge requests found
Show whitespace changes
Inline Side-by-side
doc/_doclib/api.rst
+ 1
− 0
View file @ 12d72047
......@@ -17,4 +17,5 @@ API
api_pynspect.lexer
api_pynspect.gparser
api_pynspect.rules
api_pynspect.traversers
api_pynspect.filters
doc/_doclib/api_pynspect.traversers.rst 0 → 100644
+ 9
− 0
View file @ 12d72047
.. _section-api-pynspect-traversers:
pynspect.traversers module
================================================================================
.. automodule:: pynspect.traversers
:show-inheritance:
:members:
:undoc-members:
pynspect/filters.py
+ 2
− 1
View file @ 12d72047
......@@ -68,7 +68,8 @@ import calendar
import ipranges
from pynspect.rules import IPV4Rule, IPV6Rule, IntegerRule, FloatRule, NumberRule, VariableRule,\
LogicalBinOpRule, UnaryOperationRule, ComparisonBinOpRule, MathBinOpRule, ListRule,\
RuleTreeTraverser, FilteringRuleException
FilteringRuleException
from pynspect.traversers import RuleTreeTraverser
from pynspect.jpath import jpath_values
......
pynspect/rules.py
+ 2
− 256
View file @ 12d72047
......@@ -39,12 +39,12 @@ parsing string rules using :py:mod:`pynspect.gparser`.
Working with rule tree is then done via objects implementing rule tree
traverser interface:
* :py:class:`RuleTreeTraverser`
* :py:class:`pynspect.traversers.RuleTreeTraverser`
There is a simple example implementation of rule tree traverser capable of
printing rule tree into a formated string:
* :py:class:`PrintingTreeTraverser`
* :py:class:`pynspect.traversers.PrintingTreeTraverser`
Rule evaluation
^^^^^^^^^^^^^^^
......@@ -477,252 +477,6 @@ class UnaryOperationRule(OperationRule):
#-------------------------------------------------------------------------------
class RuleTreeTraverser():
"""
Base class and interface definition for all rule tree traversers. This is a
mandatory interface that is required for an object to be able to traverse
through given :py:class:`pynspect.rules.Rule` tree.
"""
def ipv4(self, rule, **kwargs):
"""
Callback method for rule tree traversing. Will be called at proper time
from :py:class:`pynspect.rules.IPV4Rule.traverse` method.
:param pynspect.rules.Rule rule: Reference to rule.
:param dict kwargs: Optional callback arguments.
"""
raise NotImplementedError()
def ipv6(self, rule, **kwargs):
"""
Callback method for rule tree traversing. Will be called at proper time
from :py:class:`pynspect.rules.IPV6Rule.traverse` method.
:param pynspect.rules.Rule rule: Reference to rule.
:param dict kwargs: Optional callback arguments.
"""
raise NotImplementedError()
def integer(self, rule, **kwargs):
"""
Callback method for rule tree traversing. Will be called at proper time
from :py:class:`pynspect.rules.IntegerRule.traverse` method.
:param pynspect.rules.Rule rule: Reference to rule.
:param dict kwargs: Optional callback arguments.
"""
raise NotImplementedError()
def float(self, rule, **kwargs):
"""
Callback method for rule tree traversing. Will be called at proper time
from :py:class:`pynspect.rules.FloatRule.traverse` method.
:param pynspect.rules.Rule rule: Reference to rule.
:param dict kwargs: Optional callback arguments.
"""
raise NotImplementedError()
def constant(self, rule, **kwargs):
"""
Callback method for rule tree traversing. Will be called at proper time
from :py:class:`pynspect.rules.ConstantRule.traverse` method.
:param pynspect.rules.Rule rule: Reference to rule.
:param dict kwargs: Optional callback arguments.
"""
raise NotImplementedError()
def variable(self, rule, **kwargs):
"""
Callback method for rule tree traversing. Will be called at proper time
from :py:class:`pynspect.rules.VariableRule.traverse` method.
:param pynspect.rules.Rule rule: Reference to rule.
:param dict kwargs: Optional callback arguments.
"""
raise NotImplementedError()
def list(self, rule, **kwargs):
"""
Callback method for rule tree traversing. Will be called at proper time
from :py:class:`pynspect.rules.ListRule.traverse` method.
:param pynspect.rules.Rule rule: Reference to rule.
:param dict kwargs: Optional callback arguments.
"""
raise NotImplementedError()
def binary_operation_logical(self, rule, left, right, **kwargs):
"""
Callback method for rule tree traversing. Will be called at proper time
from :py:class:`pynspect.rules.LogicalBinOpRule.traverse` method.
:param pynspect.rules.Rule rule: Reference to rule.
:param left: Left operand for operation.
:param right: right operand for operation.
:param dict kwargs: Optional callback arguments.
"""
raise NotImplementedError()
def binary_operation_comparison(self, rule, left, right, **kwargs):
"""
Callback method for rule tree traversing. Will be called at proper time
from :py:class:`pynspect.rules.ComparisonBinOpRule.traverse` method.
:param pynspect.rules.Rule rule: Reference to rule.
:param left: Left operand for operation.
:param right: right operand for operation.
:param dict kwargs: Optional callback arguments.
"""
raise NotImplementedError()
def binary_operation_math(self, rule, left, right, **kwargs):
"""
Callback method for rule tree traversing. Will be called at proper time
from :py:class:`pynspect.rules.MathBinOpRule.traverse` method.
:param pynspect.rules.Rule rule: Reference to rule.
:param left: Left operand for operation.
:param right: right operand for operation.
:param dict kwargs: Optional callback arguments.
"""
raise NotImplementedError()
def unary_operation(self, rule, right, **kwargs):
"""
Callback method for rule tree traversing. Will be called at proper time
from :py:class:`pynspect.rules.UnaryOperationRule.traverse` method.
:param pynspect.rules.Rule rule: Reference to rule.
:param right: right operand for operation.
:param dict kwargs: Optional callback arguments.
"""
raise NotImplementedError()
class PrintingTreeTraverser(RuleTreeTraverser):
"""
Demonstation of simple rule tree traverser - printing traverser.
"""
def ipv4(self, rule, **kwargs):
"""
Callback method for rule tree traversing. Will be called at proper time
from :py:class:`pynspect.rules.IPV4Rule.traverse` method.
:param pynspect.rules.Rule rule: Reference to rule.
:param dict kwargs: Optional callback arguments.
"""
return "IPV4({})".format(rule.value)
def ipv6(self, rule, **kwargs):
"""
Callback method for rule tree traversing. Will be called at proper time
from :py:class:`pynspect.rules.IPV6Rule.traverse` method.
:param pynspect.rules.Rule rule: Reference to rule.
:param dict kwargs: Optional callback arguments.
"""
return "IPV6({})".format(rule.value)
def integer(self, rule, **kwargs):
"""
Callback method for rule tree traversing. Will be called at proper time
from :py:class:`pynspect.rules.IntegerRule.traverse` method.
:param pynspect.rules.Rule rule: Reference to rule.
:param dict kwargs: Optional callback arguments.
"""
return "INTEGER({})".format(rule.value)
def float(self, rule, **kwargs):
"""
Callback method for rule tree traversing. Will be called at proper time
from :py:class:`pynspect.rules.FloatRule.traverse` method.
:param pynspect.rules.Rule rule: Reference to rule.
:param dict kwargs: Optional callback arguments.
"""
return "FLOAT({})".format(rule.value)
def constant(self, rule, **kwargs):
"""
Callback method for rule tree traversing. Will be called at proper time
from :py:class:`pynspect.rules.ConstantRule.traverse` method.
:param pynspect.rules.Rule rule: Reference to rule.
:param dict kwargs: Optional callback arguments.
"""
return "CONSTANT({})".format(rule.value)
def variable(self, rule, **kwargs):
"""
Callback method for rule tree traversing. Will be called at proper time
from :py:class:`pynspect.rules.VariableRule.traverse` method.
:param pynspect.rules.Rule rule: Reference to rule.
:param dict kwargs: Optional callback arguments.
"""
return "VARIABLE({})".format(rule.value)
def list(self, rule, **kwargs):
"""
Callback method for rule tree traversing. Will be called at proper time
from :py:class:`pynspect.rules.ListRule.traverse` method.
:param pynspect.rules.Rule rule: Reference to rule.
:param dict kwargs: Optional callback arguments.
"""
return "LIST({})".format(', '.join([str(v) for v in rule.value]))
def binary_operation_logical(self, rule, left, right, **kwargs):
"""
Callback method for rule tree traversing. Will be called at proper time
from :py:class:`pynspect.rules.LogicalBinOpRule.traverse` method.
:param pynspect.rules.Rule rule: Reference to rule.
:param left: Left operand for operation.
:param right: right operand for operation.
:param dict kwargs: Optional callback arguments.
"""
return "LOGBINOP({};{};{})".format(rule.operation, left, right)
def binary_operation_comparison(self, rule, left, right, **kwargs):
"""
Callback method for rule tree traversing. Will be called at proper time
from :py:class:`pynspect.rules.ComparisonBinOpRule.traverse` method.
:param pynspect.rules.Rule rule: Reference to rule.
:param left: Left operand for operation.
:param right: right operand for operation.
:param dict kwargs: Optional callback arguments.
"""
return "COMPBINOP({};{};{})".format(rule.operation, left, right)
def binary_operation_math(self, rule, left, right, **kwargs):
"""
Callback method for rule tree traversing. Will be called at proper time
from :py:class:`pynspect.rules.MathBinOpRule.traverse` method.
:param pynspect.rules.Rule rule: Reference to rule.
:param left: Left operand for operation.
:param right: right operand for operation.
:param dict kwargs: Optional callback arguments.
"""
return "MATHBINOP({};{};{})".format(rule.operation, left, right)
def unary_operation(self, rule, right, **kwargs):
"""
Callback method for rule tree traversing. Will be called at proper time
from :py:class:`pynspect.rules.UnaryOperationRule.traverse` method.
:param pynspect.rules.Rule rule: Reference to rule.
:param right: right operand for operation.
:param dict kwargs: Optional callback arguments.
"""
return "UNOP({};{})".format(rule.operation, right)
#
# Perform the demonstration.
#
......@@ -762,11 +516,3 @@ if __name__ == "__main__":
RULE_UNOP = UnaryOperationRule('OP_NOT', RULE_VAR)
print("STR: {}".format(str(RULE_UNOP)))
print("REPR: {}".format(repr(RULE_UNOP)))
print("\n* Traverser usage:")
RULE_TRAVERSER = PrintingTreeTraverser()
print("{}".format(RULE_BINOP_L.traverse(RULE_TRAVERSER)))
print("{}".format(RULE_BINOP_C.traverse(RULE_TRAVERSER)))
print("{}".format(RULE_BINOP_M.traverse(RULE_TRAVERSER)))
print("{}".format(RULE_BINOP.traverse(RULE_TRAVERSER)))
print("{}".format(RULE_UNOP.traverse(RULE_TRAVERSER)))
pynspect/tests/test_rules.py
+ 1
− 32
View file @ 12d72047
......@@ -23,8 +23,7 @@ import unittest
from pprint import pformat
from pynspect.rules import IPV4Rule, IPV6Rule, IntegerRule, FloatRule, VariableRule, ConstantRule,\
LogicalBinOpRule, UnaryOperationRule, ComparisonBinOpRule, MathBinOpRule, ListRule,\
PrintingTreeTraverser
LogicalBinOpRule, UnaryOperationRule, ComparisonBinOpRule, MathBinOpRule, ListRule
#-------------------------------------------------------------------------------
......@@ -84,36 +83,6 @@ class TestPynspectRules(unittest.TestCase):
self.assertEqual(repr(rule_unop), "UNOP(OP_NOT VARIABLE('Test'))")
class TestPynspectPrintingTreeTraverser(unittest.TestCase):
"""
Unit test class for testing the PrintingTreeTraverser from :py:mod:`pynspect.rules` module.
"""
def setUp(self):
self.tvs = PrintingTreeTraverser()
def test_01_basic(self):
"""
Demonstrate and test the PrintingTreeTraverser object.
"""
self.maxDiff = None
rule_binop_l = LogicalBinOpRule('OP_OR', VariableRule("Test"), IntegerRule(10))
self.assertEqual(rule_binop_l.traverse(self.tvs), 'LOGBINOP(OP_OR;VARIABLE(Test);INTEGER(10))')
rule_binop_c = ComparisonBinOpRule('OP_GT', VariableRule("Test"), IntegerRule(15))
self.assertEqual(rule_binop_c.traverse(self.tvs), 'COMPBINOP(OP_GT;VARIABLE(Test);INTEGER(15))')
rule_binop_m = MathBinOpRule('OP_PLUS', VariableRule("Test"), IntegerRule(10))
self.assertEqual(rule_binop_m.traverse(self.tvs), 'MATHBINOP(OP_PLUS;VARIABLE(Test);INTEGER(10))')
rule_binop = LogicalBinOpRule('OP_OR', ComparisonBinOpRule('OP_GT', MathBinOpRule('OP_PLUS', VariableRule("Test"), IntegerRule(10)), IntegerRule(20)), ComparisonBinOpRule('OP_LT', VariableRule("Test"), IntegerRule(5)))
self.assertEqual(rule_binop.traverse(self.tvs), 'LOGBINOP(OP_OR;COMPBINOP(OP_GT;MATHBINOP(OP_PLUS;VARIABLE(Test);INTEGER(10));INTEGER(20));COMPBINOP(OP_LT;VARIABLE(Test);INTEGER(5)))')
rule_unop = UnaryOperationRule('OP_NOT', VariableRule("Test"))
self.assertEqual(rule_unop.traverse(self.tvs), 'UNOP(OP_NOT;VARIABLE(Test))')
#-------------------------------------------------------------------------------
......
pynspect/tests/test_traversers.py 0 → 100644
+ 69
− 0
View file @ 12d72047
#!/usr/bin/env python3
# -*- coding: utf-8 -*-
#-------------------------------------------------------------------------------
# This file is part of Pynspect package (https://pypi.python.org/pypi/pynspect).
# Originally part of Mentat system (https://mentat.cesnet.cz/).
#
# Copyright (C) since 2016 CESNET, z.s.p.o (http://www.ces.net/).
# Copyright (C) since 2016 Jan Mach <honza.mach.ml@gmail.com>
# Use of this source is governed by the MIT license, see LICENSE file.
#-------------------------------------------------------------------------------
"""
Unit test module for testing the :py:mod:`pynspect.traversers` module.
"""
__author__ = "Jan Mach <jan.mach@cesnet.cz>"
__credits__ = "Pavel Kácha <pavel.kacha@cesnet.cz>"
import unittest
from pynspect.rules import IntegerRule, VariableRule, LogicalBinOpRule, UnaryOperationRule,\
ComparisonBinOpRule, MathBinOpRule
from pynspect.traversers import PrintingTreeTraverser
#-------------------------------------------------------------------------------
# NOTE: Sorry for the long lines in this file. They are deliberate, because the
# assertion permutations are (IMHO) more readable this way.
#-------------------------------------------------------------------------------
class TestPynspectPrintingTreeTraverser(unittest.TestCase):
"""
Unit test class for testing the PrintingTreeTraverser from :py:mod:`pynspect.rules` module.
"""
def setUp(self):
self.tvs = PrintingTreeTraverser()
def test_01_basic(self):
"""
Demonstrate and test the PrintingTreeTraverser object.
"""
self.maxDiff = None
rule_binop_l = LogicalBinOpRule('OP_OR', VariableRule("Test"), IntegerRule(10))
self.assertEqual(rule_binop_l.traverse(self.tvs), 'LOGBINOP(OP_OR;VARIABLE(Test);INTEGER(10))')
rule_binop_c = ComparisonBinOpRule('OP_GT', VariableRule("Test"), IntegerRule(15))
self.assertEqual(rule_binop_c.traverse(self.tvs), 'COMPBINOP(OP_GT;VARIABLE(Test);INTEGER(15))')
rule_binop_m = MathBinOpRule('OP_PLUS', VariableRule("Test"), IntegerRule(10))
self.assertEqual(rule_binop_m.traverse(self.tvs), 'MATHBINOP(OP_PLUS;VARIABLE(Test);INTEGER(10))')
rule_binop = LogicalBinOpRule('OP_OR', ComparisonBinOpRule('OP_GT', MathBinOpRule('OP_PLUS', VariableRule("Test"), IntegerRule(10)), IntegerRule(20)), ComparisonBinOpRule('OP_LT', VariableRule("Test"), IntegerRule(5)))
self.assertEqual(rule_binop.traverse(self.tvs), 'LOGBINOP(OP_OR;COMPBINOP(OP_GT;MATHBINOP(OP_PLUS;VARIABLE(Test);INTEGER(10));INTEGER(20));COMPBINOP(OP_LT;VARIABLE(Test);INTEGER(5)))')
rule_unop = UnaryOperationRule('OP_NOT', VariableRule("Test"))
self.assertEqual(rule_unop.traverse(self.tvs), 'UNOP(OP_NOT;VARIABLE(Test))')
#-------------------------------------------------------------------------------
if __name__ == '__main__':
unittest.main()
pynspect/traversers.py 0 → 100644
+ 305
− 0
View file @ 12d72047
#!/usr/bin/env python3
# -*- coding: utf-8 -*-
#-------------------------------------------------------------------------------
# This file is part of Pynspect package (https://pypi.python.org/pypi/pynspect).
# Originally part of Mentat system (https://mentat.cesnet.cz/).
#
# Copyright (C) since 2016 CESNET, z.s.p.o (http://www.ces.net/).
# Copyright (C) since 2016 Jan Mach <honza.mach.ml@gmail.com>
# Use of this source is governed by the MIT license, see LICENSE file.
#-------------------------------------------------------------------------------
"""
This module contains implementation of object representations of rule tree traversers,
that are supposed to be used to work with rule tree structures.
The base implementation and interface definition is can be found in following class:
* :py:class:`pynspect.traversers.RuleTreeTraverser`
There is also a simple example implementation of rule tree traverser capable of
printing rule tree into a formated string:
* :py:class:`pynspect.traversers.PrintingTreeTraverser`
"""
__author__ = "Jan Mach <jan.mach@cesnet.cz>"
__credits__ = "Pavel Kácha <pavel.kacha@cesnet.cz>, Andrea Kropáčová <andrea.kropacova@cesnet.cz>"
class RuleTreeTraverser():
"""
Base class and interface definition for all rule tree traversers. This is a
mandatory interface that is required for an object to be able to traverse
through given :py:class:`pynspect.rules.Rule` tree.
"""
def ipv4(self, rule, **kwargs):
"""
Callback method for rule tree traversing. Will be called at proper time
from :py:class:`pynspect.rules.IPV4Rule.traverse` method.
:param pynspect.rules.Rule rule: Reference to rule.
:param dict kwargs: Optional callback arguments.
"""
raise NotImplementedError()
def ipv6(self, rule, **kwargs):
"""
Callback method for rule tree traversing. Will be called at proper time
from :py:class:`pynspect.rules.IPV6Rule.traverse` method.
:param pynspect.rules.Rule rule: Reference to rule.
:param dict kwargs: Optional callback arguments.
"""
raise NotImplementedError()
def integer(self, rule, **kwargs):
"""
Callback method for rule tree traversing. Will be called at proper time
from :py:class:`pynspect.rules.IntegerRule.traverse` method.
:param pynspect.rules.Rule rule: Reference to rule.
:param dict kwargs: Optional callback arguments.
"""
raise NotImplementedError()
def float(self, rule, **kwargs):
"""
Callback method for rule tree traversing. Will be called at proper time
from :py:class:`pynspect.rules.FloatRule.traverse` method.
:param pynspect.rules.Rule rule: Reference to rule.
:param dict kwargs: Optional callback arguments.
"""
raise NotImplementedError()
def constant(self, rule, **kwargs):
"""
Callback method for rule tree traversing. Will be called at proper time
from :py:class:`pynspect.rules.ConstantRule.traverse` method.
:param pynspect.rules.Rule rule: Reference to rule.
:param dict kwargs: Optional callback arguments.
"""
raise NotImplementedError()
def variable(self, rule, **kwargs):
"""
Callback method for rule tree traversing. Will be called at proper time
from :py:class:`pynspect.rules.VariableRule.traverse` method.
:param pynspect.rules.Rule rule: Reference to rule.
:param dict kwargs: Optional callback arguments.
"""
raise NotImplementedError()
def list(self, rule, **kwargs):
"""
Callback method for rule tree traversing. Will be called at proper time
from :py:class:`pynspect.rules.ListRule.traverse` method.
:param pynspect.rules.Rule rule: Reference to rule.
:param dict kwargs: Optional callback arguments.
"""
raise NotImplementedError()
def binary_operation_logical(self, rule, left, right, **kwargs):
"""
Callback method for rule tree traversing. Will be called at proper time
from :py:class:`pynspect.rules.LogicalBinOpRule.traverse` method.
:param pynspect.rules.Rule rule: Reference to rule.
:param left: Left operand for operation.
:param right: right operand for operation.
:param dict kwargs: Optional callback arguments.
"""
raise NotImplementedError()
def binary_operation_comparison(self, rule, left, right, **kwargs):
"""
Callback method for rule tree traversing. Will be called at proper time
from :py:class:`pynspect.rules.ComparisonBinOpRule.traverse` method.
:param pynspect.rules.Rule rule: Reference to rule.
:param left: Left operand for operation.
:param right: right operand for operation.
:param dict kwargs: Optional callback arguments.
"""
raise NotImplementedError()
def binary_operation_math(self, rule, left, right, **kwargs):
"""
Callback method for rule tree traversing. Will be called at proper time
from :py:class:`pynspect.rules.MathBinOpRule.traverse` method.
:param pynspect.rules.Rule rule: Reference to rule.
:param left: Left operand for operation.
:param right: right operand for operation.
:param dict kwargs: Optional callback arguments.
"""
raise NotImplementedError()
def unary_operation(self, rule, right, **kwargs):
"""
Callback method for rule tree traversing. Will be called at proper time
from :py:class:`pynspect.rules.UnaryOperationRule.traverse` method.
:param pynspect.rules.Rule rule: Reference to rule.
:param right: right operand for operation.
:param dict kwargs: Optional callback arguments.
"""
raise NotImplementedError()
class PrintingTreeTraverser(RuleTreeTraverser):
"""
Demonstation of simple rule tree traverser - printing traverser.
"""
def ipv4(self, rule, **kwargs):
"""
Callback method for rule tree traversing. Will be called at proper time
from :py:class:`pynspect.rules.IPV4Rule.traverse` method.
:param pynspect.rules.Rule rule: Reference to rule.
:param dict kwargs: Optional callback arguments.
"""
return "IPV4({})".format(rule.value)
def ipv6(self, rule, **kwargs):
"""
Callback method for rule tree traversing. Will be called at proper time
from :py:class:`pynspect.rules.IPV6Rule.traverse` method.
:param pynspect.rules.Rule rule: Reference to rule.
:param dict kwargs: Optional callback arguments.
"""
return "IPV6({})".format(rule.value)
def integer(self, rule, **kwargs):
"""
Callback method for rule tree traversing. Will be called at proper time
from :py:class:`pynspect.rules.IntegerRule.traverse` method.
:param pynspect.rules.Rule rule: Reference to rule.
:param dict kwargs: Optional callback arguments.
"""
return "INTEGER({})".format(rule.value)
def float(self, rule, **kwargs):
"""
Callback method for rule tree traversing. Will be called at proper time
from :py:class:`pynspect.rules.FloatRule.traverse` method.
:param pynspect.rules.Rule rule: Reference to rule.
:param dict kwargs: Optional callback arguments.
"""
return "FLOAT({})".format(rule.value)
def constant(self, rule, **kwargs):
"""
Callback method for rule tree traversing. Will be called at proper time
from :py:class:`pynspect.rules.ConstantRule.traverse` method.
:param pynspect.rules.Rule rule: Reference to rule.
:param dict kwargs: Optional callback arguments.
"""
return "CONSTANT({})".format(rule.value)
def variable(self, rule, **kwargs):
"""
Callback method for rule tree traversing. Will be called at proper time
from :py:class:`pynspect.rules.VariableRule.traverse` method.
:param pynspect.rules.Rule rule: Reference to rule.
:param dict kwargs: Optional callback arguments.
"""
return "VARIABLE({})".format(rule.value)
def list(self, rule, **kwargs):
"""
Callback method for rule tree traversing. Will be called at proper time
from :py:class:`pynspect.rules.ListRule.traverse` method.
:param pynspect.rules.Rule rule: Reference to rule.
:param dict kwargs: Optional callback arguments.
"""
return "LIST({})".format(', '.join([str(v) for v in rule.value]))
def binary_operation_logical(self, rule, left, right, **kwargs):
"""
Callback method for rule tree traversing. Will be called at proper time
from :py:class:`pynspect.rules.LogicalBinOpRule.traverse` method.
:param pynspect.rules.Rule rule: Reference to rule.
:param left: Left operand for operation.
:param right: right operand for operation.
:param dict kwargs: Optional callback arguments.
"""
return "LOGBINOP({};{};{})".format(rule.operation, left, right)
def binary_operation_comparison(self, rule, left, right, **kwargs):
"""
Callback method for rule tree traversing. Will be called at proper time
from :py:class:`pynspect.rules.ComparisonBinOpRule.traverse` method.
:param pynspect.rules.Rule rule: Reference to rule.
:param left: Left operand for operation.
:param right: right operand for operation.
:param dict kwargs: Optional callback arguments.
"""
return "COMPBINOP({};{};{})".format(rule.operation, left, right)
def binary_operation_math(self, rule, left, right, **kwargs):
"""
Callback method for rule tree traversing. Will be called at proper time
from :py:class:`pynspect.rules.MathBinOpRule.traverse` method.
:param pynspect.rules.Rule rule: Reference to rule.
:param left: Left operand for operation.
:param right: right operand for operation.
:param dict kwargs: Optional callback arguments.
"""
return "MATHBINOP({};{};{})".format(rule.operation, left, right)
def unary_operation(self, rule, right, **kwargs):
"""
Callback method for rule tree traversing. Will be called at proper time
from :py:class:`pynspect.rules.UnaryOperationRule.traverse` method.
:param pynspect.rules.Rule rule: Reference to rule.
:param right: right operand for operation.
:param dict kwargs: Optional callback arguments.
"""
return "UNOP({};{})".format(rule.operation, right)
#-------------------------------------------------------------------------------
#
# Perform the demonstration.
#
if __name__ == "__main__":
from pynspect.rules import IntegerRule, VariableRule, LogicalBinOpRule,\
UnaryOperationRule, ComparisonBinOpRule, MathBinOpRule
# Create couple of test rules.
RULE_VAR = VariableRule("Test")
RULE_INTEGER = IntegerRule(15)
RULE_BINOP_L = LogicalBinOpRule('OP_OR', RULE_VAR, RULE_INTEGER)
RULE_BINOP_C = ComparisonBinOpRule('OP_GT', RULE_VAR, RULE_INTEGER)
RULE_BINOP_M = MathBinOpRule('OP_PLUS', RULE_VAR, RULE_INTEGER)
RULE_BINOP = LogicalBinOpRule('OP_OR', ComparisonBinOpRule('OP_GT', MathBinOpRule('OP_PLUS', VariableRule("Test"), IntegerRule(10)), IntegerRule(20)), ComparisonBinOpRule('OP_LT', VariableRule("Test"), IntegerRule(5)))
RULE_UNOP = UnaryOperationRule('OP_NOT', RULE_VAR)
print("* Traverser usage:")
RULE_TRAVERSER = PrintingTreeTraverser()
print("{}".format(RULE_BINOP_L.traverse(RULE_TRAVERSER)))
print("{}".format(RULE_BINOP_C.traverse(RULE_TRAVERSER)))
print("{}".format(RULE_BINOP_M.traverse(RULE_TRAVERSER)))
print("{}".format(RULE_BINOP.traverse(RULE_TRAVERSER)))
print("{}".format(RULE_UNOP.traverse(RULE_TRAVERSER)))
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment