1
|
|
|
from dataclasses import dataclass |
|
|
|
|
2
|
|
|
from typing import Sequence, TypeVar |
3
|
|
|
|
4
|
|
|
from mandos.model.pubchem_api import PubchemApi |
|
|
|
|
5
|
|
|
from mandos.search.pubchem import PubchemHit, PubchemSearch |
6
|
|
|
|
7
|
|
|
|
8
|
|
|
@dataclass(frozen=True, order=True, repr=True) |
9
|
|
|
class _DrugbankInteractionHit(PubchemHit): |
10
|
|
|
"""""" |
11
|
|
|
|
12
|
|
|
gene_symbol: str |
13
|
|
|
protein_id: str |
14
|
|
|
target_name: str |
15
|
|
|
general_function: str |
16
|
|
|
specific_function: str |
17
|
|
|
|
18
|
|
|
|
19
|
|
|
@dataclass(frozen=True, order=True, repr=True) |
|
|
|
|
20
|
|
|
class DrugbankTargetHit(_DrugbankInteractionHit): |
21
|
|
|
"""""" |
22
|
|
|
|
23
|
|
|
|
24
|
|
|
@dataclass(frozen=True, order=True, repr=True) |
|
|
|
|
25
|
|
|
class DrugbankGeneralFunctionHit(_DrugbankInteractionHit): |
26
|
|
|
"""""" |
27
|
|
|
|
28
|
|
|
|
29
|
|
|
T = TypeVar("T", bound=_DrugbankInteractionHit, covariant=True) |
|
|
|
|
30
|
|
|
|
31
|
|
|
|
32
|
|
|
class _DrugbankInteractionSearch(PubchemSearch[T]): |
33
|
|
|
"""""" |
34
|
|
|
|
35
|
|
|
@property |
36
|
|
|
def data_source(self) -> str: |
|
|
|
|
37
|
|
|
return "DrugBank" |
38
|
|
|
|
39
|
|
|
@classmethod |
40
|
|
|
def _attr(cls): |
41
|
|
|
raise NotImplementedError() |
42
|
|
|
|
43
|
|
|
def find(self, inchikey: str) -> Sequence[T]: |
|
|
|
|
44
|
|
|
data = self.api.fetch_data(inchikey) |
45
|
|
|
return [ |
46
|
|
|
self.__class__.get_h()( |
47
|
|
|
record_id=dd.record_id, |
48
|
|
|
origin_inchikey=inchikey, |
49
|
|
|
matched_inchikey=data.names_and_identifiers.inchikey, |
50
|
|
|
compound_id=str(data.cid), |
51
|
|
|
compound_name=data.name, |
52
|
|
|
predicate=dd.action, |
53
|
|
|
object_id=dd.protein_id, |
54
|
|
|
object_name=getattr(dd, self.__class__._attr()), |
|
|
|
|
55
|
|
|
search_key=self.key, |
56
|
|
|
search_class=self.search_class, |
57
|
|
|
data_source=self.data_source, |
58
|
|
|
gene_symbol=dd.gene_symbol, |
59
|
|
|
protein_id=dd.protein_id, |
60
|
|
|
target_name=dd.target_name, |
61
|
|
|
general_function=dd.general_function, |
62
|
|
|
specific_function=dd.specific_function, |
63
|
|
|
) |
64
|
|
|
for dd in data.biomolecular_interactions_and_pathways.drugbank_interactions |
65
|
|
|
] |
66
|
|
|
|
67
|
|
|
|
68
|
|
|
class DrugbankTargetSearch(_DrugbankInteractionSearch[_DrugbankInteractionHit]): |
|
|
|
|
69
|
|
|
"""""" |
70
|
|
|
|
71
|
|
|
@classmethod |
72
|
|
|
def _attr(cls): |
73
|
|
|
return "targetname" |
74
|
|
|
|
75
|
|
|
|
76
|
|
|
class DrugbankGeneralFunctionSearch(_DrugbankInteractionSearch[_DrugbankInteractionHit]): |
|
|
|
|
77
|
|
|
"""""" |
78
|
|
|
|
79
|
|
|
@classmethod |
80
|
|
|
def _attr(cls): |
81
|
|
|
return "generalfunction" |
82
|
|
|
|
83
|
|
|
|
84
|
|
|
__all__ = [ |
85
|
|
|
"DrugbankTargetHit", |
86
|
|
|
"DrugbankGeneralFunctionHit", |
87
|
|
|
"DrugbankTargetSearch", |
88
|
|
|
"DrugbankGeneralFunctionSearch", |
89
|
|
|
] |
90
|
|
|
|