rev |
line source |
me@261
|
1 import sys |
me@261
|
2 import os |
me@262
|
3 import os.path |
me@261
|
4 from tempfile import NamedTemporaryFile |
me@262
|
5 import urllib2 |
me@261
|
6 |
me@261
|
7 import config |
me@261
|
8 from graph import Graph |
me@262
|
9 from Bio.PDB.DSSP import make_dssp_dict |
bnagaev@249
|
10 from fasta import save_fasta |
me@260
|
11 import data.codes |
me@260
|
12 |
me@260
|
13 class MonomerType(object): |
me@260
|
14 """Class of monomer types. |
me@260
|
15 |
me@260
|
16 Each MonomerType object represents a known monomer type, e.g. Valine, |
me@260
|
17 and is referenced to by each instance of monomer in a given sequence. |
me@260
|
18 |
me@260
|
19 - `name`: full name of monomer type |
me@260
|
20 - `code1`: one-letter code |
me@260
|
21 - `code3`: three-letter code |
me@260
|
22 - `is_modified`: either of True or False |
me@260
|
23 |
me@260
|
24 class atributes: |
me@260
|
25 |
me@260
|
26 - `by_code1`: a mapping from one-letter code to MonomerType object |
me@260
|
27 - `by_code3`: a mapping from three-letter code to MonomerType object |
me@260
|
28 - `by_name`: a mapping from monomer name to MonomerType object |
me@260
|
29 - `instance_type`: class of Monomer objects to use when creating new |
me@260
|
30 objects; this must be redefined in descendent classes |
me@260
|
31 |
me@260
|
32 All of the class attributes MUST be redefined when subclassing. |
me@260
|
33 """ |
me@260
|
34 |
me@260
|
35 by_code1 = {} |
me@260
|
36 by_code3 = {} |
me@260
|
37 by_name = {} |
me@260
|
38 instance_type = None |
me@260
|
39 |
me@260
|
40 def __init__(self, name="", code1="", code3="", is_modified=False): |
me@260
|
41 self.name = name.capitalize() |
me@260
|
42 self.code1 = code1.upper() |
me@260
|
43 self.code3 = code3.upper() |
me@260
|
44 self.is_modified = bool(is_modified) |
me@260
|
45 if not is_modified: |
me@260
|
46 self.by_code1[self.code1] = self |
me@260
|
47 self.by_code3[code3] = self |
me@260
|
48 self.by_name[name] = self |
me@260
|
49 # We duplicate distinguished long names into MonomerType itself, |
me@260
|
50 # so that we can use MonomerType.from_code3 to create the relevant |
me@260
|
51 # type of monomer. |
me@260
|
52 MonomerType.by_code3[code3] = self |
me@260
|
53 MonomerType.by_name[name] = self |
me@260
|
54 |
me@260
|
55 @classmethod |
me@260
|
56 def _initialize(cls, type_letter, codes=data.codes.codes): |
me@260
|
57 """Create all relevant instances of MonomerType. |
me@260
|
58 |
me@260
|
59 `type_letter` is either of: |
me@260
|
60 |
me@260
|
61 - 'p' for protein |
me@260
|
62 - 'd' for DNA |
me@260
|
63 - 'r' for RNA |
me@260
|
64 |
me@260
|
65 `codes` is a table of monomer codes |
me@260
|
66 """ |
me@260
|
67 for type, code1, is_modified, code3, name in codes: |
me@260
|
68 if type == type_letter: |
me@260
|
69 cls(name, code1, code3, is_modified) |
me@260
|
70 |
me@260
|
71 @classmethod |
me@260
|
72 def from_code1(cls, code1): |
me@260
|
73 """Return monomer type by one-letter code.""" |
me@260
|
74 return cls.by_code1[code1.upper()] |
me@260
|
75 |
me@260
|
76 @classmethod |
me@260
|
77 def from_code3(cls, code3): |
me@260
|
78 """Return monomer type by three-letter code.""" |
me@260
|
79 return cls.by_code3[code3.upper()] |
me@260
|
80 |
me@260
|
81 @classmethod |