Mercurial > repos > bgruening > openbabel_genprop
view remove_protonation_state.py @ 6:ebe0654a8b85 draft
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/openbabel commit 3ba9268697beb4827bd9e8e639e70166f224cb8b
author | bgruening |
---|---|
date | Mon, 03 Sep 2018 16:40:42 -0400 |
parents | df6d948e95ba |
children | a72ae2711a97 |
line wrap: on
line source
#!/usr/bin/env python """ Input: molecular input file. Output: Molecule file with removed ions and fragments. Copyright 2013, Bjoern Gruening and Xavier Lucas """ import sys, os import argparse import openbabel openbabel.obErrorLog.StopLogging() import pybel def parse_command_line(): parser = argparse.ArgumentParser() parser.add_argument('--iformat', default='sdf' , help='input file format') parser.add_argument('-i', '--input', required=True, help='input file name') parser.add_argument('-o', '--output', required=True, help='output file name') return parser.parse_args() def remove_protonation( args ): outfile = pybel.Outputfile(args.iformat, args.output, overwrite=True) for mol in pybel.readfile(args.iformat, args.input): [atom.OBAtom.SetFormalCharge(0) for atom in mol.atoms] outfile.write( mol ) outfile.close() def __main__(): """ Remove any protonation state from each atom in each molecule. """ args = parse_command_line() remove_protonation( args ) if __name__ == "__main__" : __main__()