From 9eea3c14e9956d8e32f43b106f2bf26dbf517a6e Mon Sep 17 00:00:00 2001
From: Richard Neumann <mail@richard-neumann.de>
Date: Sat, 26 Dec 2020 01:03:23 +0100
Subject: [PATCH] Added prompt formatting with host and port.

---
 rcon/console.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/rcon/console.py b/rcon/console.py
index 2e3b849..77080b9 100644
--- a/rcon/console.py
+++ b/rcon/console.py
@@ -14,7 +14,7 @@ EXIT_COMMANDS = {'exit', 'quit'}
 MSG_LOGIN_ABORTED = '\nLogin aborted. Bye.'
 MSG_EXIT = '\nBye.'
 MSG_SESSION_TIMEOUT = 'Session timed out. Please login again.'
-PROMPT = 'RCON> '
+PROMPT = 'RCON {host}:{port}> '
 
 
 def read_host() -> str:
@@ -135,6 +135,8 @@ def rconcmd(host: str, port: int, passwd: str, *, prompt: str = PROMPT):
         print(MSG_EXIT)
         return
 
+    prompt = prompt.format(host=host, port=port)
+
     with Client(host, port) as client:
         try:
             passwd = login(client, passwd)