84f8d103596f49a1d7d326bac5797c1168695231
[gpgme.git] / lang / python / examples / howto / encrypt-to-group.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 sys
29 from groups import group_lists
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 gpg.errors.InvalidRecipients as e:
73         for i in range(len(e.recipients)):
74             for n in range(len(logrus)):
75                 if logrus[n].fpr == e.recipients[i].fpr:
76                     logrus.remove(logrus[n])
77                 else:
78                     pass
79         try:
80             ciphertext, result, sign_result = c.encrypt(text,
81                                                         recipients=logrus,
82                                                         add_encrypt_to=True,
83                                                         always_trust=True)
84         except:
85             pass
86     with open("{0}.asc".format(filepath), "wb") as f:
87               f.write(ciphertext)
88 else:
89     pass
90
91 # EOF