Change async_rcon to class

By changing this to a class we clean up code duplication. Now we just declare the host, port, and password then call the method with just a command and arguments. tested to confirm working

README.md

@@ -35,13 +35,19 @@ If you prefer to use Source RCON in an asynchronous environment, you can use
 `rcon()`.
 
 ```python
-from rcon.source import rcon
+from rcon.source import Rcon
 
-response = await rcon(
+response = await Rcon.rcon(
     'some_command', 'with', 'some', 'arguments',
     host='127.0.0.1', port=5000, passwd='mysecretpassword'
 )
 print(response)
+
+or
+
+connect = Rcon('127.0.0.1', 12715, 'mysecretpassword')
+response = await connect.rcon('some_command','with', 'some', 'arguments')
+print(response)
 ```
 
 ### BattlEye RCon

rcon/__init__.py

@@ -4,7 +4,7 @@ from typing import Any, Coroutine
 from warnings import warn
 
 from rcon.exceptions import EmptyResponse, SessionTimeout, WrongPassword
-from rcon.source import rcon as _rcon
+from rcon.source import Rcon as _rcon
 from rcon.source import Client as _Client
 
 

rcon/source/__init__.py

@@ -1,7 +1,7 @@
 """Source RCON implementation."""
 
-from rcon.source.async_rcon import rcon
+from rcon.source.async_rcon import Rcon
 from rcon.source.client import Client
 
 
-__all__ = ["Client", "rcon"]
+__all__ = ["Client", "Rcon"]

rcon/source/async_rcon.py

@@ -6,17 +6,31 @@ from rcon.exceptions import SessionTimeout, WrongPassword
 from rcon.source.proto import Packet, Type
 
 
-__all__ = ["rcon"]
+__all__ = ["Rcon"]
 
 
-async def close(writer: StreamWriter) -> None:
+class Rcon:
+    """
+    Set's the variables needed for RCON connection
+    """
+    def __init__(self,
+                 host: str,
+                 port: int = 27015,
+                 password: str = ''
+                 ):
+
+        self.host = host
+        self.port = port
+        self.password = password
+
+    async def close(self, writer: StreamWriter) -> None:
         """Close socket asynchronously."""
 
         writer.close()
         await writer.wait_closed()
 
-
     async def communicate(
+        self,
         reader: StreamReader,
         writer: StreamWriter,
         packet: Packet,
@@ -41,21 +55,27 @@ async def communicate(
 
         return response
 
-
     async def rcon(
+        self,
         command: str,
         *arguments: str,
-    host: str,
+        host: str = None,
-    port: int,
+        port: int = None,
-    passwd: str,
+        passwd: str = None,
         encoding: str = "utf-8",
         frag_threshold: int = 4096,
         frag_detect_cmd: str = "",
     ) -> str:
         """Run a command asynchronously."""
+        try:
+            host = host or self.host
+            port = port or self.port
+            passwd = passwd or self.password
+        except AttributeError:
+            return print("Make sure you declare the host, port, or password.")
 
         reader, writer = await open_connection(host, port)
-    response = await communicate(
+        response = await self.communicate(
             reader,
             writer,
             Packet.make_login(passwd, encoding=encoding),
@@ -69,12 +89,12 @@ async def rcon(
             response = await Packet.aread(reader)
 
         if response.id == -1:
-        await close(writer)
+            await self.close(writer)
             raise WrongPassword()
 
         request = Packet.make_command(command, *arguments, encoding=encoding)
-    response = await communicate(reader, writer, request)
+        response = await self.communicate(reader, writer, request)
-    await close(writer)
+        await self.close(writer)
 
         if response.id != request.id:
             raise SessionTimeout()