56cfe259d9d5e5760e7ded8a9e10e834b00a4198
[gpgme.git] / lang / python / examples / howto / import-key.py
1 #!/usr/bin/env python3
2 # -*- coding: utf-8 -*-
3
4 from __future__ import absolute_import, division, unicode_literals
5
6 # Copyright (C) 2018 Ben McGinnes <ben@gnupg.org>
7 #
8 # This program is free software; you can redistribute it and/or modify it under
9 # the terms of the GNU General Public License as published by the Free Software
10 # Foundation; either version 2 of the License, or (at your option) any later
11 # version.
12 #
13 # This program is free software; you can redistribute it and/or modify it under
14 # the terms of the GNU Lesser General Public License as published by the Free
15 # Software Foundation; either version 2.1 of the License, or (at your option)
16 # any later version.
17 #
18 # This program is distributed in the hope that it will be useful, but WITHOUT
19 # ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
20 # FOR A PARTICULAR PURPOSE.  See the GNU General Public License and the GNU
21 # Lesser General Public Licensefor more details.
22 #
23 # You should have received a copy of the GNU General Public License and the GNU
24 # Lesser General Public along with this program; if not, see
25 # <http://www.gnu.org/licenses/>.
26
27 import gpg
28 import os.path
29 import sys
30
31 print("""
32 This script exports one or more public keys.
33 """)
34
35 c = gpg.Context(armor=True)
36
37 if len(sys.argv) >= 3:
38     keyfile = sys.argv[1]
39     homedir = sys.argv[2]
40 elif len(sys.argv) == 2:
41     keyfile = sys.argv[1]
42     homedir = input("Enter the GPG configuration directory path (optional): ")
43 else:
44     keyfile = input("Enter the path and filename to import the key(s) from: ")
45     homedir = input("Enter the GPG configuration directory path (optional): ")
46
47 if homedir.startswith("~"):
48     if os.path.exists(os.path.expanduser(homedir)) is True:
49         c.home_dir = os.path.expanduser(homedir)
50     else:
51         pass
52 elif os.path.exists(homedir) is True:
53     c.home_dir = homedir
54 else:
55     pass
56
57 if os.path.isfile(keyfile) is True:
58     with open(keyfile, "rb") as f:
59         incoming = f.read()
60     result = c.key_import(incoming)
61 else:
62     result = None
63
64 if result is not None and hasattr(result, "considered") is False:
65     print(result)
66 elif result is not None and hasattr(result, "considered") is True:
67     num_keys = len(result.imports)
68     new_revs = result.new_revocations
69     new_sigs = result.new_signatures
70     new_subs = result.new_sub_keys
71     new_uids = result.new_user_ids
72     new_scrt = result.secret_imported
73     nochange = result.unchanged
74     print("""
75 The total number of keys considered for import was:  {0}
76
77    Number of keys revoked:  {1}
78  Number of new signatures:  {2}
79     Number of new subkeys:  {3}
80    Number of new user IDs:  {4}
81 Number of new secret keys:  {5}
82  Number of unchanged keys:  {6}
83
84 The key IDs for all considered keys were:
85 """.format(num_keys, new_revs, new_sigs, new_subs, new_uids, new_scrt,
86            nochange))
87     for i in range(num_keys):
88         print(result.imports[i].fpr)
89     print("")
90 elif result is None:
91     print("You must specify a key file to import.")