130 lines
13 KiB
Plaintext
130 lines
13 KiB
Plaintext
|
Ë
|
|||
|
|
ÔŸØi, ã óÆ — d Z ddlZddlZddlmZ ddlmZmZmZm Z ddl
|
|||
|
|
Z
|
|||
|
|
ddl m Z m
|
|||
|
|
Z
|
|||
|
|
mZmZ e
|
|||
|
|
j e« Z G d„ d« Zd ed
|
|||
|
|
ed efd „Z dd
|
|||
|
|
edee d efd„Zy)u:
|
|||
|
|
SchemaåŠ è½½å™¨ - 解æž�JSON/DDLæ ¼å¼�çš„æ•°æ�®åº“Schema
|
|||
|
|
é N)ÚPath)ÚListÚDictÚOptionalÚTupleé )ÚTableÚColumnÚ
|
|||
|
|
ForeignKeyÚDatabaseSchemac óº — e Zd ZdZddefd„Z ddedee defd„Zd ed
|
|||
|
|
edefd „Z d e
|
|||
|
|
de fd
|
|||
|
|
„Z dede
|
|||
|
|
e fd„Zdede
|
|||
|
|
e fd„Zde
|
|||
|
|
e de
|
|||
|
|
e fd„Zde
|
|||
|
|
e fd„Zy)Ú SchemaLoaderu( SchemaåŠ è½½å™¨ - 支æŒ�JSONå’ŒDDLæ ¼å¼�NÚ
|
|||
|
|
schema_dirc ó8 — |rt |« | _ yd| _ y)uj
|
|||
|
|
åˆ�始化SchemaåŠ è½½å™¨
|
|||
|
|
|
|||
|
|
Args:
|
|||
|
|
schema_dir: Schema文件目录路径
|
|||
|
|
N)r r )Úselfr s úJC:\Users\24019\Desktop\backman-camel\text2sql_agent_camel\schema\loader.pyÚ__init__zSchemaLoader.__init__ s € ñ /9œ$˜zÓ*ˆ�¸dˆ�ó Ú json_pathÚg3sb_meta_pathÚreturnc óF — t |« }|j « st d|› �« ‚t |dd¬« 5 }t j
|
|||
|
|
|« }ddd« j
|
|||
|
|
d|j « }|j
|
|||
|
|
d« }|j
|
|||
|
|
d« }t |t « �rI|�rFi } |rtt |« }
|
|||
|
|
|
|
|||
|
|
j « rCt |
|
|||
|
|
dd¬« 5 } t j
|
|||
|
|
| « } ddd« j
|
|||
|
|
di « xs i } nt j d |« g }
|
|||
|
|
|j « D ]Œ \ }}t |t « sŒd}| r3t | j
|
|||
|
|
|« t « r| | j
|
|||
|
|
d
|
|||
|
|
« }| j |« }|
|
|||
|
|
j! t# |||| j% |« g ¬ « « ŒŽ t j' d |t) |
|
|||
|
|
« « t+ ||
|
|||
|
|
¬
|
|||
|
|
« S t |t, « rR|rP|D �cg c] }| j/ |« ‘Œ }
|
|||
|
|
}t j' d|› dt) |
|
|||
|
|
« › d�« t+ ||
|
|||
|
|
¬
|
|||
|
|
« S g }
|
|||
|
|
t j' d|› dt) |
|
|||
|
|
« › d�« t+ ||
|
|||
|
|
¬
|
|||
|
|
« S # 1 sw Y �Œ8xY w# 1 sw Y �Œ¨xY wc c}w )u.
|
|||
|
|
从JSONæ–‡ä»¶åŠ è½½Schema
|
|||
|
|
|
|||
|
|
JSONæ ¼å¼�示例：
|
|||
|
|
{
|
|||
|
|
"database": "db_name",
|
|||
|
|
"tables": [
|
|||
|
|
{
|
|||
|
|
"name": "table1",
|
|||
|
|
"comment": "表�述",
|
|||
|
|
"columns": [
|
|||
|
|
{"name": "col1", "type": "INT", "comment": "...", "nullable": false}
|
|||
|
|
],
|
|||
|
|
"primary_keys": ["col1"],
|
|||
|
|
"foreign_keys": [
|
|||
|
|
{"columns": ["col2"], "ref_table": "table2", "ref_columns": ["col1"]}
|
|||
|
|
]
|
|||
|
|
}
|
|||
|
|
]
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
亦支æŒ� G3SB table_structure.json：顶层为 ``schemas`` å—典（与 ``tables`` 数组二选一
|
|||
|
|
时优先使用�空的 ``schemas``）。�选 ``g3sb_meta_path`` 指� table_meta.json
|
|||
|
|
以�并表级注释。
|
|||
|
|
|
|||
|
|
Args:
|
|||
|
|
json_path: JSON文件路径
|
|||
|
|
g3sb_meta_path: �选，G3SB table_meta.json（� tables.{表�}.comment）
|
|||
|
|
|
|||
|
|
Returns:
|
|||
|
|
DatabaseSchema对象
|
|||
|
|
u Schema文件ä¸�å˜åœ¨: Úrúutf-8©ÚencodingNÚdatabaseÚschemasÚtablesu2 G3SB meta 文件ä¸�å˜åœ¨ï¼Œå°†è·³è¿‡è¡¨æ³¨é‡Š: %sÚcomment©Únamer ÚcolumnsÚ primary_keysÚ foreign_keysu6 [OK] åŠ è½½Schema完æˆ�(G3SB schemas): %s, å…±%då¼ è¡¨©r" r u [OK] åŠ è½½Schema完æˆ�: u , å…±õ å¼ è¡¨)r ÚexistsÚFileNotFoundErrorÚopenÚjsonÚloadÚgetÚstemÚ
|
|||
|
|
isinstanceÚdictÚis_fileÚloggerÚwarningÚitemsÚstrÚ_parse_g3sb_table_structureÚappendr Ú_extract_primary_keysÚinfoÚlenr ÚlistÚ _parse_table)r r r ÚpathÚfÚdataÚdb_nameÚ schemas_mapÚ tables_dataÚ
|
|||
|
|
table_metaÚmpÚmfÚ meta_datar Ú
|
|||
|
|
table_nameÚ
|
|||
|
|
structure_strr r# Ú
|
|||
|
|
table_datas r Úload_from_jsonzSchemaLoader.load_from_json sR € ôF �I‹ˆØ�{‰{Œ}Ü#Ð&=¸i¸[Ð$IÓJÐ Jä
|
|||
|
|
�$˜ gÕ
|
|||
|
|
.°!Ü—9‘9˜Q“<ˆD÷ /ð —(‘(˜: t§y¡yÓ1ˆà—h‘h˜yÓ)ˆ Ø—h‘h˜xÓ(ˆ ô �k¤4Õ (ª[Ø !ˆJ٠ܘ.Ó)�Ø—:‘:”<Ü ˜b #°Õ8¸BÜ$(§I¡I¨b£M˜ ÷ 9à!*§ ¡ ¨x¸Ó!<Ò!BÀ‘Jä—N‘NØLÈnôð ˆFØ-8×->Ñ->Ö-@Ñ)�
|
|||
|
|
˜MÜ! -´Ô5Ø Ø �Ù ¤*¨Z¯^©^¸JÓ-GÌÔ"NØ (¨Ñ 4× 8Ñ 8¸Ó C�GØ ×:Ñ:¸=ÓI�Ø—
|
|||
|
|
‘
|
|||
|
|
ÜØ 'Ø 'Ø 'Ø%)×%?Ñ%?ÀÓ%HØ%'ô õð .Aô
|
|||
|
|
�K‰KØHÈ'ÔSVÐW]ÓS^ô
|
|||
|
|
ô " w°vÔ>Ð >ä �k¤4Ô (©[ÙFQÓRÁk¸
|
|||
|
|
�d×'Ñ'¨
|
|||
|
|
Õ3ÀkˆFÐRÜ �K‰KÐ3°G°9¸EÄ#ÀfÃ+À ÈfÐUÔ VÜ! w°vÔ>Ð >àˆÜ� ‰ Ð/°¨y¸¼cÀ&»k¸]È&ÐQÔRÜ 7°6Ô:Ð:÷i /Ñ
|
|||
|
|
.ú÷ 9Ñ8üò> Ss ¸JÃJÈJ ÊJÊJÚ meta_pathÚstructure_pathc ó‚ — t |dd¬« 5 }t j |« }ddd« j di « }t |dd¬« 5 }t j |« }ddd« j di « }g }|j « D ]g \ } }
|
|||
|
|
|j | i « j d« } | j |
|
|||
|
|
« } t
|
|||
|
|
| | | | j | « g ¬« }
|
|||
|
|
|j |
|
|||
|
|
« Œi t j d t |« › d
|
|||
|
|
�« t d |¬ « S # 1 sw Y �ŒxY w# 1 sw Y ŒÓxY w)
|
|||
|
|
u
|
|||
|
|
åŠ è½½G3SB系统的数æ�®å—å…¸æ ¼å¼�（两个JSON文件）
|
|||
|
|
|
|||
|
|
Args:
|
|||
|
|
meta_path: table_meta.json路径（表注释）
|
|||
|
|
structure_path: table_structure.json路径（表结构）
|
|||
|
|
|
|||
|
|
Returns:
|
|||
|
|
DatabaseSchema对象
|
|||
|
|
r r r Nr r r r! u! [OK] åŠ è½½G3SB Schema完æˆ�: å…±r' ÚG3SB_DBr& )
|
|||
|
|
r* r+ r, r- r4 r6 r r8 r7 r2 r9 r: r )r rK rL r> rF rC Ústructure_dataÚtable_structuresr rG rH r r# Útables r Úload_from_g3sb_formatz"SchemaLoader.load_from_g3sb_formaty s# € ô �)˜S¨7Õ
|
|||
|
|
3°qÜ Ÿ ™ !› ˆI÷ 4à —]‘] 8¨RÓ0ˆ
|
|||
|
|
ô �. #°Õ
|
|||
|
|
8¸AÜ !ŸY™Y q›\ˆN÷ 9à)×-Ñ-¨i¸Ó<Ðð ˆØ)9×)?Ñ)?Ö)AÑ %ˆJ˜
|
|||
|
|
Ø —n‘n Z°Ó4×8Ñ8¸ÓCˆGð ×6Ñ6°}ÓEˆGäØ Ø Ø Ø !× 7Ñ 7¸Ó @Ø ô ˆEð
|
|||
|
|
�M‰M˜%Õ ð *Bô � ‰ Ð7¼¸F» °}ÀFÐKÔLÜ 9°VÔ<Ð<÷9 4Ñ
|
|||
|
|
3ú÷
|
|||
|
|
9Ð
|
|||
|
|
8ús �D(Á
|
|||
|
|
D5Ä(D2Ä5D>r? c óÆ — g }|j dg « D ]X }t |d |d |j d« |j dd« |j dd« ¬ « }|j |« ŒZ g }|j d
|
|||
|
|
g « D ]* }t |d |d |d ¬
|
|||
|
|
« }|j |« Œ, t |d |j d« ||j dg « |¬« S )u 解æž�å�•è¡¨JSONæ•°æ�®r# r" Útyper ÚnullableTÚis_primary_keyF)r" Ú data_typer rU rV r% Ú ref_tableÚ ref_columns)r# rX rY r$ r! )r- r
|
|||
|
|
r7 r r )r r? r# Úcol_dataÚcolr% Úfk_dataÚfks r r< zSchemaLoader._parse_table£ sñ € àˆØ Ÿ™ ¨BÖ/ˆHÜØ ˜fÑ%Ø" 6Ñ*Ø Ÿ ™ YÓ/Ø!Ÿ ™ j°$Ó7Ø 'Ÿ|™|Ð,<¸eÓ Dô ˆCð
|
|||
|
|
�N‰N˜3Õ ð 0ð ˆ Ø—x‘x °Ö3ˆGÜØ Ñ*Ø! +Ñ.Ø # MÑ 2ôˆBð
|
|||
|
|
|
|||
|
|
× Ñ Õ #ð
|
|||
|
|
4ô Ø�f‘ Ø—H‘H˜YÓ'ØØ Ÿ™ .°"Ó5Ø%ô
|