From 09803c4a81b9431fd4c8f30abb1c60c4c735f0cb Mon Sep 17 00:00:00 2001 From: Justus Winter Date: Tue, 24 May 2016 12:29:32 +0200 Subject: python: Improve support for edit callbacks. * lang/python/helpers.c (pyEditCb): Stash exceptions. * lang/python/pyme/core.py (Context.op_edit): Hand in 'self'. * lang/python/tests/Makefile.am (py_tests): Add new test. * lang/python/tests/t-callbacks.py: Test edit callbacks. * lang/python/tests/t-edit.py: New file. Signed-off-by: Justus Winter --- lang/python/tests/t-edit.py | 62 +++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 62 insertions(+) create mode 100755 lang/python/tests/t-edit.py (limited to 'lang/python/tests/t-edit.py') diff --git a/lang/python/tests/t-edit.py b/lang/python/tests/t-edit.py new file mode 100755 index 00000000..64255c95 --- /dev/null +++ b/lang/python/tests/t-edit.py @@ -0,0 +1,62 @@ +#!/usr/bin/env python3 + +# Copyright (C) 2005 Igor Belyi +# Copyright (C) 2016 g10 Code GmbH +# +# This file is part of GPGME. +# +# GPGME is free software; you can redistribute it and/or modify it +# under the terms of the GNU General Public License as published by +# the Free Software Foundation; either version 2 of the License, or +# (at your option) any later version. +# +# GPGME is distributed in the hope that it will be useful, but WITHOUT +# ANY WARRANTY; without even the implied warranty of MERCHANTABILITY +# or FITNESS FOR A PARTICULAR PURPOSE. See the GNU Lesser General +# Public License for more details. +# +# You should have received a copy of the GNU Lesser General Public +# License along with this program; if not, see . + +import sys +import os +from pyme import core, constants +import support + +class KeyEditor(object): + def __init__(self): + self.steps = ["fpr", "expire", "1", "primary", "quit"] + self.step = 0 + self.done = False + self.verbose = int(os.environ.get('verbose', 0)) > 1 + + def edit_fnc(self, status, args, out): + if args == "keyedit.prompt": + result = self.steps[self.step] + self.step += 1 + elif args == "keyedit.save.okay": + result = "Y" + self.done = self.step == len(self.steps) + elif args == "keygen.valid": + result = "0" + else: + result = None + + if self.verbose: + sys.stderr.write("Code: {}, args: {!r}, Returning: {!r}\n" + .format(status, args, result)) + + return result + +support.init_gpgme(constants.PROTOCOL_OpenPGP) + +c = core.Context() +c.set_pinentry_mode(constants.PINENTRY_MODE_LOOPBACK) +c.set_passphrase_cb(lambda *args: "abc") +c.set_armor(True) +sink = core.Data() + +editor = KeyEditor() +c.op_edit(c.get_key("A0FF4590BB6122EDEF6E3C542D727CC768697734", False), + editor.edit_fnc, sink, sink) +assert editor.done -- cgit v1.2.3