Merge branch 'fix/missing-constants' into 'develop'

Add missing constants See merge request ProtonVPN/linux/linux-cli!8
iamzoltan · Apr 16, 2021 · edf7ba1 · edf7ba1
2 parents 212e6c2 + 1ea79e7
commit edf7ba1
Showing 1 changed file with 24 additions and 4 deletions.
diff --git a/protonvpn_cli/cli_dialog.py b/protonvpn_cli/cli_dialog.py
@@ -4,15 +4,35 @@
 from protonvpn_nm_lib import exceptions
 from protonvpn_nm_lib.core.subprocess_wrapper import subprocess
 from protonvpn_nm_lib.country_codes import country_codes
-from protonvpn_nm_lib.enums import ProtocolEnum, ServerTierEnum
+from protonvpn_nm_lib.enums import (FeatureEnum, KillswitchStatusEnum,
+                                    ProtocolEnum, ServerTierEnum)
+
 from .logger import logger
-from .constants import SUPPORTED_FEATURES, SERVER_TIERS
 
 
 class ProtonVPNDialog:
 
     def __init__(self, protonvpn):
         self.protonvpn = protonvpn
+        self.SUPPORTED_FEATURES = {
+            FeatureEnum.NORMAL: "",
+            FeatureEnum.SECURE_CORE: "Secure-Core",
+            FeatureEnum.TOR: "Tor",
+            FeatureEnum.P2P: "P2P",
+            FeatureEnum.STREAMING: "Streaming",
+            FeatureEnum.IPv6: "IPv6"
+        }
+        self.SERVER_TIERS = {
+            ServerTierEnum.FREE: "Free",
+            ServerTierEnum.BASIC: "Basic",
+            ServerTierEnum.PLUS_VISIONARY: "Plus/Visionary",
+            ServerTierEnum.PM: "PMTEAM"
+        }
+        self.KILLSWITCH_STATUS_TEXT = {
+            KillswitchStatusEnum.HARD: "Permanent",
+            KillswitchStatusEnum.SOFT: "On",
+            KillswitchStatusEnum.DISABLED: "Off",
+        }
 
     def start(self):
         """Connect to server with a dialog menu.
@@ -91,9 +111,9 @@ def display_servers(self, country, countries):
             )
             load = str(int(server.load)).rjust(3, " ")
             features = ", ".join(
-                [SUPPORTED_FEATURES[feature] for feature in server.features]
+                [self.SUPPORTED_FEATURES[feature] for feature in server.features]
             )
-            tier = SERVER_TIERS[ServerTierEnum(server.tier)]
+            tier = self.SERVER_TIERS[ServerTierEnum(server.tier)]
 
             choices.append(
                 (