Skip to content

Instantly share code, notes, and snippets.

Embed
What would you like to do?
from core.modules import NodeCreator, RelationshipMaker
from py2neo import Graph
class EmailTC(NodeCreator):
def __init__(self):
NodeCreator.__init__(self)
self.Node_Label = 'Email'
self.Plugin_name = 'ThreatCrowd'
self.node_key_field = 'email_address'
class DomainTC(NodeCreator):
def __init__(self):
NodeCreator.__init__(self)
self.Node_Label = 'Domain'
self.Plugin_name = 'ThreatCrowd'
self.node_key_field = 'domain'
class Sub_DomainTC(NodeCreator):
def __init__(self):
NodeCreator.__init__(self)
self.Node_Label = 'Sub-Domain'
self.Plugin_name = 'ThreatCrowd'
self.node_key_field = 'sub-domain'
class IP_addressTC(NodeCreator):
def __init__(self):
NodeCreator.__init__(self)
self.Node_Label = 'IP_address'
self.Plugin_name = 'ThreatCrowd'
self.node_key_field = 'ip_address'
class Hashes_TC(NodeCreator):
def __init__(self):
NodeCreator.__init__(self)
self.Node_Label = 'Hashes'
self.Plugin_name = 'ThreatCrowd'
self.node_key_field = 'Hashes'
#Relationships
class Domain_to_IP_address(RelationshipMaker):
def __init__(self):
RelationshipMaker.__init__(self)
self.Relationship_Label = "Resolves"
self.to_node_type = "IP_address"
self.from_node_type = "Domain"
class Domain_Sub_domain_Relationship(RelationshipMaker):
def __init__(self):
RelationshipMaker.__init__(self)
self.Relationship_Label = "Hosts"
self.to_node_type = "Sub-Domain"
self.from_node_type = "Domain"
class Email_Domain_TC_Relationship(RelationshipMaker):
def __init__(self):
RelationshipMaker.__init__(self)
self.Relationship_Label = "Owns"
self.to_node_type = "Domain"
self.from_node_type = "Email"
class Domains_Hashes_Relationship(RelationshipMaker):
def __init__(self):
RelationshipMaker.__init__(self)
self.Relationship_Label = "Associated"
self.to_node_type = "Hashes"
self.from_node_type = "Domain"
class Hashes_IP_Address_Relationship(RelationshipMaker):
def __init__(self):
NodeCreator.__init__(self)
self.Relationship_Label = "Associated"
self.to_node_type = "Hashes"
self.from_node_type = "IP_address"
class Hashes_Sub_Domain_Relationship(RelationshipMaker):
def __int__(self):
RelationshipMaker.__init__(self)
self.Relationship_Label = "Associated"
self.to_node_type = "Hashes"
self.from_node_type = "sub-domain"
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
You can’t perform that action at this time.