Skip to content
Snippets Groups Projects
Select Git revision
  • master default protected
  • registrar-entities-contacts
2 results
Blame Permalink
entity_details.py 3.69 KiB
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 






#!/usr/bin/env python3

import sys
import argparse
import requests
from xml.etree import ElementTree as ET


# DEFINE SOME VARS

entity_id = None
metadata_file = None
root = None 
tec_contact = ''
sup_contact = ''
adm_contact = ''


# ARGPARSE

parser = argparse.ArgumentParser(description='Show detailed information about an eduGAIN entity.',
                                 formatter_class=argparse.RawDescriptionHelpFormatter,
                                 epilog='''
Examples

- Retrieve details about a single entity using eduGAIN metadata:\n
./entity_details.py https://idp.dir.garr.it/idp/shibboleth\n\n

- Retrieve details for a list of idps using a local metadata file:\n
./entity_details.py -e idp_list -f edugain-v2.xml
''')
group = parser.add_mutually_exclusive_group()
group.add_argument('ENTITYID', nargs='?', help='entityID of the entity to look up')
group.add_argument('-e', type=str, help='load entityIDs from file E')
parser.add_argument('-f', type=str, help='load metadata from file F')
parser.add_argument('-u', type=str, default='https://mds.edugain.org/edugain-v2.xml', 
                    help='download metadata from url U (default to https://mds.edugain.org/edugain-v2.xml)')
args = parser.parse_args()

# METHDO TO RETRIVE FED NAME

feds_request = requests.get("https://technical.edugain.org/api.php?action=list_feds&format")
feds = feds_request.json()

def get_fed_name(registration_authority):
    for key in feds:
        if feds[key]['reg_auth'] == registration_authority:
            return feds[key]['name']
    return None
    
# MAIN

entities = []

if args.ENTITYID:
    entities.append(args.ENTITYID)
elif args.e:
    with open(args.e, 'r') as entitiesfile:
        entities = entitiesfile.readlines()
else:
    parser.parse_args(['-h'])
    exit(1)

if args.f:
    tree = ET.parse(args.f)
    root = tree.getroot()
else:
    xml_req = requests.get(args.u)
    root = ET.fromstring(xml_req.content)

orgs = set()

ns = {
    'md': 'urn:oasis:names:tc:SAML:2.0:metadata',
    'mdui': 'urn:oasis:names:tc:SAML:metadata:ui',
    'shibmd': 'urn:mace:shibboleth:metadata:1.0',
    'remd': 'http://refeds.org/metadata',
    'icmd': 'http://id.incommon.org/metadata',
    'mdrpi': 'urn:oasis:names:tc:SAML:metadata:rpi',
}

print('entityID,FederationName,RegistrationAuthority,OrganizationName,TechnicalContact,SupportContact,AdministrativeContact')

for entity_id in entities: 

    entity = root.find(f'./md:EntityDescriptor[@entityID="{entity_id.strip()}"]', ns)
    
    if entity:

        registration_authority = None
        registration_info = entity.find('./md:Extensions/mdrpi:RegistrationInfo', ns)

        if registration_info:
            registration_authority = registration_info.attrib['registrationAuthority'].strip()
        fed_name = get_fed_name(registration_authority)

        orgname = entity.find('./md:Organization/md:OrganizationDisplayName', ns).text.strip()
        
        tec_contact_el = entity.find('./md:ContactPerson[@contactType="technical"]/md:EmailAddress', ns)
        if tec_contact_el is not None:
            tec_contact = tec_contact_el.text.replace('mailto:', '')
        sup_contact_el = entity.find('./md:ContactPerson[@contactType="support"]/md:EmailAddress', ns)
        if sup_contact_el is not None:
            sup_contact = sup_contact_el.text.replace('mailto:', '')
        adm_contact_el = entity.find('./md:ContactPerson[@contactType="administrative"]/md:EmailAddress', ns)
        if adm_contact_el is not None:
            adm_contact = adm_contact_el.text.replace('mailto:', '')
            
        print('{},{},{},{},{},{}'.format(entity_id, fed_name, registration_authority, orgname, tec_contact, sup_contact, adm_contact))

    else:

        print(f'No such entityID: {entity_id}')