From 2931d6d784c2aa64ad3a951390fb121eaa39d2f3 Mon Sep 17 00:00:00 2001
From: Richard Neumann <mail@richard-neumann.de>
Date: Sun, 13 Feb 2022 18:15:09 +0100
Subject: [PATCH] Fix socket type initialization

---
 rcon/battleye/client.py | 3 ++-
 rcon/client.py          | 6 ++++--
 2 files changed, 6 insertions(+), 3 deletions(-)

diff --git a/rcon/battleye/client.py b/rcon/battleye/client.py
index 76b2e05..6de9f2e 100644
--- a/rcon/battleye/client.py
+++ b/rcon/battleye/client.py
@@ -1,6 +1,7 @@
 """BattlEye RCon client."""
 
 from ipaddress import IPv4Address
+from socket import SOCK_DGRAM
 from typing import Union
 
 from rcon.battleye.proto import Command, LoginRequest
@@ -13,7 +14,7 @@ __all__ = ['Client']
 Host = Union[str, IPv4Address]
 
 
-class Client(BaseClient):
+class Client(BaseClient, socket_type=SOCK_DGRAM):
     """BattlEye RCon client."""
 
     def communicate(self, data: bytes, *, recv: int = 4096) -> bytes:
diff --git a/rcon/client.py b/rcon/client.py
index 1ab14db..7b9fc07 100644
--- a/rcon/client.py
+++ b/rcon/client.py
@@ -14,17 +14,19 @@ class BaseClient:
 
     def __init__(
             self, host: str, port: int, *,
-            type: SocketKind = SOCK_STREAM,
             timeout: Optional[float] = None,
             passwd: Optional[str] = None
     ):
         """Initializes the base client with the SOCK_STREAM socket type."""
-        self._socket = socket(type=type)
+        self._socket = socket(type=self._socket_type)
         self.host = host
         self.port = port
         self.timeout = timeout
         self.passwd = passwd
 
+    def __init_subclass__(cls, *, socket_type: SocketKind = SOCK_STREAM):
+        cls._socket_type = socket_type
+
     def __enter__(self):
         """Attempts an auto-login if a password is set."""
         self._socket.__enter__()