value = ""

weight = 100

displayInformation = None

additionalFields = []

iconURL = ""

entityType = "Phrase"

def __init__(self,eT=None,v=None):

if (eT is not None):

self.entityType = eT

if (v is not None):

self.value = sanitise(v)

self.additionalFields = []

self.displayInformation = None

def setType(self,eT=None):

if (eT is not None):

self.entityType = eT

def setValue(self,eV=None):

if (eV is not None):

self.value = sanitise(eV)

def setWeight(self,w=None):

if (w is not None):

self.weight = w

def setDisplayInformation(self,di=None):

if (di is not None):

self.displayInformation = di

def addAdditionalFields(self,fieldName=None,displayName=None,matchingRule=False,value=None):

self.additionalFields.append([sanitise(fieldName),sanitise(displayName),matchingRule,sanitise(value)])

def setIconURL(self,iU=None):

if (iU is not None):

self.iconURL = iU

def setLinkColor(self,color):

self.addAdditionalFields('link#maltego.link.color','LinkColor','',color)

def setLinkStyle(self,style):

self.addAdditionalFields('link#maltego.link.style','LinkStyle','',style)

def setLinkThickness(self,thick):

self.addAdditionalFields('link#maltego.link.thickness','Thickness','',str(thick))

def setLinkLabel(self,label):

self.addAdditionalFields('link#maltego.link.label','Label','',label)

def setBookmark(self,bookmark):

self.addAdditionalFields('bookmark#','Bookmark','',bookmark)

def setNote(self,note):

self.addAdditionalFields('notes#','Notes','',note)

def returnEntity(self):

print "<Entity Type=\"" + str(self.entityType) + "\">"

print "<Value>" + str(self.value) + "</Value>"

print "<Weight>" + str(self.weight) + "</Weight>"

if (self.displayInformation is not None):

print "<DisplayInformation><Label Name=\"\" Type=\"text/html\"><![CDATA[" + str(self.displayInformation) + "]]></Label></DisplayInformation>"

if (len(self.additionalFields) > 0):

print "<AdditionalFields>"

for i in range(len(self.additionalFields)):

if (str(self.additionalFields[i][2]) <> "strict"):

print "<Field Name=\"" + str(self.additionalFields[i][0]) + "\" DisplayName=\"" + str(self.additionalFields[i][1]) + "\">" + str(self.additionalFields[i][3]) + "</Field>"

else:

print "<Field MatchingRule=\"" + str(self.additionalFields[i][2]) + "\" Name=\"" + str(self.additionalFields[i][0]) + "\" DisplayName=\"" + str(self.additionalFields[i][1]) + "\">" + str(self.additionalFields[i][3]) + "</Field>"

print "</AdditionalFields>"

if (len(self.iconURL) > 0):

print "<IconURL>" + self.iconURL + "</IconURL>"

entities = []

exceptions = []

UIMessages = []

values = {}

def __init__(self):

values = {}

value = None

def parseArguments(self,argv):

if (argv[1] is not None):

self.value = argv[1]

if (len(argv) > 2):

if (argv[2] is not None):

vars = argv[2].split('#')

for x in range(0,len(vars)):

vars_values = vars[x].split('=',1)

if (len(vars_values) == 2):

self.values[vars_values[0]] = vars_values[1]

def getValue(self):

if (self.value is not None):

return self.value

def getVar(self,varName):

if (varName in self.values.keys()):

if (self.values[varName] is not None):

return self.values[varName]

def addEntity(self,enType,enValue):

me = MaltegoEntity(enType,enValue)

self.addEntityToMessage(me)

return self.entities[len(self.entities)-1]

def addEntityToMessage(self,maltegoEntity):

self.entities.append(maltegoEntity)

def addUIMessage(self,message,messageType="Inform"):

self.UIMessages.append([messageType,message])

def addException(self,exceptionString):

self.exceptions.append(exceptionString)

def throwExceptions(self):

print "<MaltegoMessage>"

print "<MaltegoTransformExceptionMessage>"

print "<Exceptions>"

for i in range(len(self.exceptions)):

print "<Exception>" + self.exceptions[i] + "</Exception>"

print "</Exceptions>"

print "</MaltegoTransformExceptionMessage>"

print "</MaltegoMessage>"

exit()

def returnOutput(self):

print "<MaltegoMessage>"

print "<MaltegoTransformResponseMessage>"

print "<Entities>"

for i in range(len(self.entities)):

self.entities[i].returnEntity()

print "</Entities>"

print "<UIMessages>"

for i in range(len(self.UIMessages)):

print "<UIMessage MessageType=\"" + self.UIMessages[i][0] + "\">" + self.UIMessages[i][1] + "</UIMessage>"

print "</UIMessages>"

print "</MaltegoTransformResponseMessage>"

print "</MaltegoMessage>"

def writeSTDERR(self,msg):

sys.stderr.write(str(msg))

def heartbeat(self):

self.writeSTDERR("+")

def progress(self,percent):

self.writeSTDERR("%" + str(percent))

def debug(self,msg):

replace_these = ["&",">","<"]

replace_with = ["&amp;","&gt;","&lt;"]

tempvalue = value

for i in range(0,len(replace_these)):

tempvalue = tempvalue.replace(replace_these[i],replace_with[i])

return tempvalue