From e0570df5538648248356067bdd53d92c4edf8eb3 Mon Sep 17 00:00:00 2001 From: Alex <61382599+NeffIsBack@users.noreply.github.com> Date: Wed, 16 Aug 2023 11:14:09 +0200 Subject: [PATCH] Implement color coding from config for nla --- cme/protocols/rdp.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/cme/protocols/rdp.py b/cme/protocols/rdp.py index 6882b25f..8f7ed278 100644 --- a/cme/protocols/rdp.py +++ b/cme/protocols/rdp.py @@ -12,6 +12,7 @@ from impacket.krb5.ccache import CCache from cme.connection import * from cme.helpers.bloodhound import add_user_bh from cme.logger import CMEAdapter +from cme.config import host_info_colors from aardwolf.connection import RDPConnection from aardwolf.commons.queuedata.constants import VIDEO_FORMAT @@ -103,7 +104,7 @@ class rdp(connection): ) def print_host_info(self): - nla = colored(f"nla:{self.nla}", 'blue', attrs=['bold']) if self.nla else colored(f"nla:{self.nla}", 'cyan', attrs=['bold']) + nla = colored(f"nla:{self.nla}", host_info_colors[3], attrs=['bold']) if self.nla else colored(f"nla:{self.nla}", host_info_colors[2], attrs=['bold']) if self.domain is None: self.logger.display("Probably old, doesn't not support HYBRID or HYBRID_EX" f" ({nla})") else: