Python 3 port of PyME
[gpgme.git] / lang / py3-pyme / examples / t-edit.py
1 #!/usr/bin/env python3
2 # $Id$
3 # Copyright (C) 2005 Igor Belyi <belyi@users.sourceforge.net>
4 #
5 #    This program is free software; you can redistribute it and/or modify
6 #    it under the terms of the GNU General Public License as published by
7 #    the Free Software Foundation; either version 2 of the License, or
8 #    (at your option) any later version.
9 #
10 #    This program is distributed in the hope that it will be useful,
11 #    but WITHOUT ANY WARRANTY; without even the implied warranty of
12 #    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
13 #    GNU General Public License for more details.
14 #
15 #    You should have received a copy of the GNU General Public License
16 #    along with this program; if not, write to the Free Software
17 #    Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
18
19 import os
20 from pyme import core
21 from pyme.core import Data, Context
22
23 core.check_version(None)
24
25 class KeyEditor:
26     def __init__(self):
27         self.steps = ["fpr", "expire", "1", "primary", "quit"]
28         self.step = 0
29
30     def edit_fnc(self, status, args, out):
31         print("[-- Response --]")
32         out.seek(0,0)
33         print(out.read(), end=' ')
34         print("[-- Code: %d, %s --]" % (status, args))
35     
36         if args == "keyedit.prompt":
37             result = self.steps[self.step]
38             self.step += 1
39         elif args == "keyedit.save.okay":
40             result = "Y"
41         elif args == "keygen.valid":
42             result = "0"
43         else:
44             result = None
45
46         return result
47
48 if not os.getenv("GNUPGHOME"):
49     print("Please, set GNUPGHOME env.var. pointing to GPGME's tests/gpg dir")
50 else:
51     c = Context()
52     c.set_passphrase_cb(lambda x,y,z: "abc")
53     out = Data()
54     c.op_keylist_start("Alpha", 0)
55     key = c.op_keylist_next()
56     c.op_edit(key, KeyEditor().edit_fnc, out, out)
57     print("[-- Last response --]")
58     out.seek(0,0)
59     print(out.read(), end=' ')