Mercurial > repos > bgruening > ctb_rdkit_descriptors

annotate rdkit_descriptors.py @ 4:fcc88ab4f1d3 draft

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/rdkit commit 20df7e562341cd30e89a14d6bde9054956fadc06"
author bgruening
date Tue, 10 Mar 2020 16:56:37 +0000
parents 81233a9053f5
children 2d051db1f561
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
0
054d7b0de5c1 planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/rdkit commit ed9b6859de648aa5f7cde483732f5df20aaff90e
bgruening
parents:
diff changeset
1 #!/usr/bin/env python
054d7b0de5c1 planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/rdkit commit ed9b6859de648aa5f7cde483732f5df20aaff90e
bgruening
parents:
diff changeset
2
054d7b0de5c1 planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/rdkit commit ed9b6859de648aa5f7cde483732f5df20aaff90e
bgruening
parents:
diff changeset
3 from rdkit.Chem import Descriptors
054d7b0de5c1 planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/rdkit commit ed9b6859de648aa5f7cde483732f5df20aaff90e
bgruening
parents:
diff changeset
4 from rdkit import Chem
054d7b0de5c1 planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/rdkit commit ed9b6859de648aa5f7cde483732f5df20aaff90e
bgruening
parents:
diff changeset
5 import sys, os, re
054d7b0de5c1 planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/rdkit commit ed9b6859de648aa5f7cde483732f5df20aaff90e
bgruening
parents:
diff changeset
6 import argparse
054d7b0de5c1 planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/rdkit commit ed9b6859de648aa5f7cde483732f5df20aaff90e
bgruening
parents:
diff changeset
7 import inspect
054d7b0de5c1 planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/rdkit commit ed9b6859de648aa5f7cde483732f5df20aaff90e
bgruening
parents:
diff changeset
8
054d7b0de5c1 planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/rdkit commit ed9b6859de648aa5f7cde483732f5df20aaff90e
bgruening
parents:
diff changeset
9 def get_supplier( infile, format = 'smiles' ):
054d7b0de5c1 planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/rdkit commit ed9b6859de648aa5f7cde483732f5df20aaff90e
bgruening
parents:
diff changeset
10 """
054d7b0de5c1 planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/rdkit commit ed9b6859de648aa5f7cde483732f5df20aaff90e
bgruening
parents:
diff changeset
11 Returns a generator over a SMILES or InChI file. Every element is of RDKit
054d7b0de5c1 planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/rdkit commit ed9b6859de648aa5f7cde483732f5df20aaff90e
bgruening
parents:
diff changeset
12 molecule and has its original string as _Name property.
054d7b0de5c1 planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/rdkit commit ed9b6859de648aa5f7cde483732f5df20aaff90e
bgruening
parents:
diff changeset
13 """
054d7b0de5c1 planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/rdkit commit ed9b6859de648aa5f7cde483732f5df20aaff90e
bgruening
parents:
diff changeset
14 with open(infile) as handle:
054d7b0de5c1 planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/rdkit commit ed9b6859de648aa5f7cde483732f5df20aaff90e
bgruening
parents:
diff changeset
15 for line in handle:
054d7b0de5c1 planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/rdkit commit ed9b6859de648aa5f7cde483732f5df20aaff90e
bgruening
parents:
diff changeset
16 line = line.strip()
054d7b0de5c1 planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/rdkit commit ed9b6859de648aa5f7cde483732f5df20aaff90e
bgruening
parents:
diff changeset
17 if format == 'smiles':
054d7b0de5c1 planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/rdkit commit ed9b6859de648aa5f7cde483732f5df20aaff90e
bgruening
parents:
diff changeset
18 mol = Chem.MolFromSmiles( line, sanitize=True )
054d7b0de5c1 planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/rdkit commit ed9b6859de648aa5f7cde483732f5df20aaff90e
bgruening
parents:
diff changeset
19 elif format == 'inchi':
054d7b0de5c1 planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/rdkit commit ed9b6859de648aa5f7cde483732f5df20aaff90e
bgruening
parents:
diff changeset
20 mol = Chem.inchi.MolFromInchi( line, sanitize=True, removeHs=True, logLevel=None, treatWarningAsError=False )
054d7b0de5c1 planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/rdkit commit ed9b6859de648aa5f7cde483732f5df20aaff90e
bgruening
parents:
diff changeset
21 if mol is None:
054d7b0de5c1 planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/rdkit commit ed9b6859de648aa5f7cde483732f5df20aaff90e
bgruening
parents:
diff changeset
22 yield False
054d7b0de5c1 planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/rdkit commit ed9b6859de648aa5f7cde483732f5df20aaff90e
bgruening
parents:
diff changeset
23 else:
054d7b0de5c1 planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/rdkit commit ed9b6859de648aa5f7cde483732f5df20aaff90e
bgruening
parents:
diff changeset
24 mol.SetProp( '_Name', line.split('\t')[0] )
054d7b0de5c1 planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/rdkit commit ed9b6859de648aa5f7cde483732f5df20aaff90e
bgruening
parents:
diff changeset
25 yield mol
054d7b0de5c1 planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/rdkit commit ed9b6859de648aa5f7cde483732f5df20aaff90e
bgruening
parents:
diff changeset
26
054d7b0de5c1 planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/rdkit commit ed9b6859de648aa5f7cde483732f5df20aaff90e
bgruening
parents:
diff changeset
27 def get_rdkit_descriptor_functions():
054d7b0de5c1 planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/rdkit commit ed9b6859de648aa5f7cde483732f5df20aaff90e
bgruening
parents:
diff changeset
28 """
054d7b0de5c1 planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/rdkit commit ed9b6859de648aa5f7cde483732f5df20aaff90e
bgruening
parents:
diff changeset
29 Returns all descriptor functions under the Chem.Descriptors Module as tuple of (name, function)
054d7b0de5c1 planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/rdkit commit ed9b6859de648aa5f7cde483732f5df20aaff90e
bgruening
parents:
diff changeset
30 """
054d7b0de5c1 planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/rdkit commit ed9b6859de648aa5f7cde483732f5df20aaff90e
bgruening
parents:
diff changeset
31 ret = [ (name, f) for name, f in inspect.getmembers( Descriptors ) if inspect.isfunction( f ) and not name.startswith( '_' ) ]
054d7b0de5c1 planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/rdkit commit ed9b6859de648aa5f7cde483732f5df20aaff90e
bgruening
parents:
diff changeset
32 ret.sort()
054d7b0de5c1 planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/rdkit commit ed9b6859de648aa5f7cde483732f5df20aaff90e
bgruening
parents:
diff changeset
33 return ret
054d7b0de5c1 planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/rdkit commit ed9b6859de648aa5f7cde483732f5df20aaff90e
bgruening
parents:
diff changeset
34
054d7b0de5c1 planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/rdkit commit ed9b6859de648aa5f7cde483732f5df20aaff90e
bgruening
parents:
diff changeset
35
054d7b0de5c1 planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/rdkit commit ed9b6859de648aa5f7cde483732f5df20aaff90e
bgruening
parents:
diff changeset
36 def descriptors( mol, functions ):
054d7b0de5c1 planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/rdkit commit ed9b6859de648aa5f7cde483732f5df20aaff90e
bgruening
parents:
diff changeset
37 """
054d7b0de5c1 planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/rdkit commit ed9b6859de648aa5f7cde483732f5df20aaff90e
bgruening
parents:
diff changeset
38 Calculates the descriptors of a given molecule.
054d7b0de5c1 planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/rdkit commit ed9b6859de648aa5f7cde483732f5df20aaff90e
bgruening
parents:
diff changeset
39 """
054d7b0de5c1 planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/rdkit commit ed9b6859de648aa5f7cde483732f5df20aaff90e
bgruening
parents:
diff changeset
40 for name, function in functions:
054d7b0de5c1 planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/rdkit commit ed9b6859de648aa5f7cde483732f5df20aaff90e
bgruening
parents:
diff changeset
41 yield (name, function( mol ))
054d7b0de5c1 planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/rdkit commit ed9b6859de648aa5f7cde483732f5df20aaff90e
bgruening
parents:
diff changeset
42
054d7b0de5c1 planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/rdkit commit ed9b6859de648aa5f7cde483732f5df20aaff90e
bgruening
parents:
diff changeset
43
054d7b0de5c1 planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/rdkit commit ed9b6859de648aa5f7cde483732f5df20aaff90e
bgruening
parents:
diff changeset
44 if __name__ == "__main__":
054d7b0de5c1 planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/rdkit commit ed9b6859de648aa5f7cde483732f5df20aaff90e
bgruening
parents:
diff changeset
45 parser = argparse.ArgumentParser()
054d7b0de5c1 planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/rdkit commit ed9b6859de648aa5f7cde483732f5df20aaff90e
bgruening
parents:
diff changeset
46 parser.add_argument('-i', '--infile', required=True, help='Path to the input file.')
054d7b0de5c1 planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/rdkit commit ed9b6859de648aa5f7cde483732f5df20aaff90e
bgruening
parents:
diff changeset
47 parser.add_argument("--iformat", help="Specify the input file format.")
054d7b0de5c1 planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/rdkit commit ed9b6859de648aa5f7cde483732f5df20aaff90e
bgruening
parents:
diff changeset
48
054d7b0de5c1 planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/rdkit commit ed9b6859de648aa5f7cde483732f5df20aaff90e
bgruening
parents:
diff changeset
49 parser.add_argument('-o', '--outfile', type=argparse.FileType('w+'),
054d7b0de5c1 planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/rdkit commit ed9b6859de648aa5f7cde483732f5df20aaff90e
bgruening
parents:
diff changeset
50 default=sys.stdout, help="path to the result file, default it sdtout")
054d7b0de5c1 planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/rdkit commit ed9b6859de648aa5f7cde483732f5df20aaff90e
bgruening
parents:
diff changeset
51
054d7b0de5c1 planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/rdkit commit ed9b6859de648aa5f7cde483732f5df20aaff90e
bgruening
parents:
diff changeset
52 parser.add_argument("--header", dest="header", action="store_true",
054d7b0de5c1 planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/rdkit commit ed9b6859de648aa5f7cde483732f5df20aaff90e
bgruening
parents:
diff changeset
53 default=False,
054d7b0de5c1 planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/rdkit commit ed9b6859de648aa5f7cde483732f5df20aaff90e
bgruening
parents:
diff changeset
54 help="Write header line.")
054d7b0de5c1 planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/rdkit commit ed9b6859de648aa5f7cde483732f5df20aaff90e
bgruening
parents:
diff changeset
55
054d7b0de5c1 planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/rdkit commit ed9b6859de648aa5f7cde483732f5df20aaff90e
bgruening
parents:
diff changeset
56 args = parser.parse_args()
054d7b0de5c1 planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/rdkit commit ed9b6859de648aa5f7cde483732f5df20aaff90e
bgruening
parents:
diff changeset
57
054d7b0de5c1 planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/rdkit commit ed9b6859de648aa5f7cde483732f5df20aaff90e
bgruening
parents:
diff changeset
58 if args.iformat == 'sdf':
054d7b0de5c1 planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/rdkit commit ed9b6859de648aa5f7cde483732f5df20aaff90e
bgruening
parents:
diff changeset
59 supplier = Chem.SDMolSupplier( args.infile )
054d7b0de5c1 planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/rdkit commit ed9b6859de648aa5f7cde483732f5df20aaff90e
bgruening
parents:
diff changeset
60 elif args.iformat =='smi':
054d7b0de5c1 planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/rdkit commit ed9b6859de648aa5f7cde483732f5df20aaff90e
bgruening
parents:
diff changeset
61 supplier = get_supplier( args.infile, format = 'smiles' )
054d7b0de5c1 planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/rdkit commit ed9b6859de648aa5f7cde483732f5df20aaff90e
bgruening
parents:
diff changeset
62 elif args.iformat == 'inchi':
054d7b0de5c1 planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/rdkit commit ed9b6859de648aa5f7cde483732f5df20aaff90e
bgruening
parents:
diff changeset
63 supplier = get_supplier( args.infile, format = 'inchi' )
054d7b0de5c1 planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/rdkit commit ed9b6859de648aa5f7cde483732f5df20aaff90e
bgruening
parents:
diff changeset
64
054d7b0de5c1 planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/rdkit commit ed9b6859de648aa5f7cde483732f5df20aaff90e
bgruening
parents:
diff changeset
65 functions = get_rdkit_descriptor_functions()
054d7b0de5c1 planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/rdkit commit ed9b6859de648aa5f7cde483732f5df20aaff90e
bgruening
parents:
diff changeset
66
054d7b0de5c1 planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/rdkit commit ed9b6859de648aa5f7cde483732f5df20aaff90e
bgruening
parents:
diff changeset
67 if args.header:
2
81233a9053f5 planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/rdkit commit 3372cca3d07562b643b8152d489dcbd2325acf4a
bgruening
parents: 0
diff changeset
68 args.outfile.write( '%s\n' % '\t'.join( ['MoleculeID'] + [name for name, f in functions] ) )
0
054d7b0de5c1 planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/rdkit commit ed9b6859de648aa5f7cde483732f5df20aaff90e
bgruening
parents:
diff changeset
69
054d7b0de5c1 planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/rdkit commit ed9b6859de648aa5f7cde483732f5df20aaff90e
bgruening
parents:
diff changeset
70 for mol in supplier:
054d7b0de5c1 planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/rdkit commit ed9b6859de648aa5f7cde483732f5df20aaff90e
bgruening
parents:
diff changeset
71 if not mol:
054d7b0de5c1 planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/rdkit commit ed9b6859de648aa5f7cde483732f5df20aaff90e
bgruening
parents:
diff changeset
72 continue
054d7b0de5c1 planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/rdkit commit ed9b6859de648aa5f7cde483732f5df20aaff90e
bgruening
parents:
diff changeset
73 descs = descriptors( mol, functions )
054d7b0de5c1 planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/rdkit commit ed9b6859de648aa5f7cde483732f5df20aaff90e
bgruening
parents:
diff changeset
74 molecule_id = mol.GetProp("_Name")
2
81233a9053f5 planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/rdkit commit 3372cca3d07562b643b8152d489dcbd2325acf4a
bgruening
parents: 0
diff changeset
75 args.outfile.write( "%s\n" % '\t'.join( [molecule_id]+ [str(round(res, 6)) for name, res in descs] ) )
0
054d7b0de5c1 planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/rdkit commit ed9b6859de648aa5f7cde483732f5df20aaff90e
bgruening
parents:
diff changeset
76