436 lines
52 KiB
Plaintext
436 lines
52 KiB
Plaintext
|
Ë
|
|||
|
|
|®Üi~Ä ã ó6 — d Z ddlmZ ddlZddlZddlZddlZddlmZ ddl m
|
|||
|
|
Z
|
|||
|
|
m Z ddl m
|
|||
|
|
Z
|
|||
|
|
mZ ddlmZ ddlmZ dd lmZmZ dd
|
|||
|
|
lmZmZmZmZmZ dd l m Z ej< e « Z d;d „Z!e d
|
|||
|
|
Z"e d Z# e$h d£« Z%d<d„Z&d=d„Z'd=d„Z(d>d„Z)d?d„Z*d@d„Z+dAd„Z,dBd„Z-dCd„Z.dDd„Z/dCd„Z0 dEd„Z1dFd „Z2 e$h d £« Z3 ejh d « Z5dGd „Z6dHd „Z7dId!„Z8dJd"„Z9dJd#„Z: dKd$„Z; dLd%„Z<dMd&„Z= dNd'„Z>d(dd)œ dOd*„Z?d(d+œ dPd,„Z@d(d+œ dQd-„ZAd.d/œ dRd0„ZBd.d/œdSd1„ZC dT dUd2„ZD G d3„ d4« ZE eE« ZFdVd5„ZGdWd6„ZHdWd7„ZIdXd8„ZJdYd9„ZKe d:k( r eK« yy)Zu¯
|
|||
|
|
与 DBHub 对�的数�库工具：search_objects（元数�探索）�execute_sql（SQL 执行，对� execute-sql.ts）�
|
|||
|
|
execute_sql_all（å�Œ execute_sql 但ä¸�截æ–行数）ã€�execute_sql_count_only（仅统计行数，ä¸�返回明细）。
|
|||
|
|
Text2SQL Validator 使用 ``probe_sql_execution_status`` / ``probe_sql_execution_status_ex`` �库上探针。
|
|||
|
|
|
|||
|
|
- 模�级 ``execute_sql`` / ``execute_sql_all`` / ``execute_sql_count_only`` / ``search_objects``：便�入�，委托给 ``default_dbhub_tools``。
|
|||
|
|
- ``DbHubTools``：å�¯å®žä¾‹åŒ–的工具类，方法è¯ä¹‰ä¸Žä¸Šè¿°æ¨¡å�—函数相å�Œã€‚
|
|||
|
|
- ``default_dbhub_tools``：默认�例实例，与模�级函数共用。
|
|||
|
|
é )Ú annotationsN)ÚDecimal)ÚAnyÚLiteral)ÚinspectÚtext)ÚEngine)ÚSQLAlchemyError)Úallowed_keywords_listÚis_read_only_sql)Ú
|
|||
|
|
ConnectorTypeÚ _get_scannerÚ _TOKEN_PLAINÚsplit_sql_statementsÚstrip_comments_and_strings)Ú
|
|||
|
|
get_enginec óD — t j | « }|�2t |« j « rt |« j « S t j | j « « }|�2t |« j « rt |« j « S y)u_ 从环境å�˜é‡�读å�–ï¼ˆä¸Žæ ¹ç›®å½• .env / load_dotenv 一致）；键å��大å°�写ä¸�æ•�感。N)ÚosÚgetenvÚstrÚstripÚupper)ÚkeyÚvÚv2s ú>C:\Users\24019\Desktop\backman-camel\backend\db\dbhub_tools.pyÚ_get_optional_strr % sh € ä
|
|||
|
|
� ‰ �#‹€AØ€}œ˜Q›Ÿ ™ œ Ü�1‹v�|‰|‹~Ð Ü �‰�3—9‘9“;Ó €BØ €~œ#˜b›'Ÿ-™-œ/Ü�2‹w�}‰}‹Ð Ø ó )ÚschemaÚtableÚcolumnÚ procedureÚfunctionÚindex)ÚnamesÚsummaryÚfull> ÚsysÚmysqlÚpg_toastÚ
|
|||
|
|
pg_catalogÚinformation_schemaÚperformance_schemac ó* — g }| D ]T }|dk( r|j d« Œ|dk( r|j d« Œ1|j t j |« « ŒV t j ddj |« z dz t j
|
|||
|
|
« S )u_ SQL LIKE → regex，与 DBHub likePatternToRegex 一致（% / _，其余å—符 re.escape）。Ú%z.*Ú_Ú.Ú^Ú Ú$)ÚappendÚreÚescapeÚcompileÚjoinÚ
|
|||
|
|
IGNORECASE)ÚpatternÚpartsÚcs r Ú_like_pattern_to_regexr> ? st € à€EÛ
|
|||
|
|
ˆØ �Š8Ø �L‰L˜Õ Ø
|
|||
|
|
�#ŠXØ �L‰L˜Õ à �L‰LœŸ™ 1› Õ &ð
|
|||
|
|
ô �:‰:�c˜BŸG™G E›NÑ*¨SÑ0´"·-±-Ó @Ð@r c ód — | j « j dd« d j « S )Nr1 é éÿÿÿÿ)r Úrsplit)Únames r Ú_bare_table_namerD L s) € Ø �:‰:‹<× Ñ ˜s AÓ & rÑ *× 0Ñ 0Ó 2Ð2r c óŒ — | j « }t |« dk\ r%|d dk( r |d dk( r|dd j dd« S |S )uJ 去掉首尾空白；若整段为 [Name] å½¢å¼�则去括å�·ï¼ˆå�•å±‚）。é r Ú[rA Ú]r@ z]])r ÚlenÚreplace)rC Úts r Ú_normalize_user_object_namerL P sH € à �
|
|||
|
|
‰
|
|||
|
|
‹ €AÜ
|
|||
|
|
ˆ1ƒv�‚{�q˜‘t˜s’{ q¨¡u°¢|Ø��2ˆw�‰˜t SÓ)Ð)Ø €Hr c ó0 — t |« }|s|j « S |j « }| j | j fD ]> } ||¬« D ]/ }t t
|
|||
|
|
|« « }|j « |k( sŒ+|c c S Œ@ |S # t $ r Y ŒNw xY w)u¬
|
|||
|
|
将用户输入的表/视图å��解æž�为与 SQLAlchemy inspect 一致ã€�å�¯åœ¨ get_columns/get_indexes ä¸ä½¿ç”¨çš„å��称。
|
|||
|
|
|
|||
|
|
SQL Server ç‰åº“å¯¹æ ‡è¯†ç¬¦å¤§å°�写ä¸�æ•�感，但å��å°„æ—¶ä¼ å…¥ä¸Žç³»ç»Ÿç›®å½•ä¸�一致的大å°�写å�¯èƒ½å¯¼è‡´ get_columns 返回空；
|
|||
|
|
视图ä¸�在 get_table_names ä¸ï¼Œä»…ä¼ è¡¨å��会æ¼�匹é…�。本函数在指定 schema 下对表å��与视图å��å�šä¸�区分大å°�写对é½�。
|
|||
|
|
©r )rL r ÚlowerÚget_table_namesÚget_view_namesrD r Ú Exception)Úinspr Ú
|
|||
|
|
user_tableÚwantr ÚlisterÚrawÚbares r Ú _resolve_table_or_view_bare_namerY X s– € ô ' zÓ 2€DÙ Ø× Ñ Ó!Ð!Ø
|
|||
|
|
�*‰*‹,€CØ×'Ñ'¨×)<Ñ)<Ó=ˆð Ù V×,�Ü'¬¨C«Ó1�Ø—:‘:“< 3Ó&Ø ”Kñ -ð >ð €Køô ò Ù ð ús Á
|
|||
|
|
3B Á>B ÂB  BÂBc ó` — | D �cg c] }|sŒ|j « t vsŒ |‘Œ c}S c c}w ©N)rO Ú
|
|||
|
|
_SKIP_SCHEMAS)rW Úss r Ú_filter_schemasr^ n s) € ÙÓ C‘s�!ša A§G¡G£I´]Ò$BŠA�sÑ CÐCùÒ Cs …+�+¤+c ó„ — | j j }|j |« }|r|j |« › d|› �S |S )Nr1 )ÚdialectÚidentifier_preparerÚquoteÚ quote_schema)Úenginer Ú
|
|||
|
|
bare_tableÚprepÚtqs r Ú_qualified_table_sqlrh r sD € Ø �>‰>× -Ñ -€DØ
|
|||
|
|
�‰�JÓ €BÙ
|
|||
|
|
Ø×#Ñ# FÓ+Ð,¨A¨b¨TÐ2Ð2Ø
|
|||
|
|
€Ir c ó — t | ||« } | j « 5 }|j t d|› �« « }|j « }|€
|
|||
|
|
d d d « y t |d « cd d d « S # 1 sw Y y xY w# t $ r Y y w xY w)NzSELECT COUNT(*) AS c FROM r )rh ÚconnectÚexecuter ÚfetchoneÚintrR )rd r re ÚqÚconnÚrÚrows r Ú_table_row_countrr z sƒ € Ü ˜V V¨ZÓ8€AðØ
|
|||
|
|
�^‰^Ô
|
|||
|
|
Ø— ‘ œTÐ$>¸q¸cÐ"BÓ CÓDˆAØ—*‘*“,ˆC؈{Ø÷ Ð
|
|||
|
|
ô
|
|||
|
|
�s˜1‘v“;÷ ×
|
|||
|
|
Ò
|
|||
|
|
ûô ò Ùðús: �A< Ÿ1A0ÁA< Á
|
|||
|
|
A0Á& A< Á0A9Á5A< Á9A< Á< BÂBc ó– — | j ||¬« }|j d« xs g }t d„ |D « « S # t $ r Y yw xY w)NrN Úconstrained_columnsc 3 ó2 K — | ] }t |« –— Œ y wr[ )r )Ú.0r= s r Ú <genexpr>z _pk_columns.<locals>.<genexpr>‹ s è ø€ Ð*¡T ”S˜—V¡Tùs ‚© )Úget_pk_constraintÚgetÚtuplerR )rS r rX ÚpkÚcolss r Ú _pk_columnsr~ ‡ sS € ðØ
|
|||
|
|
×
|
|||
|
|
#Ñ
|
|||
|
|
# D°Ð
|
|||
|
|
#Ó
|
|||
|
|
8ˆØ�v‰vÐ+Ó,Ò2°ˆÜÑ*¡TÓ*Ó*Ð*øÜ ò Ùðús ‚9< ¼ AÁAc ó˜ — t | ||« } t | j ||¬« « }g }|D ]‡ }t |j d« xs g « }t |j d« « }t |« xr t
|
|||
|
|
|« |k( } |j t |j d« xs d« ||| dœ« Œ‰ |S # t $ r g cY S w xY w)NrN Ú column_namesÚuniquerC r3 )rC Úcolumnsr� Úprimary) r~ ÚlistÚ get_indexesrR rz Úboolr{ r5 r )
|
|||
|
|
rS r rX Úpk_colsÚraw_idxÚoutÚidxr} r� rƒ s
|
|||
|
|
r Ú _index_dictsr‹ � sÎ € ܘ$ ¨Ó-€GðÜ�t×'Ñ'¨°VÐ'Ó<Ó=ˆð !#€CÛˆÜ�C—G‘G˜NÓ+Ò1¨rÓ2ˆÜ�c—g‘g˜hÓ'Ó(ˆÜ�w“-Ò:¤E¨$£K°7Ñ$:ˆØ �
|
|||
|
|
‰
|
|||
|
|
ä˜CŸG™G F›OÒ 1¨rÓ2Ø Ø Ø"ñ
|
|||
|
|
õ
|
|||
|
|
ð ð €Jøô ò ØŠ ðús � B; Â; C ÃC c óè — | j ||¬« }t |t « r)|j d« xs dj « }|r|S dS y# t
|
|||
|
|
t t f$ r Y yt $ r Y yw xY w)u; 获å�–表注释，若ä¸�支æŒ�æˆ–æ— æ³¨é‡Šåˆ™è¿”å›ž None。rN r r3 N) Úget_table_commentÚ
|
|||
|
|
isinstanceÚdictrz r ÚNotImplementedErrorÚAttributeErrorÚ TypeErrorrR )rS r rX Útcr s r Ú_table_commentr” ¦ sˆ € ð
|
|||
|
|
Ø
|
|||
|
|
×
|
|||
|
|
#Ñ
|
|||
|
|
# D°Ð
|
|||
|
|
#Ó
|
|||
|
|
8ˆÜ �bœ$Ô Ø—F‘F˜6“NÒ( b×/Ñ/Ó1ˆDÙ �4Ð ) TÐ )ð ð øô
|
|||
|
|
¤ ´Ð ;ò
|
|||
|
|
Ø ð ô ò
|
|||
|
|
Ø Ø ð
|
|||
|
|
ús ‚A A Á A ÁA1Á&A1Á0A1c óˆ — t | j ||¬« « }g }|D �] }t |t « sŒt |j
|
|||
|
|
d« xs d« j « }|sŒD|j
|
|||
|
|
d« }|�t |« j « nd}|j
|
|||
|
|
d« } | du xs% t | t
|
|||
|
|
« xr | j « dk( }
|
|||
|
|
|||
|
|||
|
|
rdnd|j
|
|||
|
|
d « d
|
|||
|
|
œ} |j
|
|||
|
|
d « } | � t | « j « }
|
|||
|
|
|
|
|||
|
|
r|
|
|||
|
|
| d < |j | « �Œ |S # t $ r g cY S w xY w)
|
|||
|
|
NrN rC r3 ÚtypeÚnullableTÚYESÚNOÚdefault)Ú column_nameÚ data_typeÚ is_nullableÚcolumn_defaultÚcommentÚ description)
|
|||
|
|
r„ Ú get_columnsrR rŽ r� r rz r r r5 )rS r rX r} r‰ r= ÚfnameÚftypeÚdtyper— Únull_bÚentryrŸ Ú comment_strs r Ú
|
|||
|
|
_column_dictsr¨ ´ s? € ðÜ�D×$Ñ$ T°&Ð$Ó9Ó:ˆð !#€CÜ
|
|||
|
|
ˆÜ˜!œTÔ"Ø Ü�A—E‘E˜&“MÒ' RÓ(×.Ñ.Ó0ˆÙØ Ø—‘�f“
|
|||
|
|
ˆØ&+Ð&7”�E“
|
|||
|
|
× Ñ Ô"¸RˆØ—5‘5˜Ó$ˆØ˜TÐ!Ò^¤j°¼3Ó&?Ò&]ÀHÇNÁNÓDTÐX]ÑD]ˆà Ø Ù$*™5°Ø Ÿe™e IÓ .ñ !
|
|||
|
|
ˆð —%‘%˜ Ó"ˆØ Ð Ü ˜g›,×,Ñ,Ó.ˆKÙØ'2��mÑ$Ø �
|
|||
|
|
‰
|
|||
|
|
�5Öð- ð. €Jøô5 ò ØŠ ðús ‚ D3 Ä3 EÅ Ec
|
|||
|
|
ó˜ — g }|dvr|S d}i }|r
|
|||
|
|
|dz
|
|||
|
|
}||d< |r
|
|||
|
|
|dz
|
|||
|
|
}||d< | j « 5 }|j t |« |« D ]O }|d } |j t |d « t |d « t |d
|
|||
|
|
« | €d n
|
|||
|
|
t | « d d f« ŒQ d d d « |d v rý|rû d
|
|||
|
|
}
|
|||
|
|
i } |r
|
|||
|
|
|
|
|||
|
|
dz
|
|||
|
|
}
|
|||
|
|
|| d< |r
|
|||
|
|
|
|
|||
|
|
dz
|
|||
|
|
}
|
|||
|
|
|| d< i } j t |
|
|||
|
|
« | « D ]k }t |d « t |d « f}
|
|||
|
|
|d
|
|||
|
|
r t |d
|
|||
|
|
« j « nd }|d r t |d « j « nd }||f| |
|
|||
|
|
< Œm g }|D ]H \ }}}} }}||f}
|
|||
|
|
|
|
|||
|
|
| v r | |
|
|||
|
|
\ }}|j |||| ||f« Œ2|j |||| d d f« ŒJ |}|S |S # 1 sw Y �Œ
|
|||
|
|
xY w# t $ r Y |S w xY w)u
|
|||
|
|
返回 (schema, name, routine_type, definition_or_none, language_or_none, return_type_or_none)。
|
|||
|
|
routine_sql_type: 'PROCEDURE' | 'FUNCTION' | None（两者都�）
|
|||
|
|
|
|||
|
|
注æ„�：INFORMATION_SCHEMA.ROUTINES 在ä¸�å�Œæ•°æ�®åº“ä¸åŒ…å�«çš„å—段ä¸�å�Œï¼š
|
|||
|
|
- SQL Server: 有 EXTERNAL_LANGUAGE（但通常为 NULLï¼‰ï¼Œæ— RETURN_TYPE
|
|||
|
|
- MySQL/MariaDB: 有 EXTERNAL_LANGUAGE�DTD_IDENTIFIER（返回类型）
|
|||
|
|
- PostgreSQL: INFORMATION_SCHEMA ä¸å˜å‚¨è¿‡ç¨‹æ”¯æŒ�有é™�，通常需è¦�通过 pg_proc 查询
|
|||
|
|
)ÚmssqlÚ
|
|||
|
|
postgresqlr) Úmariadbz’
|
|||
|
|
SELECT ROUTINE_SCHEMA, ROUTINE_NAME, ROUTINE_TYPE, ROUTINE_DEFINITION
|
|||
|
|
FROM INFORMATION_SCHEMA.ROUTINES
|
|||
|
|
WHERE 1=1
|
|||
|
|
z AND ROUTINE_SCHEMA = :schÚschz AND ROUTINE_TYPE = :rtÚrté r r@ rF N)r) r¬ z³
|
|||
|
|
SELECT ROUTINE_SCHEMA, ROUTINE_NAME, EXTERNAL_LANGUAGE, DTD_IDENTIFIER
|
|||
|
|
FROM INFORMATION_SCHEMA.ROUTINES
|
|||
|
|
WHERE 1=1
|
|||
|
|
)rj rk r r5 r r rR )rd r` Ú
|
|||
|
|
schema_filterÚroutine_sql_typeÚrowsÚsqlÚparamsro rp ÚdefnÚ extra_sqlÚ extra_paramsÚ extra_mapr ÚlangÚret_typeÚ updated_rowsr rC Úrtyper0 s r Ú_fetch_routinesr½ Ô sE € ð LN€DØÐAÑA؈ ð €Cð
|
|||
|
|
€FÙØ Ð+Ñ+ˆØ%ˆˆu‰
|
|||
|
|
ÙØ Ð(Ñ(ˆØ'ˆˆt‰ à � ‰ Ô ˜TØ— ‘ œd 3›i¨Ö0ˆAØ�Q‘4ˆDØ �K‰K䘘!™“Iܘ˜!™“Iܘ˜!™“IØ ˜L‘D¬c°$«iØØð
|
|||
|
|
õ
|
|||
|
|
ñ 1÷
|
|||
|
|
ð Ð&Ñ&©4ð! ðˆIð
|
|||
|
|
,.ˆLÙ ØÐ 9Ñ9� Ø&3� ˜UÑ#Ù ØÐ 6Ñ6� Ø%5� ˜TÑ"àNPˆIØ—\‘\¤$ y£/°<Ö@�ܘ1˜Q™4“y¤# a¨¡d£)Ð,�Ø,-¨aªD”s˜1˜Q™4“y—‘Ô(°d�Ø01°!²œ3˜q ™t›9Ÿ?™?Ô,¸$�Ø"&¨Ð!1� ˜#’ð Að \^ˆLÛ04Ñ,��T˜5 $¨¨1ؘD�k�ؘ)Ñ#Ø%.¨s¡^‘N�D˜(Ø ×'Ñ'¨¨d°E¸4ÀÀxÐ(PÕQà ×'Ñ'¨¨d°E¸4ÀÀtÐ(LÕMð
|
|||
|
|
15ð ˆDð
|
|||
|
|
€Kˆ4€K÷g
|
|||
|
|
Ñ ûô^ ò à à €Kð ús µA.F/Â3C8F< Æ/F9Æ< G ÇG c ó6 — ddddddœ}|j | d« S )už
|
|||
|
|
å°† SQLAlchemy dialect.name æ˜ å°„åˆ° DBHub ConnectorType。
|
|||
|
|
未知方言按 postgres 规则å�šå�ªè¯»æ ¡éªŒï¼ˆä¸Ž DBHub 移æ¤�约定一致）。
|
|||
|
|
Úpostgresr) r¬ ÚsqliteÚ sqlserver)r« r) r¬ rÀ rª )rz )Ú dialect_nameÚms r Ú sqlalchemy_dialect_to_connectorrÄ , s- € ð !ØØ ØØ ñ #€Að
|
|||
|
|
�5‰5� ˜zÓ *Ð*r > ÚwithÚselectÚexplainz‘(?is)^(?:distinct\s+)?(?:count|sum|avg|min|max|stdev|stdevp|string_agg|group_concat|variance|var_pop|var_samp|stddev|stddev_pop|stddev_samp)\s*\(c ób — t | |« dk( ryt |« }t | « }d}d}d}d}d}||k rú || |« } | d t k7 r| d }Œ |dk( rJt j
|
|||
|
|
d| || « }
|
|||
|
|
|
|
|||
|
|
r)d}||
|
|||
|
|
j
|
|||
|
|
« z }||
|
|||
|
|
j
|
|||
|
|
« z
|
|||
|
|
}Œi|d z
|
|||
|
|
}Œo|dk( rLt j
|
|||
|
|
d
|
|||
|
|
| || « } | r1| || } t j
|
|||
|
|
d | j « « ry|d z | fS | | }
|
|||
|
|
|
|
|||
|
|
d k( r|d z
|
|||
|
|
}n$|
|
|||
|
|
d
|
|||
|
|
k( rt d|d z
|
|||
|
|
« }n|
|
|||
|
|
dk( r
|
|||
|
|
|dk( r|d z
|
|||
|
|
}|d z
|
|||
|
|
}||k rŒúy)uÕ
|
|||
|
|
定ä½�最外层 ``SELECT`` 与深度 0 上首个 ``FROM`` 之间的选择列表，返回 (列数, 列表原文æ–片)。
|
|||
|
|
``WITH`` 开头ã€�``SELECT *``ã€�æˆ–æ— æ³•é…�对到 ``FROM`` 时返回 None。
|
|||
|
|
rÅ Nr ÚLEADINGr– Úendz(?is)select\s+ÚIN_LISTr@ z
|
|||
|
|
(?is)\bfrom\bz(?is)^\s*\*\s*$Ú(Ú)Ú,) Ú_first_sql_keywordr rI r r6 ÚmatchrÊ r Úmax)r³ Ú connectorÚscanÚnÚiÚ
|
|||
|
|
list_parenÚ
|
|||
|
|
col_commasÚstateÚ
|
|||
|
|
list_startÚtokrà ÚmfÚfragr= s r Ú _parse_top_level_select_listrÝ C sq € ô
|
|||
|
|
˜#˜yÓ)¨VÒ3ØÜ ˜ Ó "€DÜ ˆC‹€AØ €AØ€JØ€JØ+4€EØ€JØ
|
|||
|
|
ˆaŠ%Ù�3˜‹lˆØ ˆv‰;œ,Ò &Ø�E‘
|
|||
|
|
ˆAØ Ø �IÒ Ü—‘Ð*¨C°°!¨HÓ5ˆAÙØ!�Ø §¡£™[�
|
|||
|
|
Ø�Q—U‘U“W‘ �ØØ
|
|||
|
|
�‰FˆAØ Ø ˜Š?Ü—‘Ð*¨C°°!¨HÓ5ˆBÙؘ: aÐ(�Ü—8‘8Ð .°·
|
|||
|
|
±
|
|||
|
|
³ Ô=Ø Ø" Q™¨Ð-Ð-Ø �‰FˆØ �Š8Ø ˜!‰O‰JØ
|
|||
|
|
�#ŠXÜ ˜Q
|
|||
|
|
¨Q¡Ó/‰JØ
|
|||
|
|
�#ŠX˜*¨š/Ø ˜!‰OˆJØ ˆQ‰ˆð9 ˆa‹%ð: r c ó — t |« }t | « }d}d}||k rk || |« }|d t k7 r|d }Œ |dk( r t j d| || « }|ry| | }|dk( r|dz
|
|||
|
|
}n|dk( rt d|dz
|
|||
|
|
« }|dz
|
|||
|
|
}||k rŒky )
|
|||
|
|
Nr r– rÊ z(?is)\bgroup\s+by\bTrÌ r@ rÍ F)r rI r r6 rÐ rÑ ) r³ rÒ rÓ rÔ rÕ ÚdepthrÚ rà r= s r Ú_has_top_level_group_byrà q s´ € Ü ˜ Ó "€DÜ ˆC‹€AØ €AØ
|
|||
|
|
€EØ
|
|||
|
|
ˆaŠ%Ù�3˜‹lˆØ ˆv‰;œ,Ò &Ø�E‘
|
|||
|
|
ˆAØ Ø �AŠ:Ü—‘Ð/°°Q°q°Ó:ˆAÙØØ �‰FˆØ �Š8Ø �Q‰J‰EØ
|
|||
|
|
�#ŠXܘ˜5 1™9Ó%ˆEØ ˆQ‰ˆð ˆa‹%ð r c óÞ — t | |« }|€y|\ }}|dk7 ryt | |« ry|j « }|rt j d|« ryt t j |« « S )u‹ å�•åˆ—ã€�æ— GROUP BYã€�且选择列表以常è§�è�šå�ˆå‡½æ•°å¼€å¤´æ—¶ï¼Œç»“æžœé›†ä¸ºæ ‡é‡�，宜直接执行内层而é�ž COUNT(*) 包裹。Fr@ z (?is)^\*\s*$)rÝ rà r r6 rÐ r† Ú_AGGREGATE_SCALAR_PREFIX)ÚinnerrÒ ÚparsedÚncolsrÜ Úfss r Ú _is_aggregate_scalar_fast_pathrç ˆ sk € ä
|
|||
|
|
)¨%°Ó
|
|||
|
|
;€FØ
|
|||
|
|
€~ØØ�K€Eˆ4Ø �‚zØÜ ˜u iÔ0ØØ
|
|||
|
|
�‰‹ €BÙ
|
|||
|
|
”—‘˜/¨2Ô.ØÜ Ô(×.Ñ.¨rÓ2Ó 3Ð3r c ó¾ — |dvr| S t |« }t | « }d}d}d}||k r€ || |« }|d t k7 r|d }Œ |dk( r1t j d| || « }|r|}||j « z
|
|||
|
|
}ŒV| | } | dk( r|dz
|
|||
|
|
}n| d k( rt
|
|||
|
|
d|dz
|
|||
|
|
« }|dz
|
|||
|
|
}||k rŒ€|dk r| S | d
|
|||
|
|
| j « j d « j « S ) uI
|
|||
|
|
SQL Server / MySQL / MariaDB：派生表内 ``ORDER BY`` è‹¥æ— TOP/OFFSET/LIMIT 会报è¯æ³•é”™è¯¯ã€‚
|
|||
|
|
在 COUNT å�查询包装å‰�去掉最外层（括å�·æ·±åº¦ä¸º 0）最å�Žä¸€æ¬¡å‡ºçŽ°çš„ ``ORDER BY`` å�Šå…¶å�Žå†…容；
|
|||
|
|
ä¸�改å�˜è¡Œæ•°ç»Ÿè®¡è¯ä¹‰ã€‚注释与å—符串内的括å�·/关键å—ä¸�å�‚与解æž�。
|
|||
|
|
©rÁ r) r¬ r rA r– rÊ z(?is)order\s+by\brÌ r@ rÍ NÚ;)r rI r r6 rÐ rÊ rÑ Úrstrip)
|
|||
|
|
r³ rÒ rÓ rÔ rÕ rß Úlast_order_by_startrÚ rà r= s
|
|||
|
|
r Ú _strip_outer_trailing_order_byrí ˜ s € ð Ð9Ñ9؈
|
|||
|
|
Ü ˜ Ó "€DÜ ˆC‹€AØ €AØ
|
|||
|
|
€EØ ÐØ
|
|||
|
|
ˆaŠ%Ù�3˜‹lˆØ ˆv‰;œ,Ò &Ø�E‘
|
|||
|
|
ˆAØ à �AŠ:Ü—‘Ð-¨s°1°Q¨xÓ8ˆAÙØ&'Ð#Ø�Q—U‘U“W‘ �ØØ �‰FˆØ �Š8Ø �Q‰J‰EØ
|
|||
|
|
�#ŠXܘ˜5 1™9Ó%ˆEØ ˆQ‰ˆð# ˆa‹%ð$ ˜QÒ Øˆ
|
|||
|
|
Ø Ð#Ð#Ð $× +Ñ +Ó -× 4Ñ 4°SÓ 9× @Ñ @Ó BÐBr c ó¬ — t | |« }|j « j « }t j d|« }|r|j d« S dS )uK 剥离注释与å—符串å—é�¢é‡�å�Žï¼Œå�–首个é�žç©ºç™½è¯�（å°�写）。z\S+r r3 )r r rO r6 ÚsearchÚgroup)r³ rÒ Úcleanedrà s r rÏ rÏ ¼ sG € ä(¨¨iÓ8€GØ�m‰m‹o×#Ñ#Ó%€GÜ
|
|||
|
|
� ‰ �&˜'Ó"€AÙˆ1�7‰7�1‹:Ð" Ð"r c ó — | j « }|s t d« ‚t ||« }d|› d�}|�-|dv r)dj d„ t |« D « « }|› d|› d�S |S ) Nõ SQL ä¸�能为空z&SELECT COUNT(*) AS __dbhub_cnt FROM (
|
|||
|
|
z
|
|||
|
|
) AS __dbhub_subqré ú, c 3 ó& K — | ] }d |› �–— Œ yw)Ú __dbhub_cNrx )rv Úks r rw z%_sql_wrapped_count.<locals>.<genexpr>Ó s è ø€ Ð@±<¨a˜I a Sœ/±<ùs ‚z (rÍ )r Ú
|
|||
|
|
ValueErrorrí r9 Úrange)Ú inner_sqlrÒ rå rã Úcorer% s r Ú_sql_wrapped_countrü Ä s‚ € ð
|
|||
|
|
|
|||
|
|
�O‰OÓ €EÙ ÜÐ+Ó,Ð,Ü *¨5°)Ó <€Eà1Ø ˆ'ð ð ð ð
|
|||
|
|
|
|||
|
|
ИYÐ*KÑ KØ— ‘ Ñ@´5¸´<Ó@Ó@ˆØ��r˜%˜ Ð"Ð"Ø €Kr c ó¸ — | j « 5 }t |« D �]% \ }}|j « }|t |« dz
|
|||
|
|
k r7|j t |« « }|j r|j « Œ_|s t d« ‚t ||« }|t v �r\t |j « |« } t | |« r||j t | « « }|j « }
|
|||
|
|
|
|
|||
|
|
�|
|
|||
|
|
d €
|
|||
|
|
ddd« y|
|
|||
|
|
d } t | t « rt! | « c cddd« S t! | « c cddd« S t# | |« } | €K|j t | « « }|j s
|
|||
|
|
ddd« yd}
|
|||
|
|
|D ] }|
|
|||
|
|
dz
|
|||
|
|
}
|
|||
|
|
Π|
|
|||
|
|
c cddd« S | \ }}t% | ||« }|j t |« « }|j « }
|
|||
|
|
|
|
|||
|
|
€
|
|||
|
|
ddd« yt! |
|
|||
|
|
d « c cddd« S |j t |« « }|j s
|
|||
|
|
ddd« yd}
|
|||
|
|
|D ] }|
|
|||
|
|
dz
|
|||
|
|
}
|
|||
|
|
Π|
|
|||
|
|
c cddd« S ddd« y# 1 sw Y yxY w)uÊ
|
|||
|
|
顺åº�执行多æ�¡è¯å�¥ï¼›å¯¹æœ€å�Žä¸€æ�¡ï¼šä¼˜å…ˆç”¨ COUNT å�查询å�ªå�–æ ‡é‡�（select/with/explain），
|
|||
|
|
�则�次执行�在 Python 侧�行计数（�组装 rows 明细）。
|
|||
|
|
r@ õ
|
|||
|
|
SQL 为空Nr )rj Ú enumerater rI rk r Ú returns_rowsÚfetchallrø rÏ Ú_COUNT_WRAP_FIRST_WORDSrí rç rl rŽ r rm rÝ rü )rd Ú
|
|||
|
|
statementsrÒ ro rÕ Ústmtr] rp Úfirstrã rq Úcellrä rÔ r0 rå Ú_fragÚwrappeds r Ú"_execute_sql_statements_count_onlyr Ø s- € ð
|
|||
|
|
� ‰ Ô ˜TÜ ×,‰GˆAˆtØ—
|
|||
|
|
‘
|
|||
|
|
“ ˆAØ”3�z“? QÑ&Ò&Ø—L‘L¤ a£Ó)�Ø—>’>Ø—J‘J”LØÙÜ Ó.Ð.Ü& q¨)Ó4ˆEØÔ/Ò/Ü6°q·w±w³yÀ)ÓL�Ü1°%¸ÔCØ Ÿ ™ ¤T¨%£[Ó1�AØŸ*™*›,�CØ�{ c¨!¡f nØ ÷#
|
|||
|
|
Ð ð$ ˜q™6�DÜ! $¬Ô0Ü " 4›yÑ(÷)
|
|||
|
|
Ñ ô* ˜t›9Ñ$÷+
|
|||
|
|
Ñ ô, 6°e¸YÓG�Ø�>Ø Ÿ ™ ¤T¨%£[Ó1�AØ Ÿ>š>Ø ÷5
|
|||
|
|
Ð ð6 �AÛ ˜Ø˜Q™™ð à ‘H÷=
|
|||
|
|
Ñ ð> &‘ ��uÜ,¨U°I¸uÓE�Ø—L‘L¤ g£ Ó/�Ø—j‘j“l�Ø�;Ø ÷I
|
|||
|
|
Ð ôJ ˜3˜q™6“{Ñ"÷K
|
|||
|
|
Ñ ðL — ‘ œT !›WÓ%ˆAØ—>’>Ø÷Q
|
|||
|
|
Ð ðR ˆAÛ�Ø�Q‘‘ð à‰H÷Y
|
|||
|
|
Ñ Ù,÷
|
|||
|
|
× Ñ úsB ‘C%IÃ?!IÄ* IÅ 5IÅ>IÆ?IÇ!IÇ:'IÈ*IÉIÉIc óü — t | t « r t | « S t | t t f« r t | « j dd¬« S t
|
|||
|
|
| d« r | j « S | S # t $ r | cY S w xY w)Nzutf-8rJ )ÚerrorsÚ isoformat) rŽ r r ÚbytesÚ bytearrayÚdecodeÚhasattrr rR )Úvals r Ú_serialize_sql_cellr sv € Ü�#”wÔ Ü�3‹xˆÜ�#œœyÐ)Ô*Ü�S‹z× Ñ °Ð Ó;Ð;܈s�KÔ ð Ø—=‘=“?Ð "ð €Jøô ò ØŠJð ús ÁA- Á- A;Á:A;c ó¼ — g }g }d}| j « 5 }d}t |« D �] \ }} | j « }
|
|||
|
|
|j t |
|
|||
|
|
« « }|t |« dz
|
|||
|
|
k r |j r|j « Œ_|j sg }g }d} nt |j « « }|€|j « } d}n<t dt |« « } |j | dz « }
|
|||
|
|
t |
|
|||
|
|
« | kD }|
|
|||
|
|
d| } g }| D ]9 }i }t |« D ] \ }}t || « ||< Œ |j |« Œ; �Œ ddd« |||fS # 1 sw Y ŒxY w)u• 顺åº�执行多æ�¡è¯å�¥ï¼Œä»…返回最å�Žä¸€æ�¡æœ‰ç»“果集è¯å�¥çš„列与行。``max_rows`` 为 None 时对最å�Žä¸€æ�¡ ``fetchall``，ä¸�截æ–。FNr@ )rj rÿ r rk r rI r r r„ ÚkeysrÑ rm Ú fetchmanyr r5 )rd r Úmax_rowsÚ last_colsÚ last_rowsÚ truncatedro ÚresultrÕ r r] Ú rows_dataÚmrÚfetchedrq Úrow_mapÚjÚcols r Ú_execute_sql_statementsr! s^ € ð €IØ)+€IØ€IØ � ‰ Ô ˜TØˆÜ ×,‰GˆAˆtØ—
|
|||
|
|
‘
|
|||
|
|
“ ˆAØ—\‘\¤$ q£'Ó*ˆFØ”3�z“? QÑ&Ò&Ø×&Ò&Ø—O‘OÔ%ØØ×&Ò&Ø � Ø � Ø !� ÙÜ ˜VŸ[™[›]Ó+ˆIØÐ Ø "ŸO™OÓ -� Ø !‘ 䘜C ›MÓ*�Ø ×*Ñ*¨2°©6Ó2�Ü ›L¨2Ñ -� Ø # C R˜L� ؈IÛ �Ø-/�Ü '¨ Ö 2‘F�A�sÜ#6°s¸1±vÓ#>�G˜C’Lð 3à× Ñ Õ)ò !ð- -÷
|
|||
|
|
ð: �i Ð *Ð*÷;
|
|||
|
|
Ð ús —D.EÅET©Úreadonlyr c ó* — | xs dj « }|s t d« ‚t d« xs dj « }|s t d« ‚t « }t |j
|
|||
|
|
j « }t ||« }|s t d« ‚|r?|D ]: }t ||« rŒdj t |« « xs d} t d| › �« ‚ |�|nt t d
|
|||
|
|
« xs d « }
|
|||
|
|
t d t d t |
|
|||
|
|
« « « }
|
|||
|
|
t j d
|
|||
|
|
t! |« › d|› d|
|
|||
|
|
› �« t# |||
|
|||
|
|
¬« \ } } }
|
|||
|
|
t! | « }| |d| |
|
|||
|
|
dœS # t$ $ r2}t j' d|› �« t t) |« « |‚d }~ww xY w)u
|
|||
|
|
在已é…�置的业务库上执行 SQL（å�¯å¤šè¯å�¥ï¼Œåˆ†å�·åˆ†éš”）。
|
|||
|
|
:param sql: 待执行 SQL å—符串。多æ�¡è¯å�¥ç”¨åˆ†å�·åˆ†éš”ï¼›
|
|||
|
|
:param readonly: 是å�¦å�¯ç”¨å�ªè¯»æ ¡éªŒï¼Œé»˜è®¤ True。为 True 时，任一æ�¡è¯å�¥ä¸�符å�ˆ G3SB å…�许的首关键å—
|
|||
|
|
（如 selectã€�withã€�explain ç‰ï¼Œéš�方言而异）或å�«å�˜æ›´ç±»å…³é”®å—则抛出 ValueError。
|
|||
|
|
为 False æ—¶ä¸�å�šä¸Šè¿°æ ¡éªŒï¼Œå�¯æ‰§è¡Œ DML/DDL（风险自负，勿用于ä¸�å�¯ä¿¡è¾“入）。
|
|||
|
|
:param max_rows: 对「最å�Žä¸€æ�¡è¿”回结果集的è¯å�¥ã€�最多å�–多少行；None 时使用é…�置项 sql_max_rows，
|
|||
|
|
并在实现侧夹紧到 1～10000。å‰�é�¢çš„è¯å�¥è‹¥äº§ç”Ÿç»“果集会被消费掉但ä¸�返回。
|
|||
|
|
:return: å—典，å�« rowsã€�countã€�source_id（固定 ``default``）ã€�columnsã€�truncated。
|
|||
|
|
最å�Žä¸€æ�¡è¯å�¥æ— 结果集时 rowsã€�columns 为空，count 为 0。未é…�ç½® database_urlã€�SQL 为空或执行失败时
|
|||
|
|
抛出 ValueError（或包装�的底层异常信�）。
|
|||
|
|
r3 ró Ú database_urlõ database_url 未é…�ç½®rþ rô ÚnoneúJRead-only mode is enabled. Only the following SQL operations are allowed: NÚ sql_max_rowsi' r@ u _execute_sql 执行è¯å�¥æ•°=ú
|
|||
|
|
readonly=z
|
|||
|
|
max_rows=©r u _execute_sql 执行失败: rš ©r² ÚcountÚ source_idr‚ r )r rø r r rÄ r` rC r r r9 r rm rÑ ÚminÚlogÚinforI r! rR Úerrorr )r³ r# r rW Úurlrd rÒ r ÚstÚkwr r} r² r Úer- s r Ú _execute_sqlr7 G sž € ð$ Š9�"×
|
|||
|
|
Ñ
|
|||
|
|
Ó
|
|||
|
|
€CÙ ÜÐ+Ó,Ð,ä ˜^Ó ,Ò 2°×
|
|||
|
|
9Ñ
|
|||
|
|
9Ó
|
|||
|
|
;€CÙ ÜÐ1Ó2Ð2ä
|
|||
|
|
‹\€FÜ/°·±×0CÑ0CÓD€IÜ% c¨9Ó5€Já ܘ Ó&Ð&áÛ ˆBÜ# B¨ Õ2Ø—Y‘YÔ 4°YÓ ?Ó@ÒJÀF�Ü Ø`ÐacÐ`dÐeóð ð ð Ð)‰¬sÔ3DÀ^Ó3TÒ3]ÐX]Ó/^€BÜ ˆQ”�Eœ3˜r›7Ó#Ó $€Bä‡H�HÐ,¬S°«_Ð,=¸ZÈÀzÐQ[Ð\^Ð[_Ð
|
|||
|
|
`Ôað(Ü 7¸À
|
|||
|
|
ÐUWÔ XÑ ˆˆd�Iô
|
|||
|
|
|
|||
|
|
�‹I€EàØØ ØØ ñ ð øô ò (Ü � ‰ Ð/°¨sÐ3Ô4Üœ˜Q›Ó aÐ'ûð(ús Ä2E Å FÅ -F
|
|||
|
|
Æ
|
|||
|
|
F©r# c ó® — | xs dj « }|s t d« ‚t d« xs dj « }|s t d« ‚t « }t |j
|
|||
|
|
j « }t ||« }|s t d« ‚|r?|D ]: }t ||« rŒdj t |« « xs d}t d|› �« ‚ t j d t |« › d
|
|||
|
|
|› �« t ||d ¬ « \ } }
|
|||
|
|
} t |
|
|||
|
|
« }
|
|||
|
|
|
|
|||
|
|
|
|
|||
|
|
d| | dœS # t $ r2} t j! d
|
|||
|
|
| › �« t t# | « « | ‚d } ~ ww xY w)uõ
|
|||
|
|
与 ``_execute_sql`` 相å�Œæ ¡éªŒä¸Žè¿”回结构，但对最å�Žä¸€æ�¡æœ‰ç»“果集的è¯å�¥ä¸�å�š ``max_rows`` 截æ–，全部 ``fetchall``。
|
|||
|
|
超大结果集会å� 用大é‡�内å˜ï¼Œä»…用于å�¯ä¿¡ SQL / 已自行 LIMIT 的场景。
|
|||
|
|
r3 ró r% r& rþ rô r' r( u! _execute_sql_all 执行è¯å�¥æ•°=r* Nr+ u _execute_sql_all 执行失败: rš r, )r rø r r rÄ r` rC r r r9 r r0 r1 rI r! rR r2 r )r³ r# rW r3 rd rÒ r r4 r5 r} r² r r6 r- s r Ú_execute_sql_allr: … sa € ð Š9�"×
|
|||
|
|
Ñ
|
|||
|
|
Ó
|
|||
|
|
€CÙ ÜÐ+Ó,Ð,ä ˜^Ó ,Ò 2°×
|
|||
|
|
9Ñ
|
|||
|
|
9Ó
|
|||
|
|
;€CÙ ÜÐ1Ó2Ð2ä
|
|||
|
|
‹\€FÜ/°·±×0CÑ0CÓD€IÜ% c¨9Ó5€Já ܘ Ó&Ð&áÛ ˆBÜ# B¨ Õ2Ø—Y‘YÔ 4°YÓ ?Ó@ÒJÀF�Ü Ø`ÐacÐ`dÐeóð ð ô ‡H�HÐ0´°Z³ Ð0AÀÈHÈ:Ð
|
|||
|
|
VÔWð(Ü 7¸À
|
|||
|
|
ÐUYÔ ZÑ ˆˆd�Iô
|
|||
|
|
|
|||
|
|
�‹I€EàØØ ØØ ñ ð øô ò (Ü � ‰ Ð3°A°3Ð7Ô8Üœ˜Q›Ó aÐ'ûð(ús Ã4D Ä EÄ"-EÅEc ón — | xs dj « }|s t d« ‚t d« xs dj « }|s t d« ‚t « }t |j
|
|||
|
|
j « }t ||« }|s t d« ‚|r?|D ]: }t ||« rŒdj t |« « xs d}t d|› �« ‚ t j d t |« › d
|
|||
|
|
|› �« t |||« } d | iS # t $ r&}
|
|||
|
|
t j! d |
|
|||
|
|
› �« d d
|
|||
|
|
icY d}
|
|||
|
|
~
|
|||
|
|
S d}
|
|||
|
|
~
|
|||
|
|
ww xY w)uµ
|
|||
|
|
在已é…�置的业务库上执行 SQL，仅返回「最å�Žä¸€æ�¡æœ‰ç»“果集è¯å�¥ã€�的行数，ä¸�返回å�•å…ƒæ ¼æ˜Žç»†ã€‚
|
|||
|
|
|
|||
|
|
对以 select / with / explain 开头（注释剥离å�Žï¼‰çš„最å�Žä¸€æ�¡è¯å�¥ï¼šè‹¥å·²ä¸ºå�•åˆ—常è§�è�šå�ˆï¼ˆå¦‚ ``count(1)``ï¼‰ä¸”æ— ``GROUP BY``，
|
|||
|
|
则直接执行该è¯å�¥å¹¶å�–æ ‡é‡�，é�¿å…� ``COUNT(*)`` 外包一层å�ªå¾—到 1 行。å�¦åˆ™åœ¨åº“侧用 ``SELECT COUNT(*) FROM ( ... )``ï¼›
|
|||
|
|
对 SQL Server / MySQL / MariaDB 会为派生表补上 ``(__dbhub_c0, ...)`` 列å��（é�¿å…�匿å��列 8155 ç‰ï¼‰ï¼Œå¹¶åœ¨åŒ…装å‰�去掉
|
|||
|
|
æœ€å¤–å±‚æ— æ„�义的 ``ORDER BY``ã€‚æ— æ³•è§£æž�选择列表时（如 ``WITH``ã€�``SELECT *``）改为拉全结果在 Python 侧é€�行计数。
|
|||
|
|
其它å�ªè¯»è¯å�¥ï¼ˆå¦‚部分 show/pragma）ä»�å�•æ¬¡æ‰§è¡Œå�Žé€�行计数，ä¸�组装为 dict。
|
|||
|
|
|
|||
|
|
多æ�¡è¯å�¥ç”¨åˆ†å�·åˆ†éš”时，å‰�é�¢çš„è¯å�¥ç…§å¸¸æ‰§è¡Œå¹¶æ¶ˆè´¹ç»“果集；仅对最å�Žä¸€æ�¡ç»™å‡ºè®¡æ•°ã€‚
|
|||
|
|
最å�Žä¸€æ�¡æ— 结果集时 count 为 0。进入执行阶段å�Žè‹¥ä»»ä¸€æ¥å¤±è´¥ï¼ˆå�«è¯æ³•ã€�æ�ƒé™�ã€�连接ä¸æ–ç‰ï¼‰ï¼Œè¿”回 ``{"count": -1}``ï¼›
|
|||
|
|
SQL 为空ã€�未é…�ç½® ``database_url``ã€�å�ªè¯»æ ¡éªŒä¸�通过ä»�抛 ``ValueError``。
|
|||
|
|
r3 ró r% r& rþ rô r' r( u( _execute_sql_count_only 执行è¯å�¥æ•°=r* u& _execute_sql_count_only 执行失败: r- rA N)r rø r r rÄ r` rC r r r9 r r0 r1 rI r rR r2 ) r³ r# rW r3 rd rÒ r r4 r5 rÔ r6 s r Ú_execute_sql_count_onlyr< · s< € ð$ Š9�"×
|
|||
|
|
Ñ
|
|||
|
|
Ó
|
|||
|
|
€CÙ ÜÐ+Ó,Ð,ä ˜^Ó ,Ò 2°×
|
|||
|
|
9Ñ
|
|||
|
|
9Ó
|
|||
|
|
;€CÙ ÜÐ1Ó2Ð2ä
|
|||
|
|
‹\€FÜ/°·±×0CÑ0CÓD€IÜ% c¨9Ó5€Já ܘ Ó&Ð&áÛ ˆBÜ# B¨ Õ2Ø—Y‘YÔ 4°YÓ ?Ó@ÒJÀF�Ü Ø`ÐacÐ`dÐeóð ð ô ‡H�HÐ7¼¸J»Ð7HÈ
|
|||
|
|
ÐS[ÐR\Ð
|
|||
|
|
]Ô^ð Ü .¨v°zÀ9Ó Mˆð
|
|||
|
|
|
|||
|
|
�Qˆ<Ðøô ò Ü � ‰ Ð:¸1¸#Ð>Ô?ؘˆ}Õ ûð ús Ã4
|
|||
|
|
D Ä D4ÄD/Ä)D4Ä/D4r@ r+ c óZ — t d« xs dj « }|sy t | d|¬« }t |j d
|
|||
|
|
« xs g « }t |« d kD ry y
|
|||
|
|
# t $ rC}t j d|› �« t
|
|||
|
|
|« j « xs d}d|fcY d }~S d }~ww xY w)uQ
|
|||
|
|
在已é…�置的业务库上对 SQL å�šå�ªè¯»è¯•æ‰§è¡Œï¼Œè¿”回状æ€�ç �与失败时的错误摘è¦�。
|
|||
|
|
|
|||
|
|
Returns:
|
|||
|
|
二元组 ``(status, error_message)``：
|
|||
|
|
|
|||
|
|
- ``(None, None)``：未�置 ``database_url``，跳过探针。
|
|||
|
|
- ``(1, None)``：执行æˆ�功，且**至少返回一行数æ�®**ï¼ˆä¸Žã€Œæœ‰åˆ—æ— è¡Œã€�çš„ SELECT 区分）。
|
|||
|
|
- ``(0, None)``：执行æˆ�功，但**æ•°æ�®è¡Œæ•°ä¸º 0**（å�¯æœ‰åˆ—å��è€Œæ— è¡Œï¼Œæˆ–æ— é�žç©ºç»“果集）。
|
|||
|
|
- ``(-1, msg)``：执行失败；``msg`` 为异常信�摘�（便于生�阶段�试）。
|
|||
|
|
r% r3 )NNTr" u) probe_sql_execution_status 执行失败: z
|
|||
|
|
unknown errorrA Nr² r )r@ N)r N)
|
|||
|
|
r r r7 rR r0 Úwarningr r„ rz rI )r³ r r3 r‰ r6 Údetailr² s r Ú probe_sql_execution_status_exr@ ë s£ € ô ˜^Ó ,Ò 2°×
|
|||
|
|
9Ñ
|
|||
|
|
9Ó
|
|||
|
|
;€CÙ Øðܘ3¨¸ÔAˆô
|
|||
|
|
�—‘˜“Ò% 2Ó &€DÜ
|
|||
|
|
ˆ4ƒy�1‚}ØØ øô ò Ü � ‰ Ð?À¸sÐCÔDÜ�Q“— ‘ “ Ò2 ?ˆØ�6ˆzÕûðús ¢A Á B*Á'8B% B*Â%B*c ó& — t | |¬« \ }}|S )uå
|
|||
|
|
在已é…�置的业务库上对 SQL å�šå�ªè¯»è¯•æ‰§è¡Œï¼Œè¿”回紧凑状æ€�ç �（用于 Validator 探针）。
|
|||
|
|
|
|||
|
|
è¯ä¹‰ä¸Ž :func:`probe_sql_execution_status_ex` 的首个返回值一致；失败细节请用 ``_ex``。
|
|||
|
|
r+ )r@ )r³ r Ústatusr0 s r Úprobe_sql_execution_statusrC s € ô .¨c¸HÔE�I€FˆAØ €Mr c ó@
|
|||
|
|
— |r
|
|||
|
|
|s t d« ‚|r| dvrt d| › d�« ‚t dt dt |« « « }|�|nd}t |« }t d « xs d
|
|||
|
|
j
|
|||
|
|
« } | s t d « ‚ t « }
|
|||
|
|
t |
|
|||
|
|
« } t | j « « }
|
|||
|
|
t! |
|
|||
|
|
« }|r.||
|
|||
|
|
vr&dj# |
|
|||
|
|
« xs d}t d|› d|› �« ‚|g}n|r|ndg}|
|
|||
|
|
j$ j& }g }| dk( r“|r|n|
|
|||
|
|
}|D �cg c] }|j) |« sŒ|‘Œ }}|D ]e }t+ |« |k\ r �nO|dk( r|j- d|i« Œ- t | j/ |¬« « }|j- |t+ |« dœ« Œg �nù| dk( �rs|D �]k }t+ |« |k\ r �nÜ t | j/ |¬« « }|D �]5 }t+ |« |k\ r ŒGt1 t |« « }|j) |« sŒ:|dk( r|j- ||dœ« ŒT|dk( rPt3 | ||« }t5 |
|
|||
|
|
||« }t7 | ||« } ||t+ |« |dœ} | r| | d< |j- | « Œ©t3 | ||« }t9 | ||« } t5 |
|
|||
|
|
||« }t7 | ||« } ||t+ |« ||D � cg c]% } | d | d | d d k( | d d!œd"| v rd"| d" ini ¥‘Œ' c} | d#œ} | r| | d< |j- | « �Œ8 �Œn �n€| dv �rÑ|D �]Ê }t+ |« |k\ r �nd |rt; | ||« g}!n1| j/ |¬« D �"cg c] }"t1 t |"« « ‘Œ }!}"|!D �]m }t+ |« |k\ r Œnt1 t |« « }| d$k( r�t3 | ||« }|D ]z } t+ |« |k\ r ŒO|j) | d « sŒ(|dk( r|j- | d ||d%œ« ŒF| d ||| d | d d k( | d d&œ}#d"| v r| d" |#d"< |j- |#« Œ| Œºt9 | ||« } | D ]¢ }$t+ |« |k\ r ŒÜt |$j= d« xs d
|
|||
|
|
« }%|j) |%« sŒC|dk( r|j- |%||d%œ« Œ^|j- |%|||$j= d'« |$j= d(« |$j= d)« d*œ« Œ¤ �Œp �ŒÍ nª| d+v r˜| d,k( rd-nd.}& t? |
|
|||
|
|
|||&¬/« }'|'D ]y \ }(})}*}+t+ |« |k\ r ns|j) |)« sŒ*|dk( r|j- |)|(dœ« ŒD|dk( r|j- |)|(|*ddd2œ« Œa|j- |)|(|*ddd|+d3œ« Œ{ nt d4| › �« ‚| ||||t+ |« |t+ |« |k( d5œS # t $ r} t t | « « | ‚d} ~ ww xY w# t $ r,} t j d | › �« t d
|
|||
|
|
| › �« | ‚d} ~ ww xY w# t $ r g }
|
|||
|
|
Y �Œww xY wc c}w # t $ r g }Y �Œ½w xY w# t $ r Y �Œ¤w xY wc c} w c c}"w # t $ r Y �ŒFw xY w# t $ r(} t jA d0|› d1| › �« g }'Y d} ~ �ŒŠd} ~ ww xY w)6u
|
|||
|
|
按对象类型在已é…�置的业务库ä¸æŽ¢ç´¢ schemaã€�表ã€�列ã€�索引ã€�å˜å‚¨è¿‡ç¨‹/函数ç‰å…ƒæ•°æ�®ï¼ˆå¯¹é½� DBHub search_objects）。
|
|||
|
|
会过滤系统 schema（如information_schemaã€�sysã€�pg_catalog ç‰ï¼‰ï¼Œå†�按 object_type 与 pattern å�š SQL LIKE é£Žæ ¼åŒ¹é…�ï¼›
|
|||
|
|
:param object_type: �探索的对象类别。schema-模��；table-数�表；column-列（�与 table�schema �用
|
|||
|
|
é™�定å�•è¡¨ï¼‰ï¼›index-索引（å�Œä¸Šï¼‰ï¼›procedure-å˜å‚¨è¿‡ç¨‹ï¼›function-å‡½æ•°ï¼ˆæ ‡é‡�/表值ç‰ï¼Œè§†åº“而定）。
|
|||
|
|
:param pattern: SQL LIKE 模å¼�，默认 "%" 表示ä¸�过滤å��称。"%" 匹é…�ä»»æ„�长度å�串，"_" 匹é…�å�•ä¸ªå—符；对表å��ã€�
|
|||
|
|
列å��ã€�索引å��ã€�例程å��ç‰å�šå¤§å°�写ä¸�æ•�感匹é…�。
|
|||
|
|
:param schema: �定在�个 schema 内查找；None 表示在多个�系统 schema 上�次查找。若给出具体�称，
|
|||
|
|
必须是库ä¸å·²å˜åœ¨çš„ schema，å�¦åˆ™æŠ› ValueError。使用å�‚æ•° table 时必须å�Œæ—¶æŒ‡å®š schema。
|
|||
|
|
:param table: 仅在 object_type 为 column 或 index æ—¶å…�è®¸ä¼ å…¥ï¼Œä¸Ž schema å…±å�Œé™�定「å�ªæŸ¥è¿™ä¸€å¼ 表或视图ã€�上的列
|
|||
|
|
或索引；用于其它 object_type 时会抛 ValueError。�称会在该 schema 下与�射得到的表��视图��
|
|||
|
|
ä¸�区分大å°�写匹é…�（并支æŒ� [Name] 写法），以兼容 SQL Server ç‰å¯¹ç›®å½•å¤§å°�写ä¸�æ•�感但 API 需真实写法的情况。
|
|||
|
|
:param detail_level: 返回粒度。names-仅对象å��å�Šå®šä½�å—段（如 schemaã€�table）；summary-å¢žåŠ ç®€è¦�元数æ�®
|
|||
|
|
（如表的 column_countã€�row_count，列的类型ã€�å�¯ç©ºç‰ï¼‰ï¼›full-表级返回列列表ã€�索引列表ç‰å®Œæ•´ç»“构。
|
|||
|
|
:param limit: 最多返回的结果æ�¡æ•°ï¼Œé»˜è®¤ 100，有效范围 1～1000ï¼ˆä¼ å…¥å€¼ä¼šè¢«å¤¹ç´§åˆ°è¯¥åŒºé—´ï¼‰ã€‚
|
|||
|
|
:return: 包å�« object_typeã€�patternã€�schemaã€�tableã€�detail_levelã€�countã€�resultsã€�truncated çš„å—典。
|
|||
|
|
未�置 database_url�连接失败或内�失败时抛出 ValueError（或其它底层异常）。
|
|||
|
|
z7The 'table' parameter requires 'schema' to be specified)r! r$ zLThe 'table' parameter only applies to object_type 'column' or 'index', not 'Ú'r@ iè Nr/ r% r3 r& u _search_objects inspect 失败: u æ•°æ�®åº“内çœ�失败: rô z(none)zSchema 'z%' does not exist. Available schemas: r r% rC rN )rC Ú table_countr )rC r r& )rC r Ú column_countÚ row_countrŸ r› rœ r� r˜ rž )rC r– r— rš r )rC r rG rH r‚ Úindexesr! )rC r r )rC r r r– r— rš r‚ r� rƒ )rC r r r‚ r� rƒ )r" r# r" Ú PROCEDUREÚFUNCTION)r° r± u+ _search_objects 例程列表失败 dialect=z: )rC r r– ÚlanguageÚ return_type)rC r r– rL Ú
|
|||
|
|
parametersrM Ú
|
|||
|
|
definitionzUnsupported object_type: )Ú object_typer; r r Ú detail_levelr- Úresultsr )!rø rÑ r/ rm r> r r r r r r
|
|||
|
|
r0 r2 r„ Úget_schema_namesrR r^ r9 r` rC rÐ rI r5 rP rD r¨ rr r” r‹ rY rz r½ r> ),rP r; r r rQ ÚlimitÚlimÚpatÚrxr3 rd r6 rS Úall_schema_namesÚfiltered_schemasÚavailÚschemas_to_searchr` rR Ú base_schemasr] Ú
|
|||
|
|
candidatesÚ schema_nameÚtblsÚ
|
|||
|
|
table_namerX r} ÚrcrŸ Ú result_dictÚidxsr= Úresult_dict_fullÚtables_to_searchrK Úresult_dict_colrŠ Úinamer® Ú routine_rowsr rC r¼ rµ s, r Ú_search_objectsri sž € ñ4
|