summaryrefslogtreecommitdiff
path: root/src/mailman_pgp/model/address.py
blob: a912e8770dd4914a9bfa006ebabb27e3fcac248b (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
# Copyright (C) 2017 Jan Jancar
#
# This file is a part of the Mailman PGP plugin.
#
# This program is free software; you can redistribute it and/or modify it under
# the terms of the GNU General Public License as published by the Free
# Software Foundation, either version 3 of the License, or (at your option)
# any later version.
#
# This program is distributed in the hope that it will be useful, but WITHOUT
# ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
# FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License for
# more details.
#
# You should have received a copy of the GNU General Public License along with
# this program.  If not, see <http://www.gnu.org/licenses/>.

"""Model for PGP enabled addresses."""

from mailman.database.types import SAUnicode
from mailman.interfaces.usermanager import IUserManager
from sqlalchemy import Boolean, Column, Integer, String
from sqlalchemy.orm import reconstructor
from zope.component import getUtility

from mailman_pgp.config import config
from mailman_pgp.model.base import Base
from mailman_pgp.model.fs_key import FSKey


class PGPAddress(Base):
    """A PGP enabled address."""

    __tablename__ = 'pgp_addresses'

    id = Column(Integer, primary_key=True)
    email = Column(SAUnicode, index=True, unique=True)
    key_fingerprint = Column(String(50))
    key_confirmed = Column(Boolean, default=False)

    def __init__(self, address):
        super().__init__(email=address.email)
        self._init()
        self._address = address

    @reconstructor
    def _init(self):
        self._address = None
        self._key = FSKey(config.pgp.keydir_config['user_keydir'],
                          self.email + '.asc', True)

    @property
    def key(self):
        """

        :return:
        :rtype: pgpy.PGPKey
        """
        self._key.reload()
        return self._key.key

    @key.setter
    def key(self, value):
        """

        :param value:
        :type value: pgpy.PGPKey
        """
        if value is None:
            self.key_fingerprint = None
        else:
            self.key_fingerprint = value.fingerprint
        self._key.key = value
        self._key.save()

    @property
    def key_path(self):
        """

        :return:
        :rtype: str
        """
        return self._key.key_path

    @property
    def address(self):
        """

        :return:
        :rtype: mailman.model.address.Address
        """
        if self._address is None:
            self._address = getUtility(IUserManager).get_address(self.email)
        return self._address

    @staticmethod
    def for_address(address):
        """

        :param address:
        :return:
        :rtype: PGPAddress|None
        """
        if address is None:
            return None
        return PGPAddress.for_email(address.email)

    @staticmethod
    def for_email(email):
        """

        :param email:
        :return:
        :rtype: PGPAddress|None
        """
        if email is None:
            return None
        return PGPAddress.query().filter_by(email=email).first()