From 593803edd2efe4b45f201b8d423b5f4af7d4d6d7 Mon Sep 17 00:00:00 2001 From: Ruben Pollan Date: Thu, 30 Jun 2016 19:05:12 +0200 Subject: [feat] add support for private keys into bitmask cli --- src/leap/bitmask/core/dispatcher.py | 22 +++++++++++++++++----- 1 file changed, 17 insertions(+), 5 deletions(-) (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 785c60dc..a50a6b62 100644 --- a/src/leap/bitmask/core/dispatcher.py +++ b/src/leap/bitmask/core/dispatcher.py @@ -153,22 +153,30 @@ class KeysCmd(SubCommand): label = 'keys' - @register_method("[[str, str]]") + @register_method("[dict]") def do_LIST(self, service, *parts, **kw): + private = False + if parts[-1] == 'private': + private = True + bonafide = kw['bonafide'] d = bonafide.do_get_active_user() - d.addCallback(service.do_list_keys) + d.addCallback(service.do_list_keys, private) return d - @register_method('str') + @register_method('dict') def do_EXPORT(self, service, *parts, **kw): if len(parts) < 3: return defer.fail("An email address is needed") address = parts[2] + private = False + if parts[-1] == 'private': + private = True + bonafide = kw['bonafide'] d = bonafide.do_get_active_user() - d.addCallback(service.do_export, address) + d.addCallback(service.do_export, address, private) return d @register_method('str') @@ -177,9 +185,13 @@ class KeysCmd(SubCommand): return defer.fail("An email address is needed") address = parts[2] + private = False + if parts[-1] == 'private': + private = True + bonafide = kw['bonafide'] d = bonafide.do_get_active_user() - d.addCallback(service.do_delete, address) + d.addCallback(service.do_delete, address, private) return d -- cgit v1.2.3