Make socket_type optional

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,7 +22,8 @@ class BaseClient:
         self.timeout = timeout
         self.passwd = passwd
 
-    def __init_subclass__(cls, *, socket_type: SocketKind = SOCK_STREAM):
+    def __init_subclass__(cls, *, socket_type: SocketKind | None = None):
+        if socket_type is not None:
             cls._socket_type = socket_type
 
     def __enter__(self):

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: