diff --git a/ippisite/ippidb/ws.py b/ippisite/ippidb/ws.py index 2c178fc1e80ca0bdb39e41d30571bde6756a9797..7a272f2a30e659ae65b408f4825b2085c2c3a9d4 100644 --- a/ippisite/ippidb/ws.py +++ b/ippisite/ippidb/ws.py @@ -78,3 +78,8 @@ def get_pfam_info(pfam_acc): domain_family = '' return {'id': pfam_id, 'description': description} + +def get_pdb_uniprot_mapping(pdb_id): + resp = requests.get('https://www.ebi.ac.uk/pdbe/api/mappings/uniprot/{}'.format(pdb_id)) + uniprot_ids = list(resp.json()[pdb_id]['UniProt'].keys()) + return uniprot_ids