From d12de7ea60c5dc04114f269f964a7a6bc03bb790 Mon Sep 17 00:00:00 2001
From: HuangHai <10402852@qq.com>
Date: Fri, 11 Jul 2025 19:52:49 +0800
Subject: [PATCH] 'commit'

---
 .../Config/__pycache__/Config.cpython-310.pyc | Bin 861 -> 911 bytes
 dsLightRag/Start.py                           |  51 +++++++++---------
 dsLightRag/Util/PostgreSQLUtil.py             |  47 +++++++++++++++-
 .../__pycache__/LightRagUtil.cpython-310.pyc  | Bin 3666 -> 4443 bytes
 .../PostgreSQLUtil.cpython-310.pyc            | Bin 625 -> 1565 bytes
 5 files changed, 71 insertions(+), 27 deletions(-)

diff --git a/dsLightRag/Config/__pycache__/Config.cpython-310.pyc b/dsLightRag/Config/__pycache__/Config.cpython-310.pyc
index 5111a038f20d9558dadced5e2deedb0fa25c5cd4..c5ecd364379714c393d32b9d644a9f5877e85929 100644
GIT binary patch
delta 392
zcmcc1*3ZtH&&$ij00drN3o^1M^2#zsOw{%f6o?W`5r`5>5sVT}5ejC|6yDfk#aJK3
zo|2iIotjr%#baoo2SO%#Mg|s{7NQIcw>S#&i%ZgrQj4RQixSiQG)1D=9DO3=9larR
zuy+)<kB@J>V}NJ8w`*h+4~XOB80;D!8srnj2jcnqySVzq`#Ji$MhUq3I=Q+ai}1So
zMu3E1`uQL#;3jZF_%5ElQ6f;yjuG)8{@$*B@xh)^u2GzRuKp%oK-)Z{_@K;S*C4QM
zlRKGYviTg{UE|$@90NSSPPoMz;2#{~9^@Jv?*U{WaRdB=kh!5iQ*H^u6$dy52Z#Fy
mxgZI-IEFZa+_{pWh#eTxMeLJ5FgY`7Og3RwXJKLFU<3d!pl@yf

delta 342
zcmeBYzsts(&&$ij00ajY6lB;><dtO%nW*i<%M&G-A`m5%A{fk|DYUV}im|?m)6mFV
z&j1JwGmXL+7;do?<>!|~u_q-Kr^crw`DqGAu{rug#yfgL=wR<#+&(_O@s0tW@!qbH
zw|GDtC&ysd_|PDqTYMm%ufL0{PrRR_uj?%VS6?Sr7i1A$hzLx(4<CdNH-Qtvck%SS
zB?8s#7!e=h@9pXrAM6?BdW+M~)!)PmXq)FPJ}5KTH3)3m<e5w|-dw(s!GS*U9{#~0
zQBZb(e-NA<3RE1$2T>E?7#tk#ALJ6n3*oysfnB|lp@<h4az(t86_}kF)h4Gft21#h
F0sx(vW61yj

diff --git a/dsLightRag/Start.py b/dsLightRag/Start.py
index ef423589..71e28f1b 100644
--- a/dsLightRag/Start.py
+++ b/dsLightRag/Start.py
@@ -147,8 +147,8 @@ async def get_tree_data():
     try:
         pg_pool = await init_postgres_pool()
         async with pg_pool.acquire() as conn:
-            async with conn.cursor() as cur:
-                await cur.execute("""
+        # 执行查询
+            rows = await conn.fetch("""
                                   SELECT id,
                                          title,
                                          parent_id,
@@ -158,8 +158,6 @@ async def get_tree_data():
                                   FROM knowledge_points
                                   ORDER BY parent_id, id
                                   """)
-                rows = await cur.fetchall()
-
         # 构建节点映射
         nodes = {}
         for row in rows:
@@ -214,28 +212,29 @@ async def update_knowledge(request: fastapi.Request):
 
         pg_pool = await init_postgres_pool()
         async with pg_pool.acquire() as conn:
-            async with conn.cursor() as cur:
-                if update_type == 'prerequisite':
-                    await cur.execute(
-                        """
-                        UPDATE knowledge_points
-                        SET prerequisite = %s
-                        WHERE id = %s
-                        """,
-                        (json.dumps([{"id": p["id"], "title": p["title"]} for p in knowledge], ensure_ascii=False),
-                         node_id)
-                    )
-                else:  # related knowledge
-                    await cur.execute(
-                        """
-                        UPDATE knowledge_points
-                        SET related = %s
-                        WHERE id = %s
-                        """,
-                        (json.dumps([{"id": p["id"], "title": p["title"]} for p in knowledge], ensure_ascii=False),
-                         node_id)
-                    )
-                await conn.commit()
+            if update_type == 'prerequisite':
+                await conn.execute("""
+                    UPDATE knowledge_points 
+                    SET prerequisite = $1 
+                    WHERE id = $2
+                """,
+                json.dumps(
+                    [{"id": p["id"], "title": p["title"]} for p in knowledge],
+                    ensure_ascii=False
+                ),
+                node_id)
+            else:  # related knowledge
+                await conn.execute("""
+                    UPDATE knowledge_points 
+                    SET related = $1 
+                    WHERE id = $2
+                """,
+                json.dumps(
+                    [{"id": p["id"], "title": p["title"]} for p in knowledge],
+                    ensure_ascii=False
+                ),
+                node_id)
+
 
         return {"code": 0, "msg": "更新成功"}
     except Exception as e:
diff --git a/dsLightRag/Util/PostgreSQLUtil.py b/dsLightRag/Util/PostgreSQLUtil.py
index c6ccf004..a908d064 100644
--- a/dsLightRag/Util/PostgreSQLUtil.py
+++ b/dsLightRag/Util/PostgreSQLUtil.py
@@ -19,4 +19,49 @@ POSTGRES_CONFIG = {
 
 # 初始化 PostgreSQL 连接池
 async def init_postgres_pool():
-    return await asyncpg.create_pool(**POSTGRES_CONFIG)
\ No newline at end of file
+    return await asyncpg.create_pool(**POSTGRES_CONFIG)
+
+
+# 查询示例
+async def query_example(pool):
+    async with pool.acquire() as conn:
+        # 执行查询
+        rows = await conn.fetch('SELECT * FROM your_table LIMIT 10')
+        # 处理结果
+        for row in rows:
+            print(dict(row))
+        return rows
+
+
+# 插入示例
+async def insert_example(pool, data):
+    async with pool.acquire() as conn:
+        # 执行插入
+        stmt = """
+        INSERT INTO your_table (column1, column2) 
+        VALUES ($1, $2)
+        """
+        # 使用参数化查询防止SQL注入
+        inserted_id = await conn.fetchval(stmt, data['value1'], data['value2'])
+        print(f"插入成功！")
+        return inserted_id
+
+
+# 使用示例
+async def main():
+    pool = await init_postgres_pool()
+
+    # 查询示例
+    await query_example(pool)
+
+    # 插入示例
+    #data_to_insert = {'value1': 'test1', 'value2': 'test2'}
+    #await insert_example(pool, data_to_insert)
+
+    await pool.close()
+
+
+if __name__ == '__main__':
+    import asyncio
+
+    asyncio.run(main())
\ No newline at end of file
diff --git a/dsLightRag/Util/__pycache__/LightRagUtil.cpython-310.pyc b/dsLightRag/Util/__pycache__/LightRagUtil.cpython-310.pyc
index b14b6b8116e173db7a22c0431e138b25aec147e7..253fd21379a22b8622357a9789683c2c1815298c 100644
GIT binary patch
delta 1010
zcmZuvO=}ZD7@pZ|(qyyQHcgAEO+Rc866i(nAf*SlByK;*2idk_MV8IZG)bH6hRHN-
zrJ73h;Gu`GAP64xpq{0FKoC4Cc<`|P0>OhIp2Uxt)zk{k%RW5MGhgrf?z~HVm{Ocr
zOag2yeQj3D!(AmMh_E}e2PeAE&cn&>n`9o3Vqr6d&ddQk=CdSY2|RuXeKo?=gwM`0
zHtDlb#?JXHH3!fb8ovwHrx4UZ`rM08ISA$BS>dG!2amsnYzUX#&G378%~c~`WG|A<
z&1W-4W-;fvmn8(t?o%lZEACtAIoxq~qMiOW%eEylP@otkRW~R<fcM=;GKKftcr4M`
zWK*7^YFR@osl^2&ujaB|tXR-jv{F_#78p4ti-po@xvFPNUi@TnQPuUUg;M5}l~GsJ
zc~#F^kloW80wW6hUIEYnttH0pKnnH&g;YQg2`E4VRHQ)~LgG;X1yKm~p)iUd=}5i>
z_u=EdCtwjEumB*S;lmTY&wCclf8^!XzySa&<OZW&w5Tnw>C_=+1$%=<E#p`^HR*Os
zr=8!XQq(kJHmkiw{Y7mJTWrT4<4Rg4COe9(^7E8>y<s#QgzJWFHn5lYs~hI7VYc>c
z%S--iQfCX>Myqxcd(+r9%XMt<1s%(@s@T|Yh=FZdqk9J4rNp!-xz2Brw%SIkmpF1P
zCV1XGkCgp%+J34%(;GFnsbe%uo0Z!$`4*GnRysm>_x(CZf%l(;xVAVT4*Xt%IRqG_
zh4m==rOS(XjlHrG+29icClF`Mi`VQLt(kRJo6)QotjiCX-I`$_90>4#A$}KvzCXgI
zL!Xs|3`K$5W;@bCKA(p9^t5{+_Ij?u98%@L-znK*;Pqj<T_cW7>fCB@usJv!njH8!
zWWepn!){8EFI{9Z48o9y%whlzaUKYVgjw;YAV@!wiWCVCh5MzL1QjUxAq&3%PvHm&

delta 212
zcmcbubV-IUpO=@50SFE*D9BJ0oX97`w1;7$P9>xG#AWJ?5);o!u%$2tGiXX~HevK-
zWaQi2#B_j#QDn0-dmbaB@8nG!*BGNGcXEE_iQ;tib#iryck%R%n%u#a!k9AoEte)^
z>SS^5<&2e+_jA|tR{*UkW&sj>j4a$mX_I|<rZIL;e!^3t*bb5bIfIE21lbtb{uX5b
yxtdIVewvJan$nX5j&Y{~1&Y!@M8@R50&09}KqdzR2O|(N@i0v`6qI7(;RgUFH!-^a

diff --git a/dsLightRag/Util/__pycache__/PostgreSQLUtil.cpython-310.pyc b/dsLightRag/Util/__pycache__/PostgreSQLUtil.cpython-310.pyc
index acaa3d2b7b8af3b1e0aad071f06b409e72e3d65b..aada6b85691079cb14517057f6b2b6d02d299487 100644
GIT binary patch
literal 1565
zcmah}&2Jk;6rY)0I}?AVQHv<5&{ir^wn8BZ^nxlxNZi(vYX`4G#42dD>`aqw_rsYT
zlP0pIL<R|Q0K^d?kw|dhf8fTAKVh$^m2%<+s1g!y){f)02WB<zb9VN<-+OQ4a=CzD
z{r*>XcRr8MA4-{Cv!J{WJBffWWFv+d39yZUYXRvXu5q2`c%IKNo#j}b&HO-Z$|#%t
z9NDw1z>2^Nti;N|i>$(`z)QT$<{lAoE4!^?S9s;5%1c9ZiZNou%C>pw@JgZUcMU&E
zq#Fc=n;b-5cc*X$p`fOTQ+)FbgLrmXk=sB_n(Di8B2&E^3!uG(i<EZVB-xJzgRorb
ze(EONrlB7>iGRpb8oG}&$*y{F7`hR2q#yFQCq20Exut{l8|>s&Ajepuhk$~QB>4${
zjq8BUJ!B!1G=D*+md=c$wTqtMuH;TPj)Szi(Xu*s+cnEsYc=oG?}{Q^bwQrpT7P$&
zC0nt01Z+dZ54k5nYWO?5vhD6{Z%RMd-hd2u1h+nHD0Q)Wkk0#&FX0QBJ;^?GMFpxN
zCP%MeQ3b(MSe-BBQMJwT2|zf;eRP6O(K~3A$@j(p_wfm2cskB)K>B1IeL@GCBvR{Z
zLp+3P?OpDZEkq!D9iE{!dK&{0%j27pxAx(jdOn8DMi?Dlw`z^rTF1C)+-bM&8wYVu
zI8dB`8;$z?ddFB^GIapa+w1v)r@04Qdb??^EBpxR%6w0P8>$SL>Rud0sV?IEB-Mak
zz$q$_SX$oeadF`A$8OjS_-kNFl${*S<2gb}1rxmVOsJZA1jzZyfiOWxb^l8UT^|sJ
z2U?$;;J(&}xwwjeYFE+e7=?i@3DXryef@thT6zhiT=TF~FvhFiv})~+QEzry(@-sV
zanK8+<u{FyuwojMS3X*8Y}PDe;Rbkbte6wKo&U@YdVDz@@s*zHji=vy_vDM8pMLr6
zldpdK^Vefj6LVm-X$Cu-7xSQ+IiV&E;D<31s=-n{k)gDUFri#X&K#e;h<;_{JBR-b
zcxwW`POjysg8y%E4eZYioN888j;ui!K~MuX9JkZ}Nod#MxG@1<bV5$YzHy7;CM2O!
z%PO1ZydX}viL(q|j8P9=KYAO^CKL!cD&dMYpBJx#l2XS}UdJ)#(iz8LvFA9Vq&(%d
zIC|jkERI-Oo{aWAnDHrfqt%{PH=&a0rODxqRm<9Hwb!Ri>#LpB4_2+3D8pwia>y{6
z1Qod<cbxHW#*d-%d(l*0*?%Q#e|kHNSuf!4DMXV6AUdKNB}69x{l%02hn8rL7HNf+
JDNKg`?|&}-Zj=B3

delta 79
zcmbQs^O1!wpO=@50SKfn6=d9GoX97`Xf{#XjFl;vL6d9aJU>PzO|Hqhtksjxvnt6+
W0oAcEa4>T)F|jZ)B15(xJUjrK#|>=&