Mercurial > repos > bgruening > chembl
annotate chembl.py @ 5:e907c14ea4be draft
"planemo upload for repository https://github.com/chembl/chembl_webresource_client commit 944ea4bb8a9cd4244152a4a4fecd0485fabc2ad0"
author | bgruening |
---|---|
date | Tue, 28 Jul 2020 11:54:13 +0000 |
parents | 6aa23d7ae03b |
children | 9dbc85cd5780 |
rev | line source |
---|---|
1
0291a4af0db6
"planemo upload for repository https://github.com/chembl/chembl_webresource_client commit 594b8bcfa2730fb21de1efa83f89e0380fdcaa0f"
bgruening
parents:
0
diff
changeset
|
1 from chembl_webresource_client.settings import Settings |
0291a4af0db6
"planemo upload for repository https://github.com/chembl/chembl_webresource_client commit 594b8bcfa2730fb21de1efa83f89e0380fdcaa0f"
bgruening
parents:
0
diff
changeset
|
2 Settings.Instance().CACHING = False |
0
b60f2077abd8
planemo upload for repository https://github.com/chembl/chembl_webresource_client commit 2e3c3c2bd7ecdc9c2968a32f91e81136e0cb3835
bgruening
parents:
diff
changeset
|
3 from chembl_webresource_client.new_client import new_client |
b60f2077abd8
planemo upload for repository https://github.com/chembl/chembl_webresource_client commit 2e3c3c2bd7ecdc9c2968a32f91e81136e0cb3835
bgruening
parents:
diff
changeset
|
4 import argparse |
b60f2077abd8
planemo upload for repository https://github.com/chembl/chembl_webresource_client commit 2e3c3c2bd7ecdc9c2968a32f91e81136e0cb3835
bgruening
parents:
diff
changeset
|
5 |
b60f2077abd8
planemo upload for repository https://github.com/chembl/chembl_webresource_client commit 2e3c3c2bd7ecdc9c2968a32f91e81136e0cb3835
bgruening
parents:
diff
changeset
|
6 def open_file(filename): |
b60f2077abd8
planemo upload for repository https://github.com/chembl/chembl_webresource_client commit 2e3c3c2bd7ecdc9c2968a32f91e81136e0cb3835
bgruening
parents:
diff
changeset
|
7 with open(filename) as f: |
4
6aa23d7ae03b
"planemo upload for repository https://github.com/chembl/chembl_webresource_client commit f5c316339ed36a0469f4862d00fa0e5cafebc041"
bgruening
parents:
2
diff
changeset
|
8 return f.readline().split()[0] |
0
b60f2077abd8
planemo upload for repository https://github.com/chembl/chembl_webresource_client commit 2e3c3c2bd7ecdc9c2968a32f91e81136e0cb3835
bgruening
parents:
diff
changeset
|
9 |
b60f2077abd8
planemo upload for repository https://github.com/chembl/chembl_webresource_client commit 2e3c3c2bd7ecdc9c2968a32f91e81136e0cb3835
bgruening
parents:
diff
changeset
|
10 def get_smiles(res): |
b60f2077abd8
planemo upload for repository https://github.com/chembl/chembl_webresource_client commit 2e3c3c2bd7ecdc9c2968a32f91e81136e0cb3835
bgruening
parents:
diff
changeset
|
11 """ |
b60f2077abd8
planemo upload for repository https://github.com/chembl/chembl_webresource_client commit 2e3c3c2bd7ecdc9c2968a32f91e81136e0cb3835
bgruening
parents:
diff
changeset
|
12 Get a list of SMILES from function results |
b60f2077abd8
planemo upload for repository https://github.com/chembl/chembl_webresource_client commit 2e3c3c2bd7ecdc9c2968a32f91e81136e0cb3835
bgruening
parents:
diff
changeset
|
13 """ |
b60f2077abd8
planemo upload for repository https://github.com/chembl/chembl_webresource_client commit 2e3c3c2bd7ecdc9c2968a32f91e81136e0cb3835
bgruening
parents:
diff
changeset
|
14 smiles = set() |
b60f2077abd8
planemo upload for repository https://github.com/chembl/chembl_webresource_client commit 2e3c3c2bd7ecdc9c2968a32f91e81136e0cb3835
bgruening
parents:
diff
changeset
|
15 for smi in res: |
5
e907c14ea4be
"planemo upload for repository https://github.com/chembl/chembl_webresource_client commit 944ea4bb8a9cd4244152a4a4fecd0485fabc2ad0"
bgruening
parents:
4
diff
changeset
|
16 try: |
e907c14ea4be
"planemo upload for repository https://github.com/chembl/chembl_webresource_client commit 944ea4bb8a9cd4244152a4a4fecd0485fabc2ad0"
bgruening
parents:
4
diff
changeset
|
17 smiles.add('{}\t{}'.format(smi['molecule_structures']['canonical_smiles'], smi['molecule_chembl_id'])) |
e907c14ea4be
"planemo upload for repository https://github.com/chembl/chembl_webresource_client commit 944ea4bb8a9cd4244152a4a4fecd0485fabc2ad0"
bgruening
parents:
4
diff
changeset
|
18 except TypeError: |
e907c14ea4be
"planemo upload for repository https://github.com/chembl/chembl_webresource_client commit 944ea4bb8a9cd4244152a4a4fecd0485fabc2ad0"
bgruening
parents:
4
diff
changeset
|
19 continue |
0
b60f2077abd8
planemo upload for repository https://github.com/chembl/chembl_webresource_client commit 2e3c3c2bd7ecdc9c2968a32f91e81136e0cb3835
bgruening
parents:
diff
changeset
|
20 return smiles |
b60f2077abd8
planemo upload for repository https://github.com/chembl/chembl_webresource_client commit 2e3c3c2bd7ecdc9c2968a32f91e81136e0cb3835
bgruening
parents:
diff
changeset
|
21 |
b60f2077abd8
planemo upload for repository https://github.com/chembl/chembl_webresource_client commit 2e3c3c2bd7ecdc9c2968a32f91e81136e0cb3835
bgruening
parents:
diff
changeset
|
22 def sim_search(smiles, tanimoto): |
b60f2077abd8
planemo upload for repository https://github.com/chembl/chembl_webresource_client commit 2e3c3c2bd7ecdc9c2968a32f91e81136e0cb3835
bgruening
parents:
diff
changeset
|
23 """ |
b60f2077abd8
planemo upload for repository https://github.com/chembl/chembl_webresource_client commit 2e3c3c2bd7ecdc9c2968a32f91e81136e0cb3835
bgruening
parents:
diff
changeset
|
24 Return compounds which are within a Tanimoto range of the SMILES input |
b60f2077abd8
planemo upload for repository https://github.com/chembl/chembl_webresource_client commit 2e3c3c2bd7ecdc9c2968a32f91e81136e0cb3835
bgruening
parents:
diff
changeset
|
25 """ |
b60f2077abd8
planemo upload for repository https://github.com/chembl/chembl_webresource_client commit 2e3c3c2bd7ecdc9c2968a32f91e81136e0cb3835
bgruening
parents:
diff
changeset
|
26 similarity = new_client.similarity |
2
30af54fed083
"planemo upload for repository https://github.com/chembl/chembl_webresource_client commit d7fca21744d3c87cf173c1539c069a5c720084f3"
bgruening
parents:
1
diff
changeset
|
27 return similarity.filter(smiles=smiles, similarity=tanimoto).only(['molecule_structures', 'molecule_chembl_id']) |
0
b60f2077abd8
planemo upload for repository https://github.com/chembl/chembl_webresource_client commit 2e3c3c2bd7ecdc9c2968a32f91e81136e0cb3835
bgruening
parents:
diff
changeset
|
28 |
b60f2077abd8
planemo upload for repository https://github.com/chembl/chembl_webresource_client commit 2e3c3c2bd7ecdc9c2968a32f91e81136e0cb3835
bgruening
parents:
diff
changeset
|
29 def substr_search(smiles): |
b60f2077abd8
planemo upload for repository https://github.com/chembl/chembl_webresource_client commit 2e3c3c2bd7ecdc9c2968a32f91e81136e0cb3835
bgruening
parents:
diff
changeset
|
30 """ |
b60f2077abd8
planemo upload for repository https://github.com/chembl/chembl_webresource_client commit 2e3c3c2bd7ecdc9c2968a32f91e81136e0cb3835
bgruening
parents:
diff
changeset
|
31 Return compounds which contain the SMILES substructure input |
b60f2077abd8
planemo upload for repository https://github.com/chembl/chembl_webresource_client commit 2e3c3c2bd7ecdc9c2968a32f91e81136e0cb3835
bgruening
parents:
diff
changeset
|
32 """ |
b60f2077abd8
planemo upload for repository https://github.com/chembl/chembl_webresource_client commit 2e3c3c2bd7ecdc9c2968a32f91e81136e0cb3835
bgruening
parents:
diff
changeset
|
33 substructure = new_client.substructure |
2
30af54fed083
"planemo upload for repository https://github.com/chembl/chembl_webresource_client commit d7fca21744d3c87cf173c1539c069a5c720084f3"
bgruening
parents:
1
diff
changeset
|
34 return substructure.filter(smiles=smiles).only(['molecule_structures', 'molecule_chembl_id']) |
0
b60f2077abd8
planemo upload for repository https://github.com/chembl/chembl_webresource_client commit 2e3c3c2bd7ecdc9c2968a32f91e81136e0cb3835
bgruening
parents:
diff
changeset
|
35 |
b60f2077abd8
planemo upload for repository https://github.com/chembl/chembl_webresource_client commit 2e3c3c2bd7ecdc9c2968a32f91e81136e0cb3835
bgruening
parents:
diff
changeset
|
36 def filter_drugs(mols): |
b60f2077abd8
planemo upload for repository https://github.com/chembl/chembl_webresource_client commit 2e3c3c2bd7ecdc9c2968a32f91e81136e0cb3835
bgruening
parents:
diff
changeset
|
37 """ |
b60f2077abd8
planemo upload for repository https://github.com/chembl/chembl_webresource_client commit 2e3c3c2bd7ecdc9c2968a32f91e81136e0cb3835
bgruening
parents:
diff
changeset
|
38 Return only compounds which are approved drugs |
b60f2077abd8
planemo upload for repository https://github.com/chembl/chembl_webresource_client commit 2e3c3c2bd7ecdc9c2968a32f91e81136e0cb3835
bgruening
parents:
diff
changeset
|
39 """ |
b60f2077abd8
planemo upload for repository https://github.com/chembl/chembl_webresource_client commit 2e3c3c2bd7ecdc9c2968a32f91e81136e0cb3835
bgruening
parents:
diff
changeset
|
40 return mols.filter(max_phase=4) |
b60f2077abd8
planemo upload for repository https://github.com/chembl/chembl_webresource_client commit 2e3c3c2bd7ecdc9c2968a32f91e81136e0cb3835
bgruening
parents:
diff
changeset
|
41 |
b60f2077abd8
planemo upload for repository https://github.com/chembl/chembl_webresource_client commit 2e3c3c2bd7ecdc9c2968a32f91e81136e0cb3835
bgruening
parents:
diff
changeset
|
42 def filter_biotherapeutic(mols): |
b60f2077abd8
planemo upload for repository https://github.com/chembl/chembl_webresource_client commit 2e3c3c2bd7ecdc9c2968a32f91e81136e0cb3835
bgruening
parents:
diff
changeset
|
43 """ |
b60f2077abd8
planemo upload for repository https://github.com/chembl/chembl_webresource_client commit 2e3c3c2bd7ecdc9c2968a32f91e81136e0cb3835
bgruening
parents:
diff
changeset
|
44 Return only biotherapeutic molecules |
b60f2077abd8
planemo upload for repository https://github.com/chembl/chembl_webresource_client commit 2e3c3c2bd7ecdc9c2968a32f91e81136e0cb3835
bgruening
parents:
diff
changeset
|
45 """ |
b60f2077abd8
planemo upload for repository https://github.com/chembl/chembl_webresource_client commit 2e3c3c2bd7ecdc9c2968a32f91e81136e0cb3835
bgruening
parents:
diff
changeset
|
46 return mols.filter(biotherapeutic__isnull=False) |
b60f2077abd8
planemo upload for repository https://github.com/chembl/chembl_webresource_client commit 2e3c3c2bd7ecdc9c2968a32f91e81136e0cb3835
bgruening
parents:
diff
changeset
|
47 |
b60f2077abd8
planemo upload for repository https://github.com/chembl/chembl_webresource_client commit 2e3c3c2bd7ecdc9c2968a32f91e81136e0cb3835
bgruening
parents:
diff
changeset
|
48 def filter_nat_prod(mols): |
b60f2077abd8
planemo upload for repository https://github.com/chembl/chembl_webresource_client commit 2e3c3c2bd7ecdc9c2968a32f91e81136e0cb3835
bgruening
parents:
diff
changeset
|
49 """ |
b60f2077abd8
planemo upload for repository https://github.com/chembl/chembl_webresource_client commit 2e3c3c2bd7ecdc9c2968a32f91e81136e0cb3835
bgruening
parents:
diff
changeset
|
50 Return only natural products |
b60f2077abd8
planemo upload for repository https://github.com/chembl/chembl_webresource_client commit 2e3c3c2bd7ecdc9c2968a32f91e81136e0cb3835
bgruening
parents:
diff
changeset
|
51 """ |
b60f2077abd8
planemo upload for repository https://github.com/chembl/chembl_webresource_client commit 2e3c3c2bd7ecdc9c2968a32f91e81136e0cb3835
bgruening
parents:
diff
changeset
|
52 return mols.filter(natural_product=1) |
b60f2077abd8
planemo upload for repository https://github.com/chembl/chembl_webresource_client commit 2e3c3c2bd7ecdc9c2968a32f91e81136e0cb3835
bgruening
parents:
diff
changeset
|
53 |
b60f2077abd8
planemo upload for repository https://github.com/chembl/chembl_webresource_client commit 2e3c3c2bd7ecdc9c2968a32f91e81136e0cb3835
bgruening
parents:
diff
changeset
|
54 def filter_ro5(mols): |
b60f2077abd8
planemo upload for repository https://github.com/chembl/chembl_webresource_client commit 2e3c3c2bd7ecdc9c2968a32f91e81136e0cb3835
bgruening
parents:
diff
changeset
|
55 """ |
b60f2077abd8
planemo upload for repository https://github.com/chembl/chembl_webresource_client commit 2e3c3c2bd7ecdc9c2968a32f91e81136e0cb3835
bgruening
parents:
diff
changeset
|
56 Return only compounds with no RO5 violations |
b60f2077abd8
planemo upload for repository https://github.com/chembl/chembl_webresource_client commit 2e3c3c2bd7ecdc9c2968a32f91e81136e0cb3835
bgruening
parents:
diff
changeset
|
57 """ |
b60f2077abd8
planemo upload for repository https://github.com/chembl/chembl_webresource_client commit 2e3c3c2bd7ecdc9c2968a32f91e81136e0cb3835
bgruening
parents:
diff
changeset
|
58 return mols.filter(molecule_properties__num_ro5_violations=0) |
b60f2077abd8
planemo upload for repository https://github.com/chembl/chembl_webresource_client commit 2e3c3c2bd7ecdc9c2968a32f91e81136e0cb3835
bgruening
parents:
diff
changeset
|
59 |
b60f2077abd8
planemo upload for repository https://github.com/chembl/chembl_webresource_client commit 2e3c3c2bd7ecdc9c2968a32f91e81136e0cb3835
bgruening
parents:
diff
changeset
|
60 def main(): |
b60f2077abd8
planemo upload for repository https://github.com/chembl/chembl_webresource_client commit 2e3c3c2bd7ecdc9c2968a32f91e81136e0cb3835
bgruening
parents:
diff
changeset
|
61 parser = argparse.ArgumentParser(description='Search ChEMBL database for compounds') |
b60f2077abd8
planemo upload for repository https://github.com/chembl/chembl_webresource_client commit 2e3c3c2bd7ecdc9c2968a32f91e81136e0cb3835
bgruening
parents:
diff
changeset
|
62 parser.add_argument('-i', '--input', help='SMILES input') |
b60f2077abd8
planemo upload for repository https://github.com/chembl/chembl_webresource_client commit 2e3c3c2bd7ecdc9c2968a32f91e81136e0cb3835
bgruening
parents:
diff
changeset
|
63 parser.add_argument('-f', '--file', help='SMILES input as file') |
b60f2077abd8
planemo upload for repository https://github.com/chembl/chembl_webresource_client commit 2e3c3c2bd7ecdc9c2968a32f91e81136e0cb3835
bgruening
parents:
diff
changeset
|
64 parser.add_argument('-o', '--output', help="SMILES output") |
b60f2077abd8
planemo upload for repository https://github.com/chembl/chembl_webresource_client commit 2e3c3c2bd7ecdc9c2968a32f91e81136e0cb3835
bgruening
parents:
diff
changeset
|
65 parser.add_argument('-t', '--tanimoto', type=int, help='Tanimoto similarity score') |
b60f2077abd8
planemo upload for repository https://github.com/chembl/chembl_webresource_client commit 2e3c3c2bd7ecdc9c2968a32f91e81136e0cb3835
bgruening
parents:
diff
changeset
|
66 parser.add_argument('-s', '--substructure', action='store_true', help='Substructure search using the SMILES input.') |
b60f2077abd8
planemo upload for repository https://github.com/chembl/chembl_webresource_client commit 2e3c3c2bd7ecdc9c2968a32f91e81136e0cb3835
bgruening
parents:
diff
changeset
|
67 parser.add_argument('-d', '--drugs', action='store_true', help='Filter approved drugs') |
b60f2077abd8
planemo upload for repository https://github.com/chembl/chembl_webresource_client commit 2e3c3c2bd7ecdc9c2968a32f91e81136e0cb3835
bgruening
parents:
diff
changeset
|
68 parser.add_argument('-b', '--biotherapeutic', action='store_true', help='Filter biotherapeutic molecules') |
b60f2077abd8
planemo upload for repository https://github.com/chembl/chembl_webresource_client commit 2e3c3c2bd7ecdc9c2968a32f91e81136e0cb3835
bgruening
parents:
diff
changeset
|
69 parser.add_argument('-n', '--nat-prod', action='store_true', help='Filter natural products') |
b60f2077abd8
planemo upload for repository https://github.com/chembl/chembl_webresource_client commit 2e3c3c2bd7ecdc9c2968a32f91e81136e0cb3835
bgruening
parents:
diff
changeset
|
70 parser.add_argument('-r', '--ro5', action='store_true', help='Filter compounds that pass Lipinski RO5') |
b60f2077abd8
planemo upload for repository https://github.com/chembl/chembl_webresource_client commit 2e3c3c2bd7ecdc9c2968a32f91e81136e0cb3835
bgruening
parents:
diff
changeset
|
71 |
b60f2077abd8
planemo upload for repository https://github.com/chembl/chembl_webresource_client commit 2e3c3c2bd7ecdc9c2968a32f91e81136e0cb3835
bgruening
parents:
diff
changeset
|
72 args = parser.parse_args() |
b60f2077abd8
planemo upload for repository https://github.com/chembl/chembl_webresource_client commit 2e3c3c2bd7ecdc9c2968a32f91e81136e0cb3835
bgruening
parents:
diff
changeset
|
73 |
b60f2077abd8
planemo upload for repository https://github.com/chembl/chembl_webresource_client commit 2e3c3c2bd7ecdc9c2968a32f91e81136e0cb3835
bgruening
parents:
diff
changeset
|
74 if args.file: # get SMILES from file rather than -i option |
b60f2077abd8
planemo upload for repository https://github.com/chembl/chembl_webresource_client commit 2e3c3c2bd7ecdc9c2968a32f91e81136e0cb3835
bgruening
parents:
diff
changeset
|
75 args.input = open_file(args.file) |
b60f2077abd8
planemo upload for repository https://github.com/chembl/chembl_webresource_client commit 2e3c3c2bd7ecdc9c2968a32f91e81136e0cb3835
bgruening
parents:
diff
changeset
|
76 |
b60f2077abd8
planemo upload for repository https://github.com/chembl/chembl_webresource_client commit 2e3c3c2bd7ecdc9c2968a32f91e81136e0cb3835
bgruening
parents:
diff
changeset
|
77 if len(args.input) < 5: |
b60f2077abd8
planemo upload for repository https://github.com/chembl/chembl_webresource_client commit 2e3c3c2bd7ecdc9c2968a32f91e81136e0cb3835
bgruening
parents:
diff
changeset
|
78 raise IOError('SMILES must be at least 5 characters long.') |
b60f2077abd8
planemo upload for repository https://github.com/chembl/chembl_webresource_client commit 2e3c3c2bd7ecdc9c2968a32f91e81136e0cb3835
bgruening
parents:
diff
changeset
|
79 |
b60f2077abd8
planemo upload for repository https://github.com/chembl/chembl_webresource_client commit 2e3c3c2bd7ecdc9c2968a32f91e81136e0cb3835
bgruening
parents:
diff
changeset
|
80 if args.substructure: # specify search type: substructure or similarity |
b60f2077abd8
planemo upload for repository https://github.com/chembl/chembl_webresource_client commit 2e3c3c2bd7ecdc9c2968a32f91e81136e0cb3835
bgruening
parents:
diff
changeset
|
81 mols = substr_search(args.input) |
b60f2077abd8
planemo upload for repository https://github.com/chembl/chembl_webresource_client commit 2e3c3c2bd7ecdc9c2968a32f91e81136e0cb3835
bgruening
parents:
diff
changeset
|
82 else: |
b60f2077abd8
planemo upload for repository https://github.com/chembl/chembl_webresource_client commit 2e3c3c2bd7ecdc9c2968a32f91e81136e0cb3835
bgruening
parents:
diff
changeset
|
83 mols = sim_search(args.input, args.tanimoto) |
b60f2077abd8
planemo upload for repository https://github.com/chembl/chembl_webresource_client commit 2e3c3c2bd7ecdc9c2968a32f91e81136e0cb3835
bgruening
parents:
diff
changeset
|
84 |
b60f2077abd8
planemo upload for repository https://github.com/chembl/chembl_webresource_client commit 2e3c3c2bd7ecdc9c2968a32f91e81136e0cb3835
bgruening
parents:
diff
changeset
|
85 # filter options: |
b60f2077abd8
planemo upload for repository https://github.com/chembl/chembl_webresource_client commit 2e3c3c2bd7ecdc9c2968a32f91e81136e0cb3835
bgruening
parents:
diff
changeset
|
86 if args.drugs: |
b60f2077abd8
planemo upload for repository https://github.com/chembl/chembl_webresource_client commit 2e3c3c2bd7ecdc9c2968a32f91e81136e0cb3835
bgruening
parents:
diff
changeset
|
87 mols = filter_drugs(mols) |
b60f2077abd8
planemo upload for repository https://github.com/chembl/chembl_webresource_client commit 2e3c3c2bd7ecdc9c2968a32f91e81136e0cb3835
bgruening
parents:
diff
changeset
|
88 |
b60f2077abd8
planemo upload for repository https://github.com/chembl/chembl_webresource_client commit 2e3c3c2bd7ecdc9c2968a32f91e81136e0cb3835
bgruening
parents:
diff
changeset
|
89 if args.biotherapeutic: |
b60f2077abd8
planemo upload for repository https://github.com/chembl/chembl_webresource_client commit 2e3c3c2bd7ecdc9c2968a32f91e81136e0cb3835
bgruening
parents:
diff
changeset
|
90 mols = filter_biotherapeutic(mols) |
b60f2077abd8
planemo upload for repository https://github.com/chembl/chembl_webresource_client commit 2e3c3c2bd7ecdc9c2968a32f91e81136e0cb3835
bgruening
parents:
diff
changeset
|
91 |
b60f2077abd8
planemo upload for repository https://github.com/chembl/chembl_webresource_client commit 2e3c3c2bd7ecdc9c2968a32f91e81136e0cb3835
bgruening
parents:
diff
changeset
|
92 if args.nat_prod: |
b60f2077abd8
planemo upload for repository https://github.com/chembl/chembl_webresource_client commit 2e3c3c2bd7ecdc9c2968a32f91e81136e0cb3835
bgruening
parents:
diff
changeset
|
93 mols = filter_nat_prod(mols) |
b60f2077abd8
planemo upload for repository https://github.com/chembl/chembl_webresource_client commit 2e3c3c2bd7ecdc9c2968a32f91e81136e0cb3835
bgruening
parents:
diff
changeset
|
94 |
b60f2077abd8
planemo upload for repository https://github.com/chembl/chembl_webresource_client commit 2e3c3c2bd7ecdc9c2968a32f91e81136e0cb3835
bgruening
parents:
diff
changeset
|
95 if args.ro5: |
b60f2077abd8
planemo upload for repository https://github.com/chembl/chembl_webresource_client commit 2e3c3c2bd7ecdc9c2968a32f91e81136e0cb3835
bgruening
parents:
diff
changeset
|
96 mols = filter_ro5(mols) |
b60f2077abd8
planemo upload for repository https://github.com/chembl/chembl_webresource_client commit 2e3c3c2bd7ecdc9c2968a32f91e81136e0cb3835
bgruening
parents:
diff
changeset
|
97 |
b60f2077abd8
planemo upload for repository https://github.com/chembl/chembl_webresource_client commit 2e3c3c2bd7ecdc9c2968a32f91e81136e0cb3835
bgruening
parents:
diff
changeset
|
98 # get SMILES from search output |
b60f2077abd8
planemo upload for repository https://github.com/chembl/chembl_webresource_client commit 2e3c3c2bd7ecdc9c2968a32f91e81136e0cb3835
bgruening
parents:
diff
changeset
|
99 mols = get_smiles(mols) |
b60f2077abd8
planemo upload for repository https://github.com/chembl/chembl_webresource_client commit 2e3c3c2bd7ecdc9c2968a32f91e81136e0cb3835
bgruening
parents:
diff
changeset
|
100 |
b60f2077abd8
planemo upload for repository https://github.com/chembl/chembl_webresource_client commit 2e3c3c2bd7ecdc9c2968a32f91e81136e0cb3835
bgruening
parents:
diff
changeset
|
101 # write to file |
b60f2077abd8
planemo upload for repository https://github.com/chembl/chembl_webresource_client commit 2e3c3c2bd7ecdc9c2968a32f91e81136e0cb3835
bgruening
parents:
diff
changeset
|
102 with open(args.output, 'w') as f: |
b60f2077abd8
planemo upload for repository https://github.com/chembl/chembl_webresource_client commit 2e3c3c2bd7ecdc9c2968a32f91e81136e0cb3835
bgruening
parents:
diff
changeset
|
103 f.write('\n'.join(mols)) |
b60f2077abd8
planemo upload for repository https://github.com/chembl/chembl_webresource_client commit 2e3c3c2bd7ecdc9c2968a32f91e81136e0cb3835
bgruening
parents:
diff
changeset
|
104 |
b60f2077abd8
planemo upload for repository https://github.com/chembl/chembl_webresource_client commit 2e3c3c2bd7ecdc9c2968a32f91e81136e0cb3835
bgruening
parents:
diff
changeset
|
105 |
b60f2077abd8
planemo upload for repository https://github.com/chembl/chembl_webresource_client commit 2e3c3c2bd7ecdc9c2968a32f91e81136e0cb3835
bgruening
parents:
diff
changeset
|
106 if __name__ == "__main__": |
b60f2077abd8
planemo upload for repository https://github.com/chembl/chembl_webresource_client commit 2e3c3c2bd7ecdc9c2968a32f91e81136e0cb3835
bgruening
parents:
diff
changeset
|
107 main() |