459df114bd6678d7a9f719e1290faa6c09c06293
[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 <http://www.gnu.org/licenses/>.
18
19 """Simple interactive editor to test editor scripts"""
20
21 from __future__ import absolute_import, print_function, unicode_literals
22 del absolute_import, print_function, unicode_literals
23
24 import sys
25 import pyme
26 import pyme.constants.status
27
28 # Get names for the status codes
29 status2str = {}
30 for name in dir(pyme.constants.status):
31     if not name.startswith('__') and name != "util":
32         status2str[getattr(pyme.constants.status, name)] = name
33
34 if len(sys.argv) != 2:
35     sys.exit("Usage: %s <Gpg key pattern>\n" % sys.argv[0])
36
37 name = sys.argv[1]
38
39 with pyme.Context() as c:
40     keys = list(c.keylist(name))
41     if len(keys) == 0:
42         sys.exit("No key matching {}.".format(name))
43     if len(keys) > 1:
44         sys.exit("More than one key matching {}.".format(name))
45
46     key = keys[0]
47     print("Editing key {} ({}):".format(key.uids[0].uid, key.subkeys[0].fpr))
48
49     def edit_fnc(status, args):
50         print("Status: {} ({}), args: {} > ".format(
51             status2str[status], status, args), end='', flush=True)
52
53         if not 'GET' in status2str[status]:
54             # no prompt
55             print()
56             return None
57
58         try:
59             return input()
60         except EOFError:
61             return "quit"
62
63     c.op_edit(key, edit_fnc, None, sys.stdout)