Mercurial > repos > bgruening > rdock_rbcavity
diff rbdock.py @ 2:d52d6ee95cf2 draft
"planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/chemicaltoolbox/rdock commit db567a29443284f2cec1444ec9db9aa6bd913fad"
author | bgruening |
---|---|
date | Fri, 03 Apr 2020 13:32:38 -0400 |
parents | |
children |
line wrap: on
line diff
--- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/rbdock.py Fri Apr 03 13:32:38 2020 -0400 @@ -0,0 +1,28 @@ +import subprocess +import argparse + +def main(): + parser = argparse.ArgumentParser(description='Simple wrapper for rbdock') + parser.add_argument('-n', '--num', type=int, help='Number of docking poses to generate') + parser.add_argument('-s', '--seed', type=int, help='Random seed') + args = parser.parse_args() + + cmd = ['rbdock', '-i', 'ligands.sdf', '-r', 'receptor.prm', '-p', 'dock.prm', '-n', str(args.num), '-o', 'rdock_output'] + if args.seed != None: + cmd += ['-s', str(args.seed)] + + ps = subprocess.Popen(cmd, stdout=subprocess.PIPE) + + error_counter = 0 + for stdout_line in iter(ps.stdout.readline, ''): + if 'RBT_DOCKING_ERROR' in str(stdout_line): + error_counter += 1 + if error_counter == 10: + print(ps.stdout) + exit(23) + if ps.poll() != None: + print(ps.stdout) + exit(int(ps.poll())) + +if __name__ == "__main__": + main()