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