Total Complexity | 2 |
Total Lines | 43 |
Duplicated Lines | 0 % |
Changes | 0 |
1 | from dataclasses import dataclass |
||
|
|||
2 | from typing import Sequence |
||
3 | |||
4 | from mandos.search.pubchem import PubchemHit, PubchemSearch |
||
5 | |||
6 | |||
7 | @dataclass(frozen=True, order=True, repr=True) |
||
8 | class DrugbankDdiHit(PubchemHit): |
||
9 | """""" |
||
10 | |||
11 | description: str |
||
12 | |||
13 | |||
14 | class DrugbankDdiSearch(PubchemSearch[DrugbankDdiHit]): |
||
15 | """""" |
||
16 | |||
17 | @property |
||
18 | def data_source(self) -> str: |
||
19 | return "DrugBank" |
||
20 | |||
21 | def find(self, inchikey: str) -> Sequence[DrugbankDdiHit]: |
||
22 | data = self.api.fetch_data(inchikey) |
||
23 | return [ |
||
24 | DrugbankDdiHit( |
||
25 | record_id=None, |
||
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="ddi", |
||
31 | object_id=dd.drug_drugbank_id, |
||
32 | object_name=dd.drug_drugbank_id, |
||
33 | search_key=self.key, |
||
34 | search_class=self.search_class, |
||
35 | data_source=self.data_source, |
||
36 | description=dd.description, |
||
37 | ) |
||
38 | for dd in data.biomolecular_interactions_and_pathways.drugbank_ddis |
||
39 | ] |
||
40 | |||
41 | |||
42 | __all__ = ["DrugbankDdiHit", "DrugbankDdiSearch"] |
||
43 |