File: test_rdkit_draw.py

package info (click to toggle)
rdkit 201809.1%2Bdfsg-6
  • links: PTS, VCS
  • area: main
  • in suites: buster
  • size: 123,688 kB
  • sloc: cpp: 230,509; python: 70,501; java: 6,329; ansic: 5,427; sql: 1,899; yacc: 1,739; lex: 1,243; makefile: 445; xml: 229; fortran: 183; sh: 123; cs: 93
file content (22 lines) | stat: -rwxr-xr-x 554 bytes parent folder | download
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
#!/usr/bin/env python
from __future__ import print_function
from rdkit import RDConfig
import os, sys
from rdkit import Chem
from rdkit.Chem import Draw, AllChem

suppl = Chem.SDMolSupplier()

if sys.argv[1].endswith('.sdf'):
  suppl = Chem.SDMolSupplier(sys.argv[1])
elif sys.argv[1].endswith('.smi'):
  suppl = Chem.SmilesMolSupplier(sys.argv[1])
else:
  print('Need a file ending in .sdf or .smi')
  exit(1)

for mol in suppl:
  print(mol.GetProp('_Name'))
  fn = mol.GetProp('_Name') + '.png'
  AllChem.Compute2DCoords(mol)
  Draw.MolToFile(mol, fn)