54 lines
1.9 KiB
Python
54 lines
1.9 KiB
Python
#!/usr/bin/env python3
|
|
# -*- coding: utf-8 -*-
|
|
|
|
from typing import Any, Self, Optional, Sequence
|
|
from abc import ABC, abstractmethod
|
|
from Interfaces.Application.AnPInterface import AnPInterface
|
|
from Application.Event import Event
|
|
from Utils.Common import Common
|
|
|
|
class WebSocketServersAbstract(ABC):
|
|
|
|
def __init__(self:Self, anp:AnPInterface, key:str, inputs:Optional[dict[str, Any|None]|Sequence[Any|None]] = None) -> None:
|
|
self.anp:AnPInterface = anp
|
|
self.key:str = key
|
|
self.host:str = self.anp.settings.get(("web_socket_server_host", "host"), inputs, "")
|
|
self.port:int = self.anp.settings.get(("web_socket_server_port", "port"), inputs, 18765)
|
|
self.__sessions:dict[str, str] = {}
|
|
self.on_new_client:Event = Event()
|
|
self.on_message:Event = Event()
|
|
self.on_close:Event = Event()
|
|
self.on_error:Event = Event()
|
|
|
|
@abstractmethod
|
|
def start(self:Self) -> None:pass
|
|
|
|
@abstractmethod
|
|
def close(self:Self) -> None:pass
|
|
|
|
@abstractmethod
|
|
def close_client(self:Self, id:int) -> None:pass
|
|
|
|
def format_data(self:Self, controller:str, action:str, data:Any|None, id:str, code:int = 200) -> str:
|
|
return Common.data_encode({
|
|
"ok" : code >= 200 and code < 300,
|
|
"code" : code,
|
|
"controller": controller,
|
|
"action": action,
|
|
"data": data,
|
|
"id": id
|
|
})
|
|
|
|
@abstractmethod
|
|
def send(self:Self, controller:str, action:str, data:Any|None, ids:Optional[str|Sequence[str]] = None, code:int = 200) -> None:pass
|
|
|
|
def set_session(self:Self, id:str, session:str) -> None:
|
|
if id not in self.__sessions:
|
|
self.__sessions[id] = session
|
|
|
|
def get_session(self:Self, id:str) -> str|None:
|
|
return self.__sessions.get(id)
|
|
|
|
def remove_session(self:Self, id:str) -> None:
|
|
if id in self.__sessions:
|
|
del self.__sessions[id] |