From 4715b9c54fb6411019595175ef6326703e2012bf Mon Sep 17 00:00:00 2001
From: jwijenbergh <jeroenwijenbergh@protonmail.com>
Date: Tue, 27 Sep 2022 16:38:25 +0200
Subject: Python Server: Add a category string for servers

---
 wrappers/python/eduvpn_common/server.py | 12 ++++++++++++
 1 file changed, 12 insertions(+)

(limited to 'wrappers/python/eduvpn_common')

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":
-- 
cgit v1.2.3