Python Server: Add a category string for servers

1 files changed, 12 insertions, 0 deletions

diff --git a/wrappers/python/eduvpn_common/server.py b/wrappers/python/eduvpn_common/server.py
index 36c3643..bae9ba4 100644
--- a/wrappers/python/eduvpn_common/server.py
+++ b/wrappers/python/eduvpn_common/server.py
@@ -36,12 +36,20 @@ class Server:
     def __str__(self):
         return self.display_name
 
+    @property
+    def category(self):
+        return "Custom Server"
+
 
 class InstituteServer(Server):
     def __init__(self, url, display_name, support_contact, profiles, expire_time):
         super().__init__(url, display_name, profiles, expire_time)
         self.support_contact = support_contact
 
+    @property
+    def category(self):
+        return "Institute Access Server"
+
 
 class SecureInternetServer(Server):
     def __init__(
@@ -58,6 +66,10 @@ class SecureInternetServer(Server):
         self.support_contact = support_contact
         self.country_code = country_code
 
+    @property
+    def category(self):
+        return "Secure Internet Server"
+
 
 def get_type_for_str(type_str: str):
     if type_str == "secure_internet":