File tree Expand file tree Collapse file tree 1 file changed +10
-5
lines changed
Expand file tree Collapse file tree 1 file changed +10
-5
lines changed Original file line number Diff line number Diff line change @@ -60,10 +60,10 @@ def load_world_file(self, path):
6060 words .add (word [0 ])
6161 return words
6262
63- def compute (self , message ): # TODO LIMIT TO DARKWEB ???
63+ def compute (self , message ):
6464 to_tag = False
6565 content = self .obj .get_content ().lower ()
66- # print(content )
66+ domain_id = self . get_message ( )
6767
6868 is_csam = False
6969 is_child_word = False
@@ -91,10 +91,15 @@ def compute(self, message): # TODO LIMIT TO DARKWEB ???
9191 print (f'CSAM DETECTED { content } ' )
9292 # print()
9393 self .add_message_to_queue (message = self .ce_tag , queue = 'Tags' )
94- # Domains
95- for dom in self . obj . get_correlation ( 'domain' ). get ( 'domain' , []) :
96- domain = Domain (dom [ 1 :] )
94+ # Domain
95+ if domain_id :
96+ domain = Domain (domain_id )
9797 self .add_message_to_queue (obj = domain , message = self .ce_tag , queue = 'Tags' )
98+ # Domains
99+ else :
100+ for dom in self .obj .get_correlation ('domain' ).get ('domain' , []):
101+ domain = Domain (dom [1 :])
102+ self .add_message_to_queue (obj = domain , message = self .ce_tag , queue = 'Tags' )
98103
99104 return to_tag
100105
You can’t perform that action at this time.
0 commit comments