Add identify and all_mappings client functions

- Add identify() to detect ID types of unknown identifiers
- Add all_mappings() to get all mappings across all target types
- Export new functions from utils __init__
- Add tests for both functions

Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>

Author: deeenes

omnipath_client/utils/__init__.py

@@ -14,10 +14,12 @@
 
 from omnipath_client.utils._mapping import (
     id_types,
+    identify,
     map_name,
     map_name0,
     map_names,
     translate,
+    all_mappings,
     translate_column,
     translate_columns,
 )

omnipath_client/utils/_mapping.py

@@ -243,3 +243,35 @@ def id_types() -> list[dict]:
     """List all supported ID types."""
 
     return _get('/mapping/id-types')
+
+
+def identify(
+    identifiers: list[str],
+    ncbi_tax_id: int = 9606,
+) -> dict[str, list[dict]]:
+    """Identify the type of given identifiers."""
+
+    return _get(
+        '/mapping/identify',
+        {
+            'identifiers': ','.join(identifiers),
+            'ncbi_tax_id': ncbi_tax_id,
+        },
+    ).get('results', {})
+
+
+def all_mappings(
+    identifiers: list[str],
+    id_type: str,
+    ncbi_tax_id: int = 9606,
+) -> dict[str, dict[str, list[str]]]:
+    """Get all known mappings for identifiers."""
+
+    return _get(
+        '/mapping/all',
+        {
+            'identifiers': ','.join(identifiers),
+            'id_type': id_type,
+            'ncbi_tax_id': ncbi_tax_id,
+        },
+    ).get('results', {})

tests/test_utils.py

@@ -283,6 +283,65 @@ def test_translate_with_min_sources(self, mock_get):
         assert call_params.get('min_sources') == 3
 
 
+
+class TestIdentify:
+
+    @patch('omnipath_client.utils._mapping._get')
+    def test_identify(self, mock_get):
+        mock_get.return_value = {
+            'results': {
+                'P04637': [
+                    {'id_type': 'uniprot', 'role': 'source', 'mappings_count': 5},
+                ],
+            },
+            'meta': {'ncbi_tax_id': 9606, 'total_input': 1},
+        }
+        from omnipath_client.utils import identify
+
+        result = identify(['P04637'])
+        assert 'P04637' in result
+        assert result['P04637'][0]['id_type'] == 'uniprot'
+
+    @patch('omnipath_client.utils._mapping._get')
+    def test_identify_params(self, mock_get):
+        mock_get.return_value = {'results': {}, 'meta': {}}
+        from omnipath_client.utils import identify
+
+        identify(['P04637', 'HMDB0000001'], ncbi_tax_id=10090)
+        call_params = mock_get.call_args[0][1]
+        assert call_params['identifiers'] == 'P04637,HMDB0000001'
+        assert call_params['ncbi_tax_id'] == 10090
+
+
+class TestAllMappings:
+
+    @patch('omnipath_client.utils._mapping._get')
+    def test_all_mappings(self, mock_get):
+        mock_get.return_value = {
+            'results': {
+                'P04637': {'genesymbol': ['TP53'], 'entrez': ['7157']},
+            },
+            'meta': {},
+        }
+        from omnipath_client.utils import all_mappings
+
+        result = all_mappings(['P04637'], 'uniprot')
+        assert 'P04637' in result
+        assert 'genesymbol' in result['P04637']
+        assert result['P04637']['genesymbol'] == ['TP53']
+
+    @patch('omnipath_client.utils._mapping._get')
+    def test_all_mappings_params(self, mock_get):
+        mock_get.return_value = {'results': {}, 'meta': {}}
+        from omnipath_client.utils import all_mappings
+
+        all_mappings(['P04637'], 'uniprot', ncbi_tax_id=10090)
+        call_params = mock_get.call_args[0][1]
+        assert call_params['identifiers'] == 'P04637'
+        assert call_params['id_type'] == 'uniprot'
+        assert call_params['ncbi_tax_id'] == 10090
+
+
 class TestDownloadImportFix:
     """Verify the dlmachine import fix."""