ail-framework/bin/modules/abstract_module.py

179 lines
5.7 KiB
Python
Raw Normal View History

2021-04-02 07:52:05 +00:00
# coding: utf-8
"""
Base Class for AIL Modules
"""
##################################
# Import External packages
##################################
from abc import ABC, abstractmethod
import os
2023-05-12 13:29:53 +00:00
import logging
import logging.config
import sys
2021-04-02 07:52:05 +00:00
import time
import traceback
2021-04-02 07:52:05 +00:00
sys.path.append(os.environ['AIL_BIN'])
2021-04-02 07:52:05 +00:00
##################################
# Import Project packages
##################################
from pubsublogger import publisher
2023-05-12 13:29:53 +00:00
from lib import ail_logger
from lib.ail_queues import AILQueue
from lib import regex_helper
from lib.exceptions import ModuleQueueError
2021-04-02 07:52:05 +00:00
2023-05-12 13:29:53 +00:00
logging.config.dictConfig(ail_logger.get_config(name='modules'))
2021-04-02 07:52:05 +00:00
class AbstractModule(ABC):
"""
Abstract Module class
"""
2023-05-12 13:29:53 +00:00
def __init__(self, module_name=None, queue=True):
2021-04-02 07:52:05 +00:00
"""
AIL Module,
2021-04-02 07:52:05 +00:00
module_name: str; set the module name if different from the instance ClassName
:param queue: Allow to push messages to other modules
2021-04-02 07:52:05 +00:00
"""
2023-05-12 13:29:53 +00:00
self.logger = logging.getLogger(f'{self.__class__.__name__}')
2021-04-02 07:52:05 +00:00
# Module name if provided else instance className
self.module_name = module_name if module_name else self._module_name()
self.pid = os.getpid()
# Setup the I/O queues
if queue:
self.queue = AILQueue(self.module_name, self.pid)
2021-04-02 07:52:05 +00:00
# Init Redis Logger
self.redis_logger = publisher
# Port of the redis instance used by pubsublogger
self.redis_logger.port = 6380
# Channel name to publish logs
# # TODO: refactor logging
2021-04-28 13:24:33 +00:00
# If provided could be a namespaced channel like script:<ModuleName>
2023-05-12 13:29:53 +00:00
self.redis_logger.channel = 'Script'
2021-05-14 12:50:42 +00:00
# Cache key
self.r_cache_key = regex_helper.generate_redis_cache_key(self.module_name)
2022-06-03 13:30:48 +00:00
self.max_execution_time = 30
2021-04-02 07:52:05 +00:00
# Run module endlessly
self.proceed = True
# Waiting time in seconds between two processed messages
2021-04-02 07:52:05 +00:00
self.pending_seconds = 10
# Debug Mode
self.debug = False
def get_message(self):
"""
Get message from the Redis Queue (QueueIn)
Input message can change between modules
ex: '<item id>'
"""
return self.queue.get_message()
def add_message_to_queue(self, message, queue_name=None):
"""
Add message to queue
:param message: message to send in queue
:param queue_name: queue or module name
ex: add_message_to_queue(item_id, 'Mail')
"""
self.queue.send_message(message, queue_name)
2021-10-29 16:48:12 +00:00
# add to new set_module
2021-04-02 07:52:05 +00:00
def get_available_queues(self):
return self.queue.get_out_queues()
def regex_search(self, regex, obj_id, content):
return regex_helper.regex_search(self.r_cache_key, regex, obj_id, content, max_time=self.max_execution_time)
def regex_finditer(self, regex, obj_id, content):
return regex_helper.regex_finditer(self.r_cache_key, regex, obj_id, content, max_time=self.max_execution_time)
def regex_findall(self, regex, obj_id, content, r_set=False):
"""
regex findall helper (force timeout)
:param regex: compiled regex
:param obj_id: object id
:param content: object content
:param r_set: return result as set
"""
return regex_helper.regex_findall(self.module_name, self.r_cache_key, regex, obj_id, content,
max_time=self.max_execution_time, r_set=r_set)
def regex_phone_iter(self, country_code, obj_id, content):
"""
regex findall helper (force timeout)
:param regex: compiled regex
:param obj_id: object id
:param content: object content
:param r_set: return result as set
"""
return regex_helper.regex_phone_iter(self.r_cache_key, country_code, obj_id, content,
max_time=self.max_execution_time)
2021-04-02 07:52:05 +00:00
def run(self):
"""
Run Module endless process
"""
2021-04-02 07:52:05 +00:00
# Endless loop processing messages from the input queue
while self.proceed:
# Get one message (ex:item id) from the Redis Queue (QueueIn)
message = self.get_message()
2021-04-28 13:24:33 +00:00
if message:
try:
# Module processing with the message from the queue
self.compute(message)
except Exception as err:
if self.debug:
self.queue.error()
raise err
# LOG ERROR
trace = traceback.format_tb(err.__traceback__)
trace = ''.join(trace)
2023-05-12 13:29:53 +00:00
self.logger.critical(f"Error in module {self.module_name}: {__name__} : {err}")
self.logger.critical(f"Module {self.module_name} input message: {message}")
self.logger.critical(trace)
if isinstance(err, ModuleQueueError):
self.queue.error()
raise err
2021-10-29 16:48:12 +00:00
# remove from set_module
## check if item process == completed
2021-04-28 13:24:33 +00:00
else:
2021-04-02 07:52:05 +00:00
self.computeNone()
# Wait before next process
2023-05-12 13:29:53 +00:00
self.logger.debug(f"{self.module_name}, waiting for new message, Idling {self.pending_seconds}s")
2021-04-02 07:52:05 +00:00
time.sleep(self.pending_seconds)
def _module_name(self):
"""
Returns the instance class name (ie the Module Name)
2021-04-02 07:52:05 +00:00
"""
return self.__class__.__name__
@abstractmethod
def compute(self, message):
"""
Main method of the Module to implement
"""
pass
def computeNone(self):
"""
Method of the Module when there is no message
"""
pass