From 6e3ea51aaa2ae3cbb42d326a2daebba5a1e0d2bb Mon Sep 17 00:00:00 2001
From: Amber <amber@wubook.net>
Date: Tue, 23 Jan 2024 09:33:24 +0100
Subject: [PATCH] tree find opt: dont_show_children_nodes

---
 src/__pycache__/tree_repr.cpython-39.pyc | Bin 5563 -> 5685 bytes
 src/tree_repr.py                         |  13 ++++++++++---
 2 files changed, 10 insertions(+), 3 deletions(-)

diff --git a/src/__pycache__/tree_repr.cpython-39.pyc b/src/__pycache__/tree_repr.cpython-39.pyc
index 524222bce526db1431f1c399101e0f60326b0f50..cff51589a09ce34d74a4533501b7a7fd63edfc42 100644
GIT binary patch
delta 726
zcmYjO&ubG=5T4n$yV-1hwMmOfl{PjJiLD+4K}11n=|Q0sDy259Me<@BEQxu$S}F4g
zxB)@%Q1;-(LwgfEh@dAyJWKrtdh%|M{sGQQ6n!xB=EwKGo!Rg6%%|DxgG|N;J~yAe
zZ<bH*XV2i#{>;0{7Cb~bCChzWE6KBRx({m;gIYya`}pDHB-az>*`G7Yx65{mEpQPa
z1Omdy(t@l5lz51$D6O_7g|<lq)fOxR+3FgJdDuv6XA8)Y_I2pawm`ds>KC1$gmg-v
zz0e{rLPow_w{%kb<R6IMA?OtO(q%xJ(lYVLQZLC>)Lo&JcA3x}5U!9%N$C}VahcF2
zmohS0g2-a3tj<w3_!v6UkMKD%8+T+%ru#s-d5G{Ss3Yb}hOQaFu<%!Y3>LX3^azav
zzbJQ*pwuBEFwn$_VceY~cNuD#J>i$j$3@v`hnvB6XLqx)-P)0=+1_k-WHY$Cpl0}|
zsfuyq#@&d2j_%?}+{j-_);x8`)F$HB`ATI}oioW4yJNfaI1JQnlP{UvRn6eZP8fg7
zcMDg!0K&KOmW?)>hkm?JxPbMzTd3n|{Jn6#H^CwWMl#^GhiZeP#{ZH!Trf6}^fCuS
zNy(D<O0cXVm<&-`5ic%TOqNJQsO=ER{#&bw#Tu#=e!J%{cFMyEgPKrvsPf!%+)!88
x)trBe4d80kwASMnW7DUm%%Ew`0kdd!F{bXvU&r30>IP;e=X`5?$xdJNoZkRVo%jF%

delta 604
zcmY*WziSjh6n<}J_Q&n+opP7e+#%%5`GfVsVIu+2MuUYy5R*Vmv`{BVP?Vcp6~Y@1
z-0QRn+gc=Gp;)MeAP82L{s)3!V`XV)<C{|<&hWl@^WN+?-+cRS?nA$Eqh61I$FC<Z
zH?DNH8;9`IcK_7@J)^s0SZnLUdY0FY?&%|Ac>D9mzL9Xp|BeQ)KF4{E{3}BN1carq
zf<)fjgeBMnD@iQC^^aDO(%1&;6-Zu*Tzhd%Sl<Q|?1){kAq84+zeY7J?+9~~q*)#g
zD75l6kksl73<o3f$k*O9@WH5%Bq`E<8%n`cu}vt>pB737B|b)`<Bkq>Z48twK#32*
zJZD0`@SFh0!q>dl(LJGKTc>*dnUWo!!d*nZYLcrz`LIKB39@7>l~K2Adi++CIymc~
zU(GikVXb=I9Qex@vcO!bzBhaO8|J9%&N#TpP|dZ5@kvfVMyOC#Z(1jCwffRp$CYY+
z>iDRI?4vUjFyv|0IJ>piaOJ+`m>53b<UhkZ1!^o=yJq7H{skpk$!5!cwM;B-MT_^l
zW+Q)ax5zVTR++<`b8re^3Np1v@`@{8tuUQEIPHWbya#m3!%3L+>U8=no=`ZYy#JP7
JR^hSG`wd}>dno__

diff --git a/src/tree_repr.py b/src/tree_repr.py
index 7ee1d90..4fa04b8 100644
--- a/src/tree_repr.py
+++ b/src/tree_repr.py
@@ -234,13 +234,20 @@ def find_children(tree, item_abs_path):
 
   return (False, [])
 
-def find_tree_struct(startpath, item_name):
+def find_tree_struct(startpath, item_name, opts={}):
   '''
   '''
   results = find_node(startpath, item_name)
+
+  dont_show_children_nodes = True if opts.get('dont_show_children_nodes') else False
+
   if not results:
     return {}
 
+  if dont_show_children_nodes: 
+    for result in results:
+      result['children'] = []
+
   tree_struct = {}
 
   for node in results:
@@ -277,8 +284,8 @@ def nerd_tree(startpath):
   tree_txt = nerd_tree_from_struct(tree_struct)
   print(tree_txt)
 
-def find_nerd_tree(startpath, item_name):
-  tree_struct = find_tree_struct(startpath, item_name)
+def find_nerd_tree(startpath, item_name, opts={}):
+  tree_struct = find_tree_struct(startpath, item_name, opts=opts)
   if not tree_struct:
     print('No results')
     return