From 808f9d3542e21c819beb8fe72224f000ae2e019c Mon Sep 17 00:00:00 2001
From: Ruben Pollan <meskio@sindominio.net>
Date: Thu, 31 Aug 2017 10:58:59 +0200
Subject: [feat] expose an API to set/get/list gateway preferences

- Related: #9010
---
 src/leap/bitmask/core/dispatcher.py | 14 ++++++++++++++
 1 file changed, 14 insertions(+)

(limited to 'src/leap/bitmask/core/dispatcher.py')

diff --git a/src/leap/bitmask/core/dispatcher.py b/src/leap/bitmask/core/dispatcher.py
index 508a925d..363594ca 100644
--- a/src/leap/bitmask/core/dispatcher.py
+++ b/src/leap/bitmask/core/dispatcher.py
@@ -237,6 +237,20 @@ class VPNCmd(SubCommand):
         d = vpn.do_list()
         return d
 
+    @register_method('list')
+    def do_LOCATIONS(self, vpn, *parts):
+        if len(parts) > 2:
+            return vpn.do_set_locations(parts[2:])
+
+        return vpn.do_get_locations()
+
+    @register_method('list')
+    def do_COUNTRIES(self, vpn, *parts):
+        if len(parts) > 2:
+            return vpn.do_set_countries(parts[2:])
+
+        return vpn.do_get_countries()
+
 
 class MailCmd(SubCommand):
 
-- 
cgit v1.2.3