Exemple #1
0
 def execute(self, args):
     """Run import method for every molecule in source."""
     m = load_method('import', MessDB())
     s = Source(MessDB())
     s.setup(args.source)
     p = Path(MessDB())
     p.setup(m.method_id)
     pybel.ob.obErrorLog.StopLogging()
     for f in s.files():
         if (f.split('.')[-1] == 'sql' or 
             f.split('.')[-1] == 'txt' or 
             f.split('.')[-1] == 'bak' or 
             f[-1] == '~'):
             continue
         for mol in pybel.readfile(f.split('.')[-1],
                                   os.path.join(s.source_dir, f)):
             decorate(mol, UnicodeDecorator)
             pybel.ob.obErrorLog.ClearLog()
             pybel.ob.obErrorLog.StartLogging()
             inchikey = mol.write('inchikey').rstrip()
             pybel.ob.obErrorLog.StopLogging()
             cansmi = mol.write('can').split()[0]
             frag_count = cansmi.count('.') + 1
             for f in cansmi.split('.'):
                 method_args = {}
                 if (frag_count > 1):
                     frag = pybel.readstring('can', f)
                     decorate(frag, UnicodeDecorator)
                     # neutralize fragments
                     #if (frag.charge != 0):
                     #    for atom in frag.atoms:
                     #        atom.OBAtom.SetFormalCharge(0)
                     pybel.ob.obErrorLog.ClearLog()
                     pybel.ob.obErrorLog.StartLogging()
                     frag_inchikey = frag.write('inchikey').rstrip()
                     pybel.ob.obErrorLog.StopLogging()
                     if not is_inchikey(frag_inchikey):
                         print("'%s' is not an importable molecule.\n" % f, 
                               file=sys.stderr)
                         continue
                     method_args['parent'] = ('from: %s' % 
                                              unicode(mol.title,
                                                      'utf-8', 'replace'))
                     method_args['inchikey'] = frag_inchikey
                     method_args['mol'] = frag
                 else:
                     if not is_inchikey(inchikey):
                         print("'%s' is not an importable molecule.\n" % f, 
                               file=sys.stderr)
                         continue
                     method_args['inchikey'] = inchikey
                     method_args['mol'] = mol
                 method_args['source'] = s
                 method_args['path'] = p
                 method_args['skip_cir'] = args.skip_cir
                 status = m.execute(method_args)
                 print('%s: %s' % (method_args['inchikey'], status))
Exemple #2
0
 def check_dir_structure(self):
     """Check that the structure of the molecules dir is consistent."""
     moldir = os.path.join(os.path.dirname(__file__), '../molecules')
     for l in os.listdir(moldir):
         lp = os.path.join(moldir, l)
         if not os.path.isdir(lp):
             print('Unexpected file in molecules dir: %s' % l)
             continue
         if not len(l) == 1:
             print('Unexpected dir in molecules dir: %s' % l)
             continue
         for ll in os.listdir(lp):
             llp = os.path.join(moldir, l, ll)
             if not os.path.isdir(llp):
                 print('Unexpected file in molecules dir: %s/%s' % (l, ll))
                 continue
             if not (len(ll) == 2 and ll.isalpha()):
                 print('Unexpected dir in molecules dir: %s/%s' % (l, ll))
                 continue
             for lll in os.listdir(llp):
                 lllp = os.path.join(moldir, l, ll, lll)
                 if not os.path.isdir(lllp):
                     print('Unexpected file in molecules dir: %s/%s/%s' %
                           (l, ll, lll))
                     continue
                 if not is_inchikey(l + ll + lll, enforce_standard=True):
                     print('Unexpected dir in molecules dir: %s/%s/%s' %
                           (l, ll, lll))
                     continue
                 self.check_molecule_dir(l + ll + lll, lllp)
Exemple #3
0
 def execute(self, args):
     """Run calculations."""
     m = load_method(args.method, MessDB())
     p = Path(MessDB())
     p.setup(m.method_id, args.parent_path)
     pybel.ob.obErrorLog.StopLogging()
     method_args = {}
     method_args['path'] = p
     for row in args.inchikeys:
         method_args['inchikey'] = row.split()[0].strip()
         if not is_inchikey(method_args['inchikey'], enforce_standard=True):
             sys.exit('%s is not a valid InChIKey.' % 
                      method_args['inchikey'])
         status = m.execute(method_args)
         print('%s: %s' % (method_args['inchikey'], status))
Exemple #4
0
 def execute(self, args):
     """Run self checks."""
     db = MessDB()
     self.c = db.cursor()
     self.c.execute('SELECT inchikey FROM molecule')
     self.db_inchikeys = set()
     # check that inchikeys are all valid
     for r in self.c:
         if is_inchikey(r.inchikey, enforce_standard=True):
             self.db_inchikeys.add(r.inchikey)
         else:
             print('%s is not a valid standard InChiKey!' % r.inchikey)
     self.check_dir_structure()
     self.check_db_structure()
     self.check_db_dir_inchikey_concordance()
     self.summary()