Mercurial > repos > bgruening > chembl

annotate chembl.py @ 6:a57de37f12c2 draft

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