8bbe4095cf9528ee7abdf930f6b8f3089326bcdc
[gpgme.git] / lang / python / examples / howto / export-secret-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
29 import os.path
30 import sys
31
32 print("""
33 This script exports one or more secret keys.
34
35 The gpg-agent and pinentry are invoked to authorise the export.
36 """)
37
38 c = gpg.Context(armor=True)
39
40 if len(sys.argv) >= 4:
41     keyfile = sys.argv[1]
42     logrus = sys.argv[2]
43     homedir = sys.argv[3]
44 elif len(sys.argv) == 3:
45     keyfile = sys.argv[1]
46     logrus = sys.argv[2]
47     homedir = input("Enter the GPG configuration directory path (optional): ")
48 elif len(sys.argv) == 2:
49     keyfile = sys.argv[1]
50     logrus = input("Enter the UID matching the secret key(s) to export: ")
51     homedir = input("Enter the GPG configuration directory path (optional): ")
52 else:
53     keyfile = input("Enter the path and filename to save the secret key to: ")
54     logrus = input("Enter the UID matching the secret key(s) to export: ")
55     homedir = input("Enter the GPG configuration directory path (optional): ")
56
57 if homedir.startswith("~"):
58     if os.path.exists(os.path.expanduser(homedir)) is True:
59         c.home_dir = os.path.expanduser(homedir)
60     else:
61         pass
62 elif os.path.exists(homedir) is True:
63     c.home_dir = homedir
64 else:
65     pass
66
67 try:
68     result = c.key_export_secret(pattern=logrus)
69 except:
70     result = c.key_export_secret(pattern=None)
71
72 if result is not None:
73     with open(keyfile, "wb") as f:
74         f.write(result)
75     os.chmod(keyfile, 0o600)
76 else:
77     pass