Initial Commit

iris_scoring_module.py

@@ -0,0 +1,36 @@
+#!/usr/bin/env python3
+
+from iris_interface.IrisModuleInterface import IrisModuleInterface
+
+class IrisScoringModule(IrisModuleInterface):
+    _module_name = interface_conf.module_name
+    _module_description = interface_conf.module_description
+    _interface_version = interface_conf.interface_version
+    _module_version = interface_conf.module_version
+    _pipeline_support = interface_conf.pipeline_support
+    _pipeline_info = interface_conf.pipeline_info
+    _module_configuration = interface_conf.module_configuration
+    _module_type = interface_conf.module_type
+    
+    def register_hooks(self, module_id: int):
+        """
+        Called by IRIS indicating it's time to register hooks.  
+        :param module_id: Module ID provided by IRIS.
+        """        
+        status = self.register_to_hook(module_id, iris_hook_name='on_postload_ioc_create')
+        if status.is_failure():
+            self.log.error(status.get_message())
+        else:
+            self.log.info(f"Successfully subscribed to on_postload_ioc_create hook")
+
+
+    def hooks_handler(self, hook_name: str, data):
+        """
+        Called by IRIS each time one of our hook is triggered. 
+        """
+    
+        if self._dict_conf.get('log_received_hook') is True:
+            self.log.info(f'Received {hook_name}')
+            self.log.info(f'Received data of type {type(data)}')
+    
+        return InterfaceStatus.I2Success(data=data, logs=list(self.message_queue))