event_parser.py

import re
# Typical string (ip_address, state)
states = ["TRAINING", "INFERENCE", "NOT_READY"]


class EventParser():
    def __init__(self, filtering):
        # The message received is typically encoded in binary
        # So, must be transformed in string
        self._message = ""
        self.filtering = filtering
    
    def set_message(self, message):
        self._message = message.decode('utf-8').replace(" ","")
    

    def ip_address(self):
        # Now the string must be parsed, in order to obtain the ip address of the device
        # 1) remove the brackets
        to_parse = re.sub(r'[\(\)]', '', self._message)

        # 2) obtain the ip address
        ip_address = re.split(r',', to_parse)[0]
        
        # 3) verify ip address
        pattern = r'\d{1,3}\.\d{1,3}\.\d{1,3}\.\d{1,3}'
        if(re.match(pattern, ip_address)):
            if not self.filtering:
                return ip_address
            else:
                if valid_iot_ip_address(ip_address):
                    return ip_address
                else:
                    return -1
        else:
            return -1
    
    def port(self, local=False):
        if local:
            return -1
        
        # 1) remove the brackets
        to_parse = re.sub(r'[\(\)]', '', self._message)

        # 2) obtain the port
        port = int(re.split(r',', to_parse)[1])

        # 3) Verify the port
        if port in range(65535):
            return port
        else:
            return -1

    def state(self, local=False):
        # Now the string must be parsed, in order to obtain the event sent by the device (which represent its state)
        # 1) remove the brackets
        to_parse = re.sub(r'[\(\)]', '', self._message)

        # 2) obtain the state
        if local:
            state = re.split(r',', to_parse)[1]
        else:
            state = re.split(r',', to_parse)[2]
        
        if state in states:
            return state
        else:
            return None
    
    def training(self):
        return states[0]
    
    def inference(self):
        return states[1]

def valid_iot_ip_address(ip_address):
    """
        This method verify that the ip address is an IoT device.
        To do that, it uses the file "filtering_file.txt" in the device_filtering folder,
        which is the file updated each 40 seconds (this is the default value, of course can change) 
        by the router.
    """
    
    with open("./device_filtering/filtering_file.txt", "r") as f:
        lines = f.readlines()
        for line in lines:
            if line.rstrip() == ip_address:
                return True
        
    
    return False