f11bb256c49a92caa3b38d324624d8cb3a3d8c93
[gpgme.git] / lang / python / examples / inter-edit.py
1 #!/usr/bin/env python
2 #
3 # Copyright (C) 2016 g10 Code GmbH
4 # Copyright (C) 2005 Igor Belyi <belyi@users.sourceforge.net>
5 #
6 # This program is free software; you can redistribute it and/or modify
7 # it under the terms of the GNU General Public License as published by
8 # the Free Software Foundation; either version 2 of the License, or
9 # (at your option) any later version.
10 #
11 # This program is distributed in the hope that it will be useful, but
12 # WITHOUT ANY WARRANTY; without even the implied warranty of
13 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
14 # General Public License for more details.
15 #
16 # You should have received a copy of the GNU General Public License
17 # along with this program; if not, see <https://www.gnu.org/licenses/>.
18 """Simple interactive editor to test editor scripts"""
19
20 from __future__ import absolute_import, print_function, unicode_literals
21
22 import sys
23 import gpg
24
25 del absolute_import, print_function, unicode_literals
26
27 if len(sys.argv) != 2:
28     sys.exit("Usage: %s <Gpg key pattern>\n" % sys.argv[0])
29
30 name = sys.argv[1]
31
32 with gpg.Context() as c:
33     keys = list(c.keylist(name))
34     if len(keys) == 0:
35         sys.exit("No key matching {}.".format(name))
36     if len(keys) > 1:
37         sys.exit("More than one key matching {}.".format(name))
38
39     key = keys[0]
40     print("Editing key {} ({}):".format(key.uids[0].uid, key.subkeys[0].fpr))
41
42     def edit_fnc(keyword, args):
43         print(
44             "Status: {}, args: {} > ".format(keyword, args),
45             end='',
46             flush=True)
47
48         if 'GET' not in keyword:
49             # no prompt
50             print()
51             return None
52
53         try:
54             return input()
55         except EOFError:
56             return "quit"
57
58     c.interact(key, edit_fnc, sink=sys.stdout)