ArhsddlmZmZmZmZmZmZmZddlm Z m Z ddl Z ddl Z ddl Z ddlmZmZddlmZddlmZddlmZddlmZmZmZdd lmZmZdd lmZdd lm Z m!Z!m"Z"m#Z#dd l$m%Z%e jLe'Z(dd l)m*Z*e*Z+e(jYdejZde dZ.ejZdddgedZ/ejZddgedZ0ejZddgedZ1ejZddgedZ2ejZddgedZ3ejZddgedZ4ejZd dged!Z5ejZd"dged#Z6d'd$Z7ejZd%dged&Z8y)()render_templateredirecturl_forflashrequestjsonify current_app)login_required current_userN)datetime timedelta)db) database_bp)api_bp)DatabaseConnectionDatabaseSchema QueryHistory)encrypt_passworddecrypt_password)token_required) SubscriptionUserSubscriptionPlan QueryUsage)AIModelFactory)DatabaseConnectorz+Database connector initialized successfullyz /connectionsctjjtjj }t d|S)z2List all database connections for the current useruser_idzdatabase/connections.html connections)rquery filter_byr idallrr s $/var/www/html/app/database/routes.pyr!r!s8%**44\__4MQQSK 6K PPz/database/connectionsGETOPTIONS)methodsctjdk(ry tjj tj j j}td|Dcgc]}|j |j|j|j|j|j|j|j|j |j"r|j"j%nd|j&r|j&j%ndd c}d}|Scc}w#t($rG}t*j-dt/|td d t/|d d fcYd}~Sd}~wwxYw) zAAPI endpoint to get all database connections for the current userr))rTN) r$name descriptiondb_typehostport database_name service_nameusername created_at updated_at)successr!zError fetching connections: FzFailed to get connections: r8message)rmethodrr"r#r r$r%rr.r/r0r1r2r3r4r5r6 isoformatr7 Exceptionloggererrorstr)r!connresponsees r&get_connectionsrE'sP ~~" )..88AUAUAXAX8Y]]_ ('' II#'#3#3#|| II II%)%7%7$($5$5 $ AE$//";";"=VZAE$//";";"=VZ   *'(  3CF8<=4SVH=    s1AD1&B;D, ! D1,D11 F:| j@tC| d$d| jDr| jDjGnd%| jHr| jHjGnd%d& d'd(fS#tJ$re} t*j,jMtNjQd)tS| t dd*tS| dd+fcYd%} ~ Sd%} ~ wwxYw),z0API endpoint to create a new database connectionr.r0sqlite)r1r2r5passwordr3FMissing required field: r9r)rractive)rstatusrplan_idconnection_limitbasic professional enterprisez*You have reached your connection limit of z3. Please upgrade your plan to add more connections.rIr/r,r1r2r4r5) r.r/r0r1r2r3r4r5rIrThas_empty_passwordN r$r.r/r0r1r2r3r4r5rYr6r7r8 connectionzError creating connection: zFailed to create connection: r;)*rjsongetextendappendrapp.auth.modelsrrr"r#r r$firstrcountrPfeatures isinstancedictrrsessionaddcommitr.r/r0r1r2r3r4r5getattrr6r=r7r>rollbackr?r@rA) datarequired_fieldsfieldrr subscriptionexisting_connections_countrQplan plan_featuresencrypted_passwordr\rDs r&create_connectionruQsd||"9- 88I ( *  " "#K L /$ ED $!9%A   C$))33LOOT\3]cce &8%=%=%G%GP\P_P_%G%`%f%f%h" #))33L?(f3O&"%&!$/."5XXj"-' OO   z"  mm")55%--""!+!9!9 * 7 7&//&-j:NPU&VCMCXCXj33==?^bCMCXCXj33==?^b   ! "   23q6(;<6s1vh?    s3AL3L3-D0L3FL33 N!c tjj|tjj }|st ddddfSt d|j|j|j|j|j|j|j|j|jt|dd|j r|j j#nd|j$r|j$j#ndd d d fS#t&$rG}t(j+d t-|t dd t-|ddfcYd}~Sd}~wwxYw)z2API endpoint to get a specific database connectionr$rFConnection not foundr9TrYNrZr[r-zError fetching connection: zFailed to get connection: r;)rr"r#r r$rcrr.r/r0r1r2r3r4r5rkr6r=r7r>r?r@rA) connection_idr\rDs r&get_connectionr{se'--77=R^RaRa7bhhj  1   mm")55%--""!+!9!9 * 7 7&//&-j:NPU&VCMCXCXj33==?^bCMCXCXj33==?^b   ! "  23q6(;<3CF8<    s%ADC D E/(rlr?r@rA)rzr\rmrDs r&update_connectionr~sZ>'--77=R^RaRa7bhhj  1  || T>"6lJO D %)-%8J "  !%iJ  T>"6lJO T>"6lJO d "'+O'rlr?r@rA)rzr\schemasschemarDs r&delete_connectionr$sC'--77=R^RaRans7tzz|  1  !&&00}Y^0_cce !F     !   8       23q6(;<6s1vh?    s&ACBC EAD?9E?Ez./database/connections//testc&tj} tjj ||j j }|stddddfSt|j}|j||jd|j|jd}|jjdk(r3|j r'|j |d <t#d |j t$j'|j| }|d rtd dddfStd|dddfS#t($rG}t*j-dt/|tddt/|ddfcYd}~Sd}~wwxYw)NrwFrxr9ry:r5rIhostPort databaseNameoracle serviceNamezAdding Oracle service_name: r0connection_configr8TConnection successfulr-r:rKzError testing connection: Failed to test connection: r;)rr rr"r#r$rcrrrIr5r1r2r3r0lowerr4print connectortest_connectionr>r?r@rA)rzr r\rIrresultrDs r&rrGs''L>.'--77=R^RaRa7bhhj  1  $J$7$78#++ %??+1Z__,=>&44      # # % 1j6M6M/9/F/F m , 01H1H0IJ K**&&/+  ) 2   !),   1#a&:;4SVH=    s+A ECE-E F r?r@rA)r rmrnrorrrDs r&test_new_connectionr~s''LF4! -  _$ ED $!9%A  Z(Z(<.$v,8 1    ? "h .>T3IdSaNb/3N/C m ,**O/+  ) 2   !),   5c!fX>?4SVH=    s5%C! C!C!A1C!C!! D1*/schemac tjj|j}|r2t d|d|}t j |j}nt d|dtjj|}|stdd|ddd fSt|j}|j||jd |j|j d }|j"j%d k(r@t'|d r4|j(r(|j(|d<t d|j(dt d|t*j-|j"|}d|vr1t.j1d|dtdd|dddfSt d|dt||j t j2|}t4j6j9|t4j6j;td||j<r|j<j?ndddfS#t@$re}t4j6jCt.j1dtE|tddtE|ddfcYd}~Sd}~wwxYw)z8API endpoint to get the schema for a database connectionrzzJ [DATABASE ROUTES] Getting stored schema from database, for connection ID  zI [DATABASE ROUTES] Fetching stored schema from source, for connection ID FzConnection with ID z not foundr9ryrrrr4r/ [DATABASE ROUTES] Adding Oracle service_name: zConnection config:rr@zError fetching schema: zFailed to fetch schema: r;z9 [DATABASE ROUTES] Creating new schema for connection ID rz schema_name schema_dataTN)r8r last_updatedr-)#rr"r#rcrr^loadsrrr_rrrIr5r1r2r3r0rhasattrr4rget_database_schemar?r@dumpsrrhrirjrr=r>rlrA)rzexisting_schemarrr\rIrrDs r& get_schemarsJ)..88}8U[[]  _`m_nnpq r$F**_%@%@AK ^_l^mmop q,1155mDJ$!4]O:N  ( (;(;| j@d}| jBjEdk(r@tG| dr4| jHr(| jH|d<t d| jHd |tJjLd!<tNjj|"j}t d#| jBd$|d |st d%tPjS| jB|}d&|vrtd'|d&dStO|| j@tUjV|(}tXjZj]|tXjZj_n*t d)tUj`|jb}|}tjted *z }t$jj't$j(|k(t$jf|k(t$jhd+k(t$j*|k\j} d,}|jEd-vr|rr|j.r|j.} t1| t2rP|jEdk(r| jd.d/sd/}ns|jEd0k(r`| jd1d/sNd/}nK|jEdk(r|jd2vrd/}n'|jEd0k(r|jdk7rd/}nd/}|st d3|d4d5}tkjl|}tPjo| jB|}|tJjLd!<tqjp}|js||| jB|jd6i}t d7||jd8d,}d9}d:|vr|d:}tqjp}||z }|d;stut j||d<|d/||xr| |jd&d=> }tXjZj]|tXjZj_td?|jd&d@dA|xr| BStut j|||dC|d,||xr| D}|s|rt%t jtj||d+d/E}tXjZj]|tXjZj]|tXjZj_nVtj|_tXjZj]|tXjZj_d,|dC|d||j|xr| |jdFd/dG} |r|| d:<tw| S#tx$r"}!tdHt{|!dcYd9}!~!Sd9}!~!wwxYw)Iz(Generate SQL query from natural languagerrzprompt model_typemodelopenaiz+[API] Getting explain plan with model_type=zMissing required parametersrrOrNrU query_limitrRdrTr;rVrWzPYou have reached your monthly query limit. Please upgrade your plan to continue.rX)r query_countrwzDatabase connection not foundryrrrr4rrrLAST_CONNECTION_CONFIGrz' [DATABASE ROUTES] Fetching schema for z connection ID z3Schema not found in DB, creating new schema record r@zFailed to get schema: rz Updating existing schema record )hoursgenerateT)rclaude openai_modelsFr claude_models)rTrVz#[API] User does not have access to z model, falling back to deepseekdeepseekconfigzResult of AI model processing: http_successNadditional_infor8r,zLWe could not generate the SQL query for you. We are investigating the issue.) rrznatural_language_query generated_sqlexecution_time is_successfulrcredit_consumed error_messagezFailed to generate SQL query: z Unknown errorrK)r sql_query)rrzrrrrrr)rr query_textrz query_type is_cached from_cache)r8r"rrquery_idrrzError generating SQL query: )>rrrr_rr%r r$rr"r#rcrrPr utcnowyearmonthrfilterrrrdrerfrgrrrIr5r1r2r3r0rrr4r rrrrr^rrrhrirjrrr rrr get_model get_db_configtime generate_sqlrrr>rA)"rmrzrrcurrent_user_idrprrtoday month_startrrrsr\rIrrr new_schemaschema_data_obj one_hour_ago cached_queryhas_model_accessrdb_config_result start_timerrrend_timer query_history query_usagerrDs" r&generate_queryrs    D 011HH_-M XXh F,'F488GX+FJ 7 | DE  v& ';<<**--O %%///HNNPL%%// 8L8L/MSSU!uzz5;;: !&&--   / 1  % % 4  %'   DMM MMM-.=M3Q+M: <<7*"%K\\^3"%K\\\1"$K "  !;!b''  $))33 ,//4 eg =sCC 3 34H''!'q(9:"00 !X-'*n2UZdZqZq+5+B+B-( @AXAX@YY[\]4EK/0 ! ! + +- + H N N PF 4Z5G5G4HXeWffh ij  DE 33J4F4FHYZ k !!$:;w;O:P"QSVW W$'"00 ;/ z"  12jj!3!34 "O ??$yq'99L##**o-'+!!\1   eg ML    !5 5 $ mT2!'')X5m>O>OP_af>g+0(#))+x7 @Q@QRach@i+0("'')X5$,,Nl:l+0(#))+x7DLLLM+ ,}%% L rsAZZZ4 ($PPC)LL%   8 $8    9:>"Q#Q%y/AB&C&P%x8f9fP9E7K!L!F9E7K@L@D9H:NOB>Q3R3j*VH=9>9v@5'RLSL^/ kF8,SL-SLr'