python bindings: howto examples
[gpgme.git] / lang / python / examples / howto / encrypt-to-group-gullible.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 sys
8 from groups import group_lists
9
10 # Copyright (C) 2018 Ben McGinnes <ben@gnupg.org>
11 #
12 # This program is free software; you can redistribute it and/or modify it under
13 # the terms of the GNU General Public License as published by the Free Software
14 # Foundation; either version 2 of the License, or (at your option) any later
15 # version.
16 #
17 # This program is free software; you can redistribute it and/or modify it under
18 # the terms of the GNU Lesser General Public License as published by the Free
19 # Software Foundation; either version 2.1 of the License, or (at your option)
20 # any later version.
21 #
22 # This program is distributed in the hope that it will be useful, but WITHOUT
23 # ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
24 # FOR A PARTICULAR PURPOSE.  See the GNU General Public License and the GNU
25 # Lesser General Public Licensefor more details.
26 #
27 # You should have received a copy of the GNU General Public License and the GNU
28 # Lesser General Public along with this program; if not, see
29 # <http://www.gnu.org/licenses/>.
30
31 """
32 Uses the groups module to encrypt to multiple recipients.
33
34 """
35
36 c = gpg.Context(armor=True)
37
38 if len(sys.argv) > 3:
39     group_id = sys.argv[1]
40     filepath = sys.argv[2:]
41 elif len(sys.argv) == 3:
42     group_id = sys.argv[1]
43     filepath = sys.argv[2]
44 elif len(sys.argv) == 2:
45     group_id = sys.argv[1]
46     filepath = input("Enter the filename to encrypt: ")
47 else:
48     group_id = input("Enter the group name to encrypt to: ")
49     filepath = input("Enter the filename to encrypt: ")
50
51 with open(filepath, "rb") as f:
52     text = f.read()
53
54 for i in range(len(group_lists)):
55     if group_lists[i][0] == group_id:
56         klist = group_lists[i][1]
57     else:
58         klist = None
59
60 logrus = []
61
62 if klist is not None:
63     for i in range(len(klist)):
64         apattern = list(c.keylist(pattern=klist[i], secret=False))
65         if apattern[0].can_encrypt == 1:
66             logrus.append(apattern[0])
67         else:
68             pass
69     try:
70         ciphertext, result, sign_result = c.encrypt(text, recipients=logrus,
71                                                     add_encrypt_to=True)
72     except:
73         ciphertext, result, sign_result = c.encrypt(text, recipients=logrus,
74                                                     add_encrypt_to=True,
75                                                     always_trust=True)
76     with open("{0}.asc".format(filepath), "wb") as f:
77         f.write(ciphertext)
78 else:
79     pass
80
81 # EOF