pbesa 2.1__py3-none-any.whl → 4.0.0__py3-none-any.whl

pbesa/{middleware/remote/AdmListenerHandler.py → remote/adm_listener_handler.py}

@@ -1,57 +1,74 @@
-import os
-import sys
-import json
-import inspect
-import socketserver
-from time import sleep
-from importlib.machinery import SourceFileLoader
-from ...kernel.system.Directory import Directory
-
-import types
-
-class AdmListenerHandler(socketserver.StreamRequestHandler):
-
-    def handle(self):
-        self.data = self.rfile.readline().strip()
-        print("{} wrote:".format(self.client_address[0]))
-        print(self.data)
-        msg = str(self.data, "utf-8")
-        info = json.loads(msg)
-        if info['command'] == 'UPDATE':        
-            agents = info['agents']
-            directory = Directory()
-            for agent in agents:
-                directory.resetAgentList()
-                directory.addAgent(agent)
-        if info['command'] == 'MOVE':   
-            appPath = os.path.abspath(os.path.dirname(sys.argv[0]))
-            appPathSplit = appPath.split(os.sep)
-
-            classPath = info['path']
-            classPathSplit = classPath.split('.')
-
-            path = appPathSplit[0]
-            findFlag = False
-
-            for x in range(1, len(appPathSplit) ):
-                if appPathSplit[x] == classPathSplit[0]:
-                    path = path + os.sep + appPathSplit[x]
-                    for y in range(1, len(classPathSplit) ):
-                        path = path + os.sep + classPathSplit[y]
-                    break
-                else:
-                    path = path + os.sep + appPathSplit[x]            
-            path = path + '.py'
-
-            module = SourceFileLoader(info['class'], path).load_module()
-            agType = getattr(module, info['class'])
-            ag = agType(info['id'])
-            ag.state = info['state']
-            ag.start()
-            
-        
-        if info['command'] == 'SENDEVENT':
-            from ...KernelTK.SystemTK.Adm import Adm
-            Adm().sendEvent(info['alias'], info['event'], info['data'])
-            rsp = 'ACK'
-            self.wfile.write(bytes(rsp + "\n", "utf-8"))
+# -*- coding: utf-8 -*-
+"""
+----------------------------------------------------------
+-------------------------- PBESA -------------------------
+----------------------------------------------------------
+
+@autor AKEN
+@version 4.0.0
+@date 08/08/24
+"""
+
+# --------------------------------------------------------
+# Define resources
+# --------------------------------------------------------
+
+import os
+import sys
+import json
+import base64
+import socketserver
+from importlib.machinery import SourceFileLoader
+
+# --------------------------------------------------------
+# Define component
+# --------------------------------------------------------
+
+class AdmListenerHandler(socketserver.StreamRequestHandler):
+    """ ADM Listener Handler class """
+
+    def handle(self):
+        """ Handle """
+        from ..mas import Directory
+        self.data = self.rfile.readline().strip()
+        print("{} wrote:".format(self.client_address[0]))
+        print(self.data)
+        msg = str(self.data, "utf-8")
+        info = json.loads(msg)    
+        if info['command'] == 'UPDATE':        
+            agents = info['agents']
+            directory = Directory()
+            directory.set_agent_list(agents)
+        if info['command'] == 'MOVE':   
+            appPath = os.path.abspath(os.path.dirname(sys.argv[0]))
+            appPathSplit = appPath.split(os.sep)
+            classPath = info['path']
+            classPathSplit = classPath.split('.')
+            path = appPathSplit[0]
+            findFlag = False
+            for x in range(1, len(appPathSplit) ):
+                if appPathSplit[x] == classPathSplit[0]:
+                    path = path + os.sep + appPathSplit[x]
+                    for y in range(1, len(classPathSplit) ):
+                        path = path + os.sep + classPathSplit[y]
+                    break
+                else:
+                    path = path + os.sep + appPathSplit[x]            
+            path = path + '.py'
+            module = SourceFileLoader(info['class'], path).load_module()
+            agType = getattr(module, info['class'])
+            ag = agType(info['id'])
+            ag.state = info['state']
+            ag.start()
+        if info['command'] == 'SENDEVENT':
+            from ..mas import Adm
+            data = info['data']
+            aux = None
+            if data and not data == 'None':
+                data = data.encode('utf-8')
+                data = base64.b64decode(data)
+                data = data.decode('utf-8')
+                aux = json.loads(data) 
+            Adm().send_event(info['id'], info['event'], aux)
+            rsp = 'ACK'
+            self.wfile.write(bytes(rsp + "\n", "utf-8"))

pbesa/remote/exceptions.py

@@ -0,0 +1,18 @@
+# -*- coding: utf-8 -*-
+"""
+------------------------------------------------------------
+-------------------------- PBESA --------------------------- 
+------------------------------------------------------------
+
+@autor AKEN & SIDRE
+@version 3.0.1
+@date 27/07/20
+"""
+
+# ----------------------------------------------------------
+# Defines system component exceptions
+# ----------------------------------------------------------
+
+class RemoteException(Exception):
+    """ Base class for exceptions of system """
+    pass

pbesa/remote/remote_adm.py

@@ -0,0 +1,42 @@
+# -*- coding: utf-8 -*-
+"""
+----------------------------------------------------------
+-------------------------- PBESA -------------------------
+----------------------------------------------------------
+
+@autor AKEN
+@version 4.0.0
+@date 09/08/24
+"""
+
+import traceback
+import socketserver
+from threading import Thread
+from .exceptions import RemoteException
+from .remote_adm_handler import RemoteAdmHandler
+
+class RemoteAdm(Thread):
+    """ RemoteAdm class """
+
+    # IP
+    __IP = None
+    # PORT
+    __PORT = None
+
+    def __init__(self, ip:str, port:str) -> None:
+        """ Constructor
+        @param ip: IP
+        @param port: PORT
+        """
+        self.__IP = ip
+        self.__PORT = port
+        super().__init__()
+
+    def run(self) -> None:
+        """ Run method """
+        try:        
+            server = socketserver.TCPServer((self.__IP, self.__PORT), RemoteAdmHandler)
+            server.serve_forever()
+        except:
+            traceback.print_exc()
+            raise RemoteException("Could not initialize master container on %s host and %d port" % (self.__IP, self.__PORT))

pbesa/{middleware/remote/RemoteAdmHandler.py → remote/remote_adm_handler.py}

@@ -1,83 +1,109 @@
-import os
-import json
-import socket
-import socketserver
-from time import sleep
-from importlib.machinery import SourceFileLoader
-from ...kernel.system.Directory import Directory
-        
-class RemoteAdmHandler(socketserver.StreamRequestHandler):
-
-    def handle(self):
-        self.data = self.rfile.readline().strip()
-        print("{} wrote:".format(self.client_address[0]))
-        print(self.data)
-        msg = str(self.data, "utf-8")
-        info = json.loads(msg)
-        if info['command'] == 'REGISTER':
-            directory = Directory()
-            directory.addContainer({'alias': info['alias'], 'ip': info['ip'], 'port': info['port']})
-            data = 'ACK'
-            self.wfile.write(bytes(data + "\n", "utf-8"))
-            sleep(2) # 2 seconds.
-            containers = directory.getContainers()
-            agents = directory.getAgents()
-            dto = '{"command":"UPDATE", "agents" : ' + json.dumps(agents, ensure_ascii=False) + '}'
-            for ctn in containers:
-                sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
-                try:                            
-                    sock.connect( (ctn['ip'], int(ctn['port'])) )                            
-                    sock.sendall(bytes(dto + "\n", "utf-8"))
-                except:
-                    pass
-                    
-                finally:
-                    sock.close()
-
-        if info['command'] == 'MOVE':
-            appPath = os.path.abspath(os.path.dirname(sys.argv[0]))
-            appPathSplit = appPath.split(os.sep)
-
-            classPath = info['path']
-            classPathSplit = classPath.split('.')
-
-            path = appPathSplit[0]
-            findFlag = False
-
-            for x in range(1, len(appPathSplit) ):
-                if appPathSplit[x] == classPathSplit[0]:
-                    path = path + os.sep + appPathSplit[x]
-                    for y in range(1, len(classPathSplit) ):
-                        path = path + os.sep + classPathSplit[y]
-                    break
-                else:
-                    path = path + os.sep + appPathSplit[x]            
-            path = path + '.py'
-
-            module = SourceFileLoader(info['class'], path).load_module()
-            agType = getattr(module, info['class'])
-            ag = agType(info['id'])
-            ag.state = info['state']
-            ag.start()
-            
-            
-        if info['command'] == 'UPDATE':        
-            agents = info['agents']
-            directory = Directory()
-            for agent in agents:
-                directory.resetAgentList()
-                directory.addAgent(agent)
-            
-            containers = directory.getContainers()
-            agents = directory.getAgents()
-            dto = '{"command":"UPDATE", "agents" : ' + json.dumps(agents, ensure_ascii=False) + '}'
-            for ctn in containers:
-                sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
-                try:                            
-                    sock.connect( (ctn['ip'], int(ctn['port'])) )                            
-                    sock.sendall(bytes(dto + "\n", "utf-8"))
-                except:
-                    pass
-                    
-                finally:
-                    sock.close()
+# -*- coding: utf-8 -*-
+"""
+----------------------------------------------------------
+-------------------------- PBESA -------------------------
+----------------------------------------------------------
+
+@autor AKEN
+@version 4.0.0
+@date 08/08/24
+"""
+
+# --------------------------------------------------------
+# Define resources
+# --------------------------------------------------------
+
+import os
+import json
+import base64
+import socket
+import socketserver
+from time import sleep
+from importlib.machinery import SourceFileLoader
+
+# --------------------------------------------------------
+# Define component
+# --------------------------------------------------------
+
+class RemoteAdmHandler(socketserver.StreamRequestHandler):
+    """ Remote ADM Handler class """
+
+    def handle(self):
+        """ Handle """
+        from ..mas import Directory
+        self.data = self.rfile.readline().strip()
+        print("{} wrote:".format(self.client_address[0]))
+        print(self.data)
+        msg = str(self.data, "utf-8")
+        info = json.loads(msg)
+        if info['command'] == 'REGISTER':
+            directory = Directory()
+            directory.add_container({'name': info['name'], 'host': info['host'], 'port': info['port']})
+            data = 'ACK'
+            self.wfile.write(bytes(data + "\n", "utf-8"))
+            sleep(2) # 2 seconds.
+            containers = directory.get_containers()
+            agents = directory.get_agents()
+            dto = '{"command":"UPDATE", "agents" : ' + json.dumps(agents, ensure_ascii=False) + '}'
+            for ctn in containers:
+                sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
+                try:                            
+                    sock.connect( (ctn['host'], int(ctn['port'])) )                            
+                    sock.sendall(bytes(dto + "\n", "utf-8"))
+                except:
+                    pass
+                    
+                finally:
+                    sock.close()
+            directory.check_full(info['name'])
+        if info['command'] == 'MOVE':
+            appPath = os.path.abspath(os.path.dirname(sys.argv[0]))
+            appPathSplit = appPath.split(os.sep)
+            classPath = info['path']
+            classPathSplit = classPath.split('.')
+            path = appPathSplit[0]
+            findFlag = False
+            for x in range(1, len(appPathSplit) ):
+                if appPathSplit[x] == classPathSplit[0]:
+                    path = path + os.sep + appPathSplit[x]
+                    for y in range(1, len(classPathSplit) ):
+                        path = path + os.sep + classPathSplit[y]
+                    break
+                else:
+                    path = path + os.sep + appPathSplit[x]            
+            path = path + '.py'
+            module = SourceFileLoader(info['class'], path).load_module()
+            agType = getattr(module, info['class'])
+            ag = agType(info['id'])
+            ag.state = info['state']
+            ag.start()
+        if info['command'] == 'ADD':
+            agent = info['agent']
+            host = agent['host']
+            directory = Directory()
+            directory.add_agent(agent)    
+            containers = directory.get_containers()
+            agents = directory.get_agents()
+            dto = '{"command":"UPDATE", "agents" : ' + json.dumps(agents, ensure_ascii=False) + '}'
+            for ctn in containers:
+                if not ctn['host'] == host:
+                    sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
+                    try:                            
+                        sock.connect( (ctn['host'], int(ctn['port'])) )                            
+                        sock.sendall(bytes(dto + "\n", "utf-8"))
+                    except:
+                        pass
+                    finally:
+                        sock.close()        
+        if info['command'] == 'SENDEVENT':
+            from ..mas import Adm
+            data = info['data']
+            aux = None
+            if data and not data == 'None':
+                data = data.encode('utf-8')
+                data = base64.b64decode(data)
+                data = data.decode('utf-8')
+                aux = json.loads(data)
+            Adm().send_event(info['id'], info['event'], data)
+            rsp = 'ACK'
+            self.wfile.write(bytes(rsp + "\n", "utf-8"))

pbesa/social/__init__.py

@@ -0,0 +1,4 @@
+from .worker import *
+from .delegator import *
+from .dispatcher_team import *
+from .collaborative_team import *