Total Complexity | 2 |
Total Lines | 43 |
Duplicated Lines | 0 % |
Changes | 0 |
1 | import abc |
||
2 | from dataclasses import dataclass |
||
3 | from typing import Sequence |
||
4 | |||
5 | from mandos.model.pubchem_api import PubchemApi |
||
6 | from mandos.search.pubchem import PubchemHit, PubchemSearch |
||
7 | |||
8 | |||
9 | @dataclass(frozen=True, order=True, repr=True) |
||
10 | class DiseaseHit(PubchemHit): |
||
11 | evidence_type: str |
||
12 | |||
13 | |||
14 | class DiseaseSearch(PubchemSearch[DiseaseHit]): |
||
15 | """""" |
||
16 | |||
17 | @property |
||
18 | def data_source(self) -> str: |
||
19 | return "Comparative Toxicogenomics Database (CTD)" |
||
20 | |||
21 | def find(self, inchikey: str) -> Sequence[DiseaseHit]: |
||
22 | data = self.api.fetch_data(inchikey) |
||
23 | return [ |
||
24 | DiseaseHit( |
||
25 | record_id=dd.gid, |
||
26 | origin_inchikey=inchikey, |
||
27 | matched_inchikey=data.names_and_identifiers.inchikey, |
||
28 | compound_id=str(data.cid), |
||
29 | compound_name=data.name, |
||
30 | predicate=f"has {dd.evidence_type} evidence for", |
||
31 | object_id=dd.disease_id, |
||
32 | object_name=dd.disease_name, |
||
33 | evidence_type=dd.evidence_type, |
||
34 | search_key=self.key, |
||
35 | search_class=self.search_class, |
||
36 | data_source=self.data_source, |
||
37 | ) |
||
38 | for dd in data.associated_disorders_and_diseases.associated_disorders_and_diseases |
||
39 | ] |
||
40 | |||
41 | |||
42 | __all__ = ["DiseaseHit", "DiseaseSearch"] |
||
43 |