From: allfro <ndouba@gmail.com>
Date: Mon, 8 Apr 2013 01:25:43 +0000 (-0400)
Subject: Made a `MaltegoTransformRequestMessage` object
X-Git-Url: https://git.nothing2do.fr/?a=commitdiff_plain;h=68b10cd143432cc65047a9b4d5125d42ab34b211;p=get-hack-src.git

Made a `MaltegoTransformRequestMessage` object
---

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):