f7f62662f488c7997b95976b499086f0a43156c4
[gpgme.git] / lang / python / examples / howto / pmkey-import-hkp.py
1 #!/usr/bin/env python3
2 # -*- coding: utf-8 -*-
3
4 from __future__ import absolute_import, division, unicode_literals
5
6 import gpg
7 import hkp4py
8 import os.path
9 import sys
10
11 del absolute_import, division, unicode_literals
12
13 # Copyright (C) 2018 Ben McGinnes <ben@gnupg.org>
14 #
15 # This program is free software; you can redistribute it and/or modify it under
16 # the terms of the GNU General Public License as published by the Free Software
17 # Foundation; either version 2 of the License, or (at your option) any later
18 # version.
19 #
20 # This program is free software; you can redistribute it and/or modify it under
21 # the terms of the GNU Lesser General Public License as published by the Free
22 # Software Foundation; either version 2.1 of the License, or (at your option)
23 # any later version.
24 #
25 # This program is distributed in the hope that it will be useful, but WITHOUT
26 # ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
27 # FOR A PARTICULAR PURPOSE.  See the GNU General Public License and the GNU
28 # Lesser General Public License for more details.
29 #
30 # You should have received a copy of the GNU General Public License and the GNU
31 # Lesser General Public along with this program; if not, see
32 # <https://www.gnu.org/licenses/>.
33
34 print("""
35 This script searches the ProtonMail key server for the specified key and
36 imports it.
37
38 Usage:  pmkey-import-hkp.py [search strings]
39 """)
40
41 c = gpg.Context(armor=True)
42 server = hkp4py.KeyServer("hkps://api.protonmail.ch")
43 keyterms = []
44 ksearch = []
45 allkeys = []
46 results = []
47 paradox = []
48 homeless = None
49
50 if len(sys.argv) > 2:
51     keyterms = sys.argv[1:]
52 elif len(sys.argv) == 2:
53     keyterm = sys.argv[1]
54     keyterms.append(keyterm)
55 else:
56     key_term = input("Enter the key ID, UID or search string: ")
57     keyterms = key_term.split()
58
59 for keyterm in keyterms:
60     if keyterm.count("@") == 2 and keyterm.startswith("@") is True:
61         ksearch.append(keyterm[1:])
62         ksearch.append(keyterm[1:])
63         ksearch.append(keyterm[1:])
64     elif keyterm.count("@") == 1 and keyterm.startswith("@") is True:
65         ksearch.append("{0}@protonmail.com".format(keyterm[1:]))
66         ksearch.append("{0}@protonmail.ch".format(keyterm[1:]))
67         ksearch.append("{0}@pm.me".format(keyterm[1:]))
68     elif keyterm.count("@") == 0:
69         ksearch.append("{0}@protonmail.com".format(keyterm))
70         ksearch.append("{0}@protonmail.ch".format(keyterm))
71         ksearch.append("{0}@pm.me".format(keyterm))
72     elif keyterm.count("@") == 2 and keyterm.startswith("@") is False:
73         uidlist = keyterm.split("@")
74         for uid in uidlist:
75             ksearch.append("{0}@protonmail.com".format(uid))
76             ksearch.append("{0}@protonmail.ch".format(uid))
77             ksearch.append("{0}@pm.me".format(uid))
78     elif keyterm.count("@") > 2:
79         uidlist = keyterm.split("@")
80         for uid in uidlist:
81             ksearch.append("{0}@protonmail.com".format(uid))
82             ksearch.append("{0}@protonmail.ch".format(uid))
83             ksearch.append("{0}@pm.me".format(uid))
84     else:
85         ksearch.append(keyterm)
86
87 for k in ksearch:
88     print("Checking for key for: {0}".format(k))
89     try:
90         keys = server.search(k)
91         if isinstance(keys, list) is True:
92             for key in keys:
93                 allkeys.append(key)
94                 try:
95                     import_result = c.key_import(key.key_blob)
96                 except Exception as e:
97                     import_result = c.key_import(key.key)
98         else:
99             paradox.append(keys)
100             import_result = None
101     except Exception as e:
102         import_result = None
103     results.append(import_result)
104
105 for result in results:
106     if result is not None and hasattr(result, "considered") is False:
107         print("{0} for {1}".format(result.decode(), k))
108     elif result is not None and hasattr(result, "considered") is True:
109         num_keys = len(result.imports)
110         new_revs = result.new_revocations
111         new_sigs = result.new_signatures
112         new_subs = result.new_sub_keys
113         new_uids = result.new_user_ids
114         new_scrt = result.secret_imported
115         nochange = result.unchanged
116         print("""
117 The total number of keys considered for import was:  {0}
118
119 With UIDs wholely or partially matching the following string:
120
121         {1}
122
123    Number of keys revoked:  {2}
124  Number of new signatures:  {3}
125     Number of new subkeys:  {4}
126    Number of new user IDs:  {5}
127 Number of new secret keys:  {6}
128  Number of unchanged keys:  {7}
129
130 The key IDs for all considered keys were:
131 """.format(num_keys, k, new_revs, new_sigs, new_subs, new_uids, new_scrt,
132            nochange))
133         for i in range(num_keys):
134             print(result.imports[i].fpr)
135         print("")
136     elif result is None:
137         pass