50 lines
6.3 KiB
Plaintext
50 lines
6.3 KiB
Plaintext
|
§
|
|||
|
|
CkØiÉ ã óš — d Z ddlmZ ddlmZ ddlmZ ddlmZ ddl m
|
|||
|
|
Z
|
|||
|
|
m Z ddl m
|
|||
|
|
Z
|
|||
|
|
mZ ddlmZ dd lmZ d#d„Zd$d„Zd%d„Zddd œd&d"„ZdS )'u
|
|||
|
|
|
|||
|
|
与 DBHub execute-sql.ts 对�的 SQL 执行工具。
|
|||
|
|
|
|||
|
|
与 HTTP sql_execute（run_readonly_sql）ä¸�å�Œï¼šæ— nl_schema_table_allowlistã€�
|
|||
|
|
æ— rewrite_mssql_logical_database_prefixã€�æ— sql_account_rlsï¼›å�ªè¯»è§„则以 DBHub
|
|||
|
|
allowed-keywords + sql-parser 为准。
|
|||
|
|
é )Ú annotations)Ú get_optional)Úlog)Ú
|
|||
|
|
get_engine)Úallowed_keywords_listÚis_read_only_sql)Ú
|
|||
|
|
ConnectorTypeÚsplit_sql_statements)Útext)ÚEngineÚ dialect_nameÚstrÚreturnr c ó> — ddddddœ}| | d¦ « S )u‘
|
|||
|
|
å°† SQLAlchemy dialect.name æ˜ å°„åˆ° DBHub ConnectorType。
|
|||
|
|
未知方言按 postgres 规则å�šå�ªè¯»æ ¡éªŒï¼ˆä¸Žè®¡åˆ’一致）。
|
|||
|
|
ÚpostgresÚmysqlÚmariadbÚsqliteÚ sqlserver)Ú
|
|||
|
|
postgresqlr r r Úmssql)Úget)r
|
|||
|
|
Úms úBD:\CodeSpace\afe-ai\ai-g3sb-backman\src\tools\dbhub_execute_sql.pyÚ sqlalchemy_dialect_to_connectorr s5 € ð !ØØ ØØ ð #ð #€Að
|
|||
|
|
�5Š5� ˜zÑ *Ô *Ð*ó ÚvalÚobjectc ót — t | d¦ « r' | ¦ « S # t $ r | cY S w xY w| S )NÚ isoformat)Úhasattrr Ú Exception)r s r Ú_serialize_cellr# $ sP € ݈s�KÑ Ô ð ð Ø—=’=‘?”?Ð "øÝð ð ð ؈JˆJˆJð øøøà €Js ’& ¦ 5´5Úenginer Ú
|
|||
|
|
statementsú list[str]Úmax_rowsÚintú/tuple[list[str], list[dict[str, object]], bool]c ó — t dt |¦ « ¦ « }g }g }d}| ¦ « 5 }d}t |¦ « D �] \ }} | ¦ « }
|
|||
|
|
| t
|
|||
|
|
|
|
|||
|
|
¦ « ¦ « }|t |¦ « dz
|
|||
|
|
k r |j r| ¦ « Œn|j sg }g }d} n£t | ¦ « ¦ « }| |dz ¦ « } t | ¦ « |k }| d|… } g }| D ]F}
|
|||
|
|
i }t |¦ « D ] \ }}t |
|
|||
|
|
| ¦ « ||< Œ | |¦ « ŒG�Œ ddd¦ « n # 1 swxY w Y |||fS )uQ 顺åº�执行多æ�¡è¯å�¥ï¼Œä»…返回最å�Žä¸€æ�¡æœ‰ç»“果集è¯å�¥çš„列与行。é FN)Úmaxr( ÚconnectÚ enumerateÚstripÚexecuter ÚlenÚ returns_rowsÚfetchallÚlistÚkeysÚ fetchmanyr# Úappend)r$ r% r' Ú last_colsÚ last_rowsÚ truncatedÚconnÚresultÚiÚstmtÚsÚfetchedÚ rows_dataÚrowÚrow_mapÚjÚcols r Ú_execute_statementsrF - sá € õ �1•c˜(‘m”mÑ$Ô$€HØ €IØ)+€IØ€IØ � Š Ñ Ô ð *˜TØˆÝ Ñ,Ô,ð *ñ *‰GˆAˆtØ—
|
|||
|
|
’
|
|||
|
|
‘ ” ˆAØ—\’\¥$ q¡'¤'Ñ*Ô*ˆFØ•3�z‘?”? QÑ&Ò&Ð&ØÔ&ð &Ø—O’OÑ%Ô%Ð%ØØÔ&ð
|
|||
|
|
Ø � Ø � Ø !� Ø�Ý ˜VŸ[š[™]œ]Ñ+Ô+ˆIØ ×&Ò& x°!¡|Ñ4Ô4ˆGݘG™ œ xÒ/ˆIØ Ô*ˆI؈IØ ð
|
|||
|
|
*ð
|
|||
|
|
*�Ø-/�Ý '¨ Ñ 2Ô 2ð ;ð ;‘F�A�sÝ#2°3°q´6Ñ#:Ô#:�G˜C‘L�LØ× Ò Ñ)Ô)Ð)Ð)ñ
|
|||
|
|
*ð'*ð *ð *ñ *ô *ð *ð *ð *ð *ð *ð *øøøð *ð *ð *ð *ð0 �i Ð *Ð*s ¸D2E6Å6E:Å=E:TN)Úreadonlyr' ÚsqlrG Úboolú
|
|||
|
|
int | Noneúdict[str, object]c óº — | pd ¦ « }|st d¦ « ‚t d¦ « pd ¦ « }|st d¦ « ‚t ¦ « }t |j j ¦ « }t ||¦ « }|st d¦ « ‚|rK|D ]H}t ||¦ « s6d t |¦ « ¦ « pd} t d| › �¦ « ‚ŒI|�|n t t d
|
|||
|
|
¦ « pd ¦ « }
|
|||
|
|
t d t d t |
|
|||
|
|
¦ « ¦ « ¦ « }
|
|||
|
|
t j d
|
|||
|
|
t! |¦ « › d|› d|
|
|||
|
|
› �¦ « t# |||
|
|||
|
|
¬¦ « \ } } }
|
|||
|
|
nF# t$ $ r9}t j d|› �¦ « t t) |¦ « ¦ « |‚d }~ww xY wt! | ¦ « }| |d| |
|
|||
|
|
dœS )uÓ
|
|||
|
|
执行 SQL（å�¯å¤šè¯å�¥ï¼Œåˆ†å�·åˆ†éš”），行为对é½� DBHub execute_sql。
|
|||
|
|
|
|||
|
|
返回 dict � rows（list[dict]）�count（int）�source_id�columns�truncated（便于 Python 使用）。
|
|||
|
|
Ú u SQL ä¸�èƒ½ä¸ºç©ºÚ database_urlu database_url 未é…�ç½®u
|
|||
|
|
SQL 为空z, ÚnonezJRead-only mode is enabled. Only the following SQL operations are allowed: NÚ sql_max_rowsi' r+ u" dbhub_execute_sql 执行è¯å�¥æ•°=z
|
|||
|
|
readonly=z
|