275a01344f506eb9978f2146b845a7f1bd09ed86
[debianmemberportfolio.git] / debianmemberportfolio / model / keyfinder.py
1 # -*- python -*-
2 # -*- coding: utf-8 -*-
3 #
4 # Debian Member Portfolio Service key finder module
5 # Copyright (c) 2009-2014 Jan Dittberner <jan@dittberner.info>
6 #
7 # This file is part of Debian Member Portfolio Service.
8 #
9 # Debian Member Portfolio Service is free software: you can redistribute it
10 # and/or modify it under the terms of the GNU Affero General Public License as
11 # published by the Free Software Foundation, either version 3 of the License,
12 # or (at your option) any later version.
13 #
14 # Debian Member Portfolio Service is distributed in the hope that it will be
15 # useful, but WITHOUT ANY WARRANTY; without even the implied warranty of
16 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU Affero
17 # General Public License for more details.
18 #
19 # You should have received a copy of the GNU Affero General Public
20 # License along with this program.  If not, see <http://www.gnu.org/licenses/>.
21 #
22 """
23 This module provides tools for finding PGP key information from a
24 given keyring.
25 """
26 import logging
27 import time
28 import sys
29
30 db = None
31 cachetimestamp = 0
32
33
34 def _get_keyring_cache():
35     global db, cachetimestamp
36     if db is None or (time.time() - cachetimestamp) > 86300:
37         import anydbm
38         import pkg_resources
39         import os.path
40         filename = pkg_resources.resource_filename(__name__,
41                                                    'keyringcache')
42         logging.debug('reading cache data from %s', filename)
43         assert os.path.exists(filename) and os.path.isfile(filename)
44         db = anydbm.open(filename, 'r')
45         cachetimestamp = time.time()
46     return db
47
48
49 def _get_cached(cachekey):
50     cache = _get_keyring_cache()
51     logging.debug('cache lookup for %s', cachekey)
52     if cachekey in cache:
53         logging.debug('found entry %s', cache[cachekey])
54         return cache[cachekey]
55     return None
56
57
58 def getFingerprintByEmail(email):
59     """
60     Gets the fingerprints associated with the given email address if
61     available.
62     """
63     return _get_cached('fpr:email:%s' % email)
64
65
66 def getRealnameByEmail(email):
67     """
68     Gets the real names associated with the given email address if
69     available.
70     """
71     return _get_cached('name:email:%s' % email)
72
73
74 def getLoginByEmail(email):
75     """
76     Gets the logins associated with the given email address if
77     available.
78     """
79     return _get_cached('login:email:%s' % email)
80
81
82 def getLoginByFingerprint(fpr):
83     """
84     Gets the login associated with the given fingerprint if available.
85     """
86     return _get_cached('login:fpr:%s' % fpr)
87
88
89 def _dump_cache():
90     cache = _get_keyring_cache()
91     fprs = []
92     for key in cache.keys():
93         if key.startswith('email:fpr:'):
94             fpr = key.replace('email:fpr:', '')
95             if not fpr in fprs:
96                 fprs.append(fpr)
97
98     for fpr in fprs:
99         login = getLoginByFingerprint(fpr)
100         email = _get_cached('email:fpr:%s' % fpr)
101         name = _get_cached('name:fpr:%s' % fpr)
102
103         print fpr, login, ':'
104         print '   ', name, email
105
106
107 if __name__ == '__main__':
108     logging.basicConfig(stream=sys.stderr, level=logging.WARNING)
109     _dump_cache()