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
|
"""
rcynic database iterator.
At least for the moment, we attempt to provide an iterator that works
with both old-style (directory tree of file objects with names similar
to what wget would use) and new style (Django ORM) databases.
"""
import os
initialized_django = False
def _uri_to_class(uri, class_map):
return class_map[uri[uri.rindex(".")+1:]]
def authenticated_objects(directory_tree = None, uri_suffix = None, class_map = None):
if class_map is None:
import rpki.POW
class_map = dict(cer = rpki.POW.X509,
crl = rpki.POW.CRL,
gbr = rpki.POW.CMS,
mft = rpki.POW.Manifest,
roa = rpki.POW.ROA)
if directory_tree:
for head, dirs, files in os.walk(directory_tree):
for fn in files:
if uri_suffix is None or fn.endswith(uri_suffix):
fn = os.path.join(head, fn)
uri = "rsync://" + fn[len(directory_tree):].lstrip("/")
yield uri, _uri_to_class(uri, class_map).derReadFile(fn)
return
global initialized_django
if not initialized_django:
os.environ.update(DJANGO_SETTINGS_MODULE = "rpki.django_settings.rcynic")
import django
django.setup()
initialized_django = True
import rpki.rcynicdb
auth = rpki.rcynicdb.models.Authenticated.objects.order_by("-started").first()
if auth is None:
return
q = auth.rpkiobject_set
for obj in q.filter(uri__endswith = uri_suffix) if uri_suffix else q.all():
yield obj.uri, _uri_to_class(obj.uri, class_map).derRead(obj.der)
|
