From 26981173c0458025efd90dfe81b67830bbbbced1 Mon Sep 17 00:00:00 2001 From: Richard Neumann Date: Sat, 19 Feb 2022 18:02:43 +0100 Subject: [PATCH] Make socket_type optional --- rcon/client.py | 7 ++++--- rcon/source/client.py | 4 +++- 2 files changed, 7 insertions(+), 4 deletions(-) diff --git a/rcon/client.py b/rcon/client.py index 0fd22f6..6093e50 100644 --- a/rcon/client.py +++ b/rcon/client.py @@ -1,6 +1,6 @@ """Common base client.""" -from socket import SOCK_STREAM, SocketKind, socket +from socket import SocketKind, socket from typing import Optional @@ -22,8 +22,9 @@ class BaseClient: self.timeout = timeout self.passwd = passwd - def __init_subclass__(cls, *, socket_type: SocketKind = SOCK_STREAM): - cls._socket_type = socket_type + def __init_subclass__(cls, *, socket_type: SocketKind | None = None): + if socket_type is not None: + cls._socket_type = socket_type def __enter__(self): """Attempts an auto-login if a password is set.""" diff --git a/rcon/source/client.py b/rcon/source/client.py index 82ff2f0..b50c082 100644 --- a/rcon/source/client.py +++ b/rcon/source/client.py @@ -1,5 +1,7 @@ """Synchronous client.""" +from socket import SOCK_STREAM + from rcon.client import BaseClient from rcon.exceptions import SessionTimeout, WrongPassword from rcon.source.proto import Packet, Type @@ -8,7 +10,7 @@ from rcon.source.proto import Packet, Type __all__ = ['Client'] -class Client(BaseClient): +class Client(BaseClient, socket_type=SOCK_STREAM): """An RCON client.""" def communicate(self, packet: Packet) -> Packet: