B 3Rc~@slddlZddlmZddlmZejZejZejZejZddd Z dd d Z d d Z dddZ dddZ dS)N)RDConfig)DbModulesysdba masterkey. ::template1c Cstjr|s>yt|||}Wn tk r<td|gSX|}|tjtjrldd| D}ndd| D}| |n2tj rddl }ddl }| |j|tj }ng}|S)z returns a list of databases that are available **Arguments** - user: the username for DB access - password: the password to be used for DB access **Returns** - a list of db names (strings) zProblems opening database: %scSsg|]}dt|dqS)z::r)str).0xr 1lib/python3.7/site-packages/rdkit/Dbase/DbInfo.py /szGetDbNames..cSsg|]}dt|dqS)z::r)r)r r r r r r 1srN)rZgetDbSqlconnect ExceptionprintcursorexecuterusePgSQLfetchallremoveZ fileWildcardos.pathglobpathjoin) userpasswordZdirNamedBasecncnamesosrr r r GetDbNamess&   r!cCs|s8yt|||}Wn tk r6td|gSX|}|sLtj}ntj}||dd|D}t j rd|kr| d|S)a returns a list of tables available in a database **Arguments** - dBase: the name of the DB file to be used - user: the username for DB access - password: the password to be used for DB access - includeViews: if this is non-null, the views in the db will also be returned **Returns** - a list of table names (strings) zProblems opening database: %scSsg|]}t|dqS)r)rupper)r r r r r r \sz!GetTableNames..Z PG_LOGDIR_LS) rrrrrZ getTablesSqlZgetTablesAndViewsSqlrrrrr)rrrZ includeViewsrrZcommrr r r GetTableNames<s   r#c CsD|dks|jdkrgSg}tjsx||jD]r}|d}|d}|tkrJd}nB|tkrXd}n4|tkrfd}n&|tkrtd}ntj d|t |f| ||fq(Wn| }|s|Sxt |D]\}}|j|d}t|} t|t rd}nF| tkrd}n8| tkrd}n(| ttfkrd}ntj d|| f| ||fqW|S)NrstringZintegerfloatZbinaryzodd type in col %s: %s ) descriptionrZ useSqlLite sqlTextTypes sqlIntTypes sqlFloatTypes sqlBinTypessysstderrwriterappendfetchone enumeratetype isinstanceintr& memoryviewbytes) rZresultsitemZcNameZcTypeZtypeStrrivtypr r r GetColumnInfoFromCursorbsD   r<*c CsH|st|||}|}d||f}|r6|d|7}||t|S)a! gets a list of columns available in a DB table along with their types **Arguments** - dBase: the name of the DB file to be used - table: the name of the table to query - user: the username for DB access - password: the password to be used for DB access - join: an optional join clause (omit the verb 'join') - what: an optional clause indicating what to select **Returns** - a list of 2-tuples containing: 1) column name 2) column type zselect %s from %sz join %s)rrrrr<) rtablerrrwhatrrcmdr r r GetColumnNamesAndTypess   rBc Csz|st|||}|}d||f}|rP|ddkrDd|}|d|7}||||j} dd| D} | S)a gets a list of columns available in a DB table **Arguments** - dBase: the name of the DB file to be used - table: the name of the table to query - user: the username for DB access - password: the password to be used for DB access - join: an optional join clause (omit the verb 'join') - what: an optional clause indicating what to select **Returns** - a list of column names zselect %s from %srrzjoin %s cSsg|]}t|dqS)r)r)r r r r r r sz"GetColumnNames..)rrrstripfindrr0r') rr?rrrr@rrrAZdescZresr r r GetColumnNamess   rF)rrrrN)rrrN)rrr=r>N)rrr=r>N)r,ZrdkitrZ rdkit.Dbaserr(r)r*r+r!r#r<rBrFr r r r  s   % &( $