From 68b10cd143432cc65047a9b4d5125d42ab34b211 Mon Sep 17 00:00:00 2001 From: allfro Date: Sun, 7 Apr 2013 21:25:43 -0400 Subject: [PATCH] Made a `MaltegoTransformRequestMessage` object --- src/canari/maltego/message.py | 14 +++++++++++++- 1 file changed, 13 insertions(+), 1 deletion(-) diff --git a/src/canari/maltego/message.py b/src/canari/maltego/message.py index 4fcedd1..f17db7c 100644 --- a/src/canari/maltego/message.py +++ b/src/canari/maltego/message.py @@ -12,7 +12,7 @@ __copyright__ = 'Copyright 2012, Canari Project' __credits__ = [] __license__ = 'GPL' -__version__ = '0.2' +__version__ = '0.3' __maintainer__ = 'Nadeem Douba' __email__ = 'ndouba@gmail.com' __status__ = 'Development' @@ -84,6 +84,18 @@ class MaltegoException(MaltegoElement, Exception): self.text = message if not isinstance(message, basestring) else message +class MaltegoTransformRequestMessage(object): + + def __init__(self, value, fields, parameters, limits=None): + self.value = value + self.fields = fields + self.params = parameters + if limits is None: + self.limits = dict(soft=500, hard=10000) + else: + self.limits = dict(soft=limits.get('SoftLimit', 500), hard=limits.get('HardLimit', 10000)) + + @XMLSubElement(name='UIMessages', propname='uimessages', type=XSSubElementType.List) @XMLSubElement(name='Entities', propname='entities', type=XSSubElementType.List) class MaltegoTransformResponseMessage(MaltegoElement): -- 2.45.1