B ž3Rc+ã @s(ddlZddlZddlmZddlmZddlmZedddZejdd d d d d ejddddddde  ¡\Z Z e j dkr’e dƒe d¡e jdks¦e jdkr¸e dƒe d¡gZgZee j dƒZx€eD]xZe ¡ d¡Ze ed¡Zedkrej ded¡qÒejeddddZe e¡e edededf¡qÒWeeƒZ xÌej!D]ÂZe ¡Ze de¡\Z"Z#e e"¡Z$e$dkr¢ej de"¡q^eje$ddddZ%e &e%eddd ¡Z'xVe(e ƒD]JZ)e'e)e jkrÐe d!ee)dee)dee)de"e#e'e)fƒqÐWq^WdS)"éN)Ú OptionParser)ÚChem)Ú DataStructsz4Program to Tversky search results as part of Fragglez–Format of input file: whole mol smiles,ID,fraggle split smiles Output: query_frag_smiles,query_smiles,query_id,retrieved_smi,retrieved_id,tversky_sim)Z descriptionZepilogz-fz--fragsÚf_fileÚstringz5File containing the query fragmentations from Fraggle)ÚdestÚtypeÚhelpz-cz--cutoffÚcutoffÚfloatgš™™™™™é?zxCutoff for Tversy similarity. Only Tversky results with similarity greater than the cutoff will be output. DEFAULT = 0.8)rrÚdefaultr z=Please specify the file containing the Fraggle fragmentationséz,Please specify a Tversky cut-off between 0-1Úrú,ézCan't generate mol for: %s éi)ZmaxPathZfpSizeZ nBitsPerHashz\s|,Fz%s,%s,%s,%s,%s,%s)*ÚsysÚreZoptparserZrdkitrrÚparserZ add_optionÚ parse_argsZoptionsÚargsrÚprintÚexitr ZqueriesZ query_infoÚopenZ q_split_inputÚlineÚrstripÚsplitÚinfoZ MolFromSmilesZqmolÚstderrÚwriteZRDKFingerprintZqfpÚappendÚlenZ fragmentsÚstdinZsmiÚidZmolZmfpZBulkTverskySimilarityZresÚrangeÚi©r&r&ú,share/RDKit/Contrib/fraggle/rdkit_tversky.pyÚ!sZ