o i @srddlmZddlmZGdddeZGdddeZGdd d eZGd d d eZGd d d ZgdZ dS)) NamedTuple) SparkSessionc@s*eZdZUeed<edBed<eed<dS)DatabasenameN description locationUri)__name__ __module__ __qualname__str__annotations__rrY/home/ubuntu/.local/lib/python3.10/site-packages/duckdb/experimental/spark/sql/catalog.pyrs   rc@s>eZdZUeed<edBed<edBed<eed<eed<dS)TablerNdatabaser tableType isTemporaryr r r r r boolrrrrr s    rc@sBeZdZUeed<edBed<eed<eed<eed<eed<dS)ColumnrNrdataTypenullable isPartitionisBucketrrrrrrs   rc@s2eZdZUeed<edBed<eed<eed<dS)FunctionrNr classNamerrrrrrrs   rc@seZdZdeddfddZdeefddZdeefdd Z dd e d e dBdee fd d Z dd e dBdee fddZd e ddfddZdS)CatalogsessionreturnNcCs ||_dSN)_session)selfrrrr__init__%s zCatalog.__init__c>|jjd}dttdtfddfdd|D}|S)Nz,select database_name from duckdb_databases()xrcSst|ddddS)Nr)rrr)rr%rrrtransform_to_database+sz4Catalog.listDatabases..transform_to_databasecg|]}|qSrr.0r%r(rr .z)Catalog.listDatabases..)r!connsqlfetchalllistr r)r"res databasesrr,r listDatabases(zCatalog.listDatabasescr$)NzEselect table_name, database_name, sql, temporary from duckdb_tables()r%rcSs"t|d|d|dd|ddS)Nrrr&)rrrrr)rr'rrrtransform_to_table4s"z.Catalog.listTables..transform_to_tablecr)rrr*r9rrr-7r.z&Catalog.listTables..)r!r/r0r1r2r r)r"r3tablesrr:r listTables1r6zCatalog.listTables tableNamedbNamecsbd|d}|r|d|d7}|jj|}dtttBdtfddfd d |D}|S) NzY select column_name, data_type, is_nullable from duckdb_columns() where table_name = 'z' z and database_name = ''r%rcSs t|dd|d|ddddS)Nrrr7F)rrrrrr)rr'rrrtransform_to_columnBs z0Catalog.listColumns..transform_to_columncr)rrr*r@rrr-Er.z'Catalog.listColumns..)r!r/r0r1r2r rr)r"r=r>queryr3columnsrrAr listColumns:szCatalog.listColumnscCtr NotImplementedErrorr"r>rrr listFunctionsHzCatalog.listFunctionscCrEr rFrHrrrsetCurrentDatabaseKrJzCatalog.setCurrentDatabaser )r r r rr#r2rr5rr<r rrDrrIrKrrrrr$s r)rrrrrN) typingrrrrrrrr__all__rrrrs   +