B 3Rc @sddlZddlZddlZddlmZddlmZddlmZeZe dZ ddZ dd Z d Z d d Zd dZddZedkredS)N)Chem)RDLogger)ChemicalFeaturesz[ \t,]cCsvdg|}||}xX|D]P}|}d||f}x.|D]&}||dkr\g||<|||qDWqW|S)Nz%s-%s)Z GetNumAtomsGetFeaturesForMol GetAtomIds GetFamilyGetTypeappend)factorymolZresfeatsfeatZidsZfeatureZid_r7lib/python3.7/site-packages/rdkit/Chem/FeatFinderCLI.pyGetAtomFeatInfos    rcCs`tjdttjd}|jdddddd|jd d d tjtd |jd tdd|jdtdd|S)z Initialize the parser z-Determine pharmacophore features of molecules)Z descriptionZepilogZformatter_classz-r reverseItF store_truez(Set to get atoms lists for each feature.)destdefaultactionhelpz-nmaxLines)rrrtype fdefFilenamez%Pharmacophore feature definition file)rrsmilesFilenamez6The smiles file should have SMILES in the first column)argparseArgumentParser_splashMessageZRawDescriptionHelpFormatter add_argumentZSUPPRESSint existingFile)parserrrr initParser%s  r#z -*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-* FeatFinderCLI Part of the RDKit (http://www.rdkit.org) -*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-* cCs tj|std||S)z2 'type' for argparse - check that filename exists z{0} does not exist)ospathexistsrZArgumentTypeErrorformat)filenamerrrr!<s r!c Csyt|j}Wn$tk r4|d|YnXt|j:}x0t|dD] \}}||j dkrlPt | d }t |}|dkrtd||fqRtd||f|jr||}x|D]<} td| | fddtd d d | DqWqRt||} xVt| D]J\} } td || | dfdd| rhtd d | ntq&WqRWWdQRXdS)Nz+Could not parse Fdef file {0.fdefFilename}.rz)Could not process smiles '%s' on line %d.z Mol-%d %sz %s-%s: )endz, cSsg|] }t|qSr)str).0xrrr XszprocessArgs..z % 2s(%d) )rZBuildFeatureFactoryr Exceptionerrorr'openr enumerater splitExprsplitstriprZ MolFromSmilesloggerZwarningprintrrrrjoinrrZGetAtomWithIdxZ GetSymbol) argsr"r ZinFZlineNolineZsmir r r ZfeatInfoivrrr processArgsCs2   " "r?cCst}|}t||dS)z Main application N)r# parse_argsr?)r"r;rrrmaincsrA__main__)rrer$ZrdkitrrZ rdkit.Chemrr8compiler5rr#rr!r?rA__name__rrrr s