B 3Rc% @sddZddlZddlZddlmZejs8ejejdZ ndZ e dddd dfd d Z d d Z d!ddZ e dkr`ee dee dee dee ddddddgZedx"eD]Zeede edddqWedx"eD]Zeede edddqWedx$eD]Zeede edddqWed x$eD]Zeede edddq@WdS)"z- utility functions with "chemical know-how" N)RDConfigz atomdb.gdbz::RDDataZ atomic_dataZsysdbaZ masterkeycCsddddg}ddlm} | |||} | } dd|D}d |krN|d |rdd |krd|d x|D]} | |krj|| qjWd |} d | |f}y| |Wntk rt d |dSX| }x|D]}i}x,t t |D]}||}||}|||<qW|d }|||<|r||d }t |||d<t |dd||d<t |dd||d<t |ddd||d<qWdS)a pulls atomic data from a database **Arguments** - atomDict: the dictionary to populate - descriptorsDesired: the descriptors to pull for each atom - dBase: the DB to use - table: the DB table to use - where: the SQL where clause - user: the user name to use with the DB - password: the password to use with the DB - includeElCounts: if nonzero, valence electron count fields are added to the _atomDict_ ZNVALZNVAL_NO_FULL_FZNVAL_NO_FULL_DZ NVAL_NO_FULLr)DbModulecSsg|] }|qS)upper).0srr4lib/python3.7/site-packages/rdkit/utils/chemutils.py 0sz!GetAtomicData..NAMEZCONFIG,zselect %s from atomic_data %szProblems executing command:N) ignoreFullF) ignoreFullD)rr)Z rdkit.DbaserZconnectZcursorappendremovejoinZexecute ExceptionprintZfetchallrangelenConfigToNumElectrons)ZatomDictZdescriptorsDesiredZdBasetablewhereuserZpasswordZincludeElCountsZ extraFieldsrZcncZfieldZtoPullZcommandresZatomZtDictiZdescvalnameconfigrrr GetAtomicDatasF            r!cCsld}t|}||}g}xJ|D]B}t|ddkrR||dt|dfq"||ddfq"W|S)a Takes a simple chemical composition and turns into a list of element,# pairs. i.e. 'Fe3Al' -> [('Fe',3),('Al',1)] **Arguments** - compStr: the composition string to be processed **Returns** - the *composVect* corresponding to _compStr_ **Note** -this isn't smart enough by half to deal with anything even remotely subtle, so be gentle. z([A-Z][a-z]?)([0-9\.]*)r r)recompilefindallrrfloat)ZcompStrtargetZtheExprZmatchesrmatchrrr SplitCompositionQs   r(cCs|d}d}xtdt|D]}||d}t|d}|rj|dkrj|dddkrjt|dkrjd}|r|d kr|dd dkrt|dkrd}||}qW|S) aO counts the number of electrons appearing in a configuration string **Arguments** - config: the configuration string (e.g. '2s^2 2p^4') - ignoreFullD: toggles not counting full d shells - ignoreFullF: toggles not counting full f shells **Returns** the number of valence electrons  rr ^f d)splitrrintfind)r rrZarrZnElrlZincrrrr rss  ** r__main__ZFeZFe3AlZFe99PdAlZ TiNiSiSO12Pz[Xe] 4f^12 6s^2z[Xe] 4f^14 5d^6 6s^2z[Xe] 4f^14 5d^10 6s^2z[Xe] 4f^14 5d^10 6s^2 6p^1z [Xe] 5d^10z ignore allz r )rrzignore dzignore fz ignore None)rr)__doc__osr"ZrdkitrZusePgSQLpathrZ RDDataDirZ _atomDbNamer!r(r__name__rZtempentryrrrr s: <"