feat(client.py): add Client class
This commit is contained in:
67
src/judas_client/client.py
Normal file
67
src/judas_client/client.py
Normal file
@@ -0,0 +1,67 @@
|
||||
# -*- coding: utf-8 -*-
|
||||
"""Client module for handling messages."""
|
||||
|
||||
from __future__ import annotations
|
||||
|
||||
import logging as lg
|
||||
import uuid
|
||||
|
||||
from judas_protocol import Message
|
||||
|
||||
from judas_client import Connector
|
||||
|
||||
|
||||
class Client:
|
||||
"""Client class for handling messages."""
|
||||
|
||||
def __init__(self, server_host: str, server_port: int) -> None:
|
||||
"""Initialize the Client.
|
||||
|
||||
Args:
|
||||
server_host (str): The server host address.
|
||||
server_port (int): The server port number.
|
||||
"""
|
||||
self.logger: lg.Logger = lg.getLogger(
|
||||
f"{__name__}.{self.__class__.__name__}"
|
||||
)
|
||||
self.logger.debug("Initializing Client...")
|
||||
|
||||
self.server_host: str = server_host
|
||||
self.server_port: int = server_port
|
||||
|
||||
self.mac_address: str = self._get_mac_address()
|
||||
self.logger.debug(f"MAC address: {self.mac_address}")
|
||||
|
||||
self.connector: Connector = Connector(
|
||||
mac_address=self.mac_address,
|
||||
host=self.server_host,
|
||||
port=self.server_port,
|
||||
on_message=self.handle_message,
|
||||
)
|
||||
|
||||
def _get_mac_address(self) -> str:
|
||||
"""Get the MAC address of the client.
|
||||
|
||||
Returns:
|
||||
str: The MAC address in standard lowercase format (e.g., "0a:1b:2c:3d:4e:5f").
|
||||
"""
|
||||
mac_address: str = ":".join(
|
||||
[
|
||||
"{:02x}".format((uuid.getnode() >> ele) & 0xFF)
|
||||
for ele in range(0, 8 * 6, 8)
|
||||
][::-1]
|
||||
)
|
||||
return mac_address
|
||||
|
||||
def handle_message(self, message: Message) -> None:
|
||||
"""Handle incoming messages.
|
||||
|
||||
Args:
|
||||
message (Message): The incoming message to handle.
|
||||
"""
|
||||
self.logger.info(f"[v] Received message: {message}")
|
||||
|
||||
def run(self) -> None:
|
||||
"""Run the client."""
|
||||
self.logger.info(f"Starting Client ({self.mac_address})...")
|
||||
self.connector.run()
|
||||
Reference in New Issue
Block a user