o Ëh2@sdZddlZddlmZmZmZmZddlmZm Z ddl m Z ddl m Z mZmZmZddlmZddlmZdd lmZdd lmZeeZe d d gd Zedeeeefdede deeeffddZedddeeeefdede de dede deeeff ddZ!e"deeeefdedede defddZ#edeeeefdede deeeffd d!Z$dS)"z[ Admin API endpoints for database management. Handles table data retrieval and management. N)ListDictAnyOptional)textinspect)Session) APIRouterDepends HTTPExceptionstatus)get_db)get_current_admin_user)User)MessageResponsez /api/adminadmin)prefixtagsz/tables current_admindbreturnc svzt|j}|}ddgfdd|D}|t|dWSty:}ztd|ttj ddd }~ww) z-Get list of available tables in the database.alembic_version user_sessionscg|]}|vr|qSr.0table system_tablesr./var/www/html/aiguide_backend/routers/admin.py #z(get_available_tables..)tablestotalu❌ Error getting tables: zError retrieving tables status_codedetailN) rbindget_table_nameslen Exceptionloggererrorr r HTTP_500_INTERNAL_SERVER_ERROR)rr inspectorr# user_tableserrr get_available_tabless" r2z/table/{table_name}2 table_namepagelimitc sRzt|j}|}||vrttjd|dd||}dd|D}td|} ||  } | r:| dnd} |d|} td |d |d | } || }g}|D]}i}t |D] \}}||||<q`| |qX|||||| | |d|d d WStyt y}zt d|d|ttjddd}~ww)z-Get data from specific table with pagination.Table ' ' not foundr%cSsg|]}|dqS)namer)rcolrrr r!Gsz"get_table_data..SELECT COUNT(*) as total FROM rr3zSELECT * FROM z LIMIT z OFFSET )r6r7r$pages)r5columnsdata paginationu"❌ Error getting table data for '': zError retrieving table dataN)rr(r)r r HTTP_404_NOT_FOUND get_columnsrexecutefetchone enumerateappendr+r,r-r.)r5r6r7rrr/r#r> column_names count_query total_resultr$offsetqueryresultrowsrowrow_dictivaluer1rrr get_table_data1sT       rSz/table/{table_name}/refreshc sz/t|j}|}||vrttjd|ddtd|d|jt d|dddWSty8t yU}zt d |d |ttj d dd }~ww) z*Refresh table data (reload from database).r8r9r%u 🔄 Table 'z' refreshed by admin z' refreshed successfullyT)messagesuccessu❌ Error refreshing table 'rAzError refreshing table dataN) rr(r)r r rBr,infoemailrr+r-r.)r5rrr/r#r1rrr refresh_table_dataqs0    rXz/statsc szet|j}|}ddgfdd|D}i}|D]=}ztd|}||}|r3|dnd||<WqtyY} ztd|d| d||<WYd } ~ qd } ~ wwt ||t | d WSty} zt d | t tjd d d } ~ ww)zGet database statistics.rrcrrrrrrr r!r"z&get_database_stats..r<rzCould not get count for table z: N) total_tables table_counts total_recordsu"❌ Error getting database stats: z$Error retrieving database statisticsr%)rr(r)rrDrEr+r,warningr*sumvaluesr-r r r.) rrr/r#r0statsrrIrMr1rrr get_database_statss:  r`)%__doc__loggingtypingrrrr sqlalchemyrrsqlalchemy.ormrfastapir r r r databaser dependenciesrmodelsrschemasr getLogger__name__r,routergetstrr2intrSpostrXr`rrrr s|          ? %