From 60ccd874dfd4c6889c713d2cd97b623d749116a7 Mon Sep 17 00:00:00 2001
From: Amber <amber@wubook.net>
Date: Mon, 29 Jan 2024 14:07:09 +0100
Subject: [PATCH] vfind command

---
 src/__pycache__/cmdline_parser.cpython-39.pyc | Bin 0 -> 655 bytes
 src/__pycache__/tree.cpython-39.pyc           | Bin 4237 -> 4260 bytes
 src/__pycache__/tree_find.cpython-39.pyc      | Bin 3802 -> 3836 bytes
 src/__pycache__/tree_repr.cpython-39.pyc      | Bin 1213 -> 1099 bytes
 src/cmdline_parser.py                         |  24 ++++++++++++++++++
 src/tree.py                                   |   4 +--
 src/tree_find.py                              |   9 ++++---
 src/tree_repr.py                              |   2 +-
 vfind.py                                      |  18 +++++++++++++
 9 files changed, 50 insertions(+), 7 deletions(-)
 create mode 100644 src/__pycache__/cmdline_parser.cpython-39.pyc
 create mode 100644 src/cmdline_parser.py
 create mode 100755 vfind.py

diff --git a/src/__pycache__/cmdline_parser.cpython-39.pyc b/src/__pycache__/cmdline_parser.cpython-39.pyc
new file mode 100644
index 0000000000000000000000000000000000000000..7cfbe69c5fb81dad00e05c5bcc165fc5bab09941
GIT binary patch
literal 655
zcmYjOy>8nu5GM6!S+-LzMYn>T3Kwv_2Sw2gU4sDW#KmBBXN#^x2_)qt8al*to}dU2
zk9~r^i`GtggQ7rzl1hgj!Q&mj<9GLgVm8Z&vtRc=@DLF4(<lF1?2->|n_K6A04)jk
zXHe2A-~lB_b|HiiK@7=V$Rms%Vi+B%LwuAzMY|Zrkii6|cj15}N5V%q+NFGq0gexp
zr<guuM`FI>1ZFt?M&RT=Jb0cdJUh@^3iOgJ=8tX;a=xH_bRlfP<NZb|=+4jBUYb^v
zY>(Q=TCsY=S|zWUhgn`%m2jt2?5dD?!EAv{qiIWPn9#^fQC}$!3uZN9_u^9_)EC1x
zwPvl6YRja3#ZDIsr?AoWmQMgpu4Q9gFP>l{%X-^KjT@;SId4&`E?AoG`Eu#Gmw6#e
z(5RNGb}xBmY^`x^b&F0)B682uC0^1g>t+mUWnVMAmd2V#>J%?#JQaG|2yJlaP?Yks
zc0We=$(i1^6)Jn-LCv$gg7r|S`BVU`-Mazj*>LM~)2F`5M>=#QJ#G5rsa+p8S}NOT
zmp8^@_3>KT??lgB(cpf+Srv7KtFq0-N}&cP0&!(@zH*GERQRv=+vcWE-&HlVCBF9(
c%uDA|It?Nk`1L1AGMWYRFbxu#1}UA>ziy<z!vFvP

literal 0
HcmV?d00001

diff --git a/src/__pycache__/tree.cpython-39.pyc b/src/__pycache__/tree.cpython-39.pyc
index 12946417c80691c06441d0b1cbeb495c70545254..4571cd0d573b22f9466d5904a0950808519ffb44 100644
GIT binary patch
delta 143
zcmeBGT%yRE$ji&c00fTnwx@oW$Scd}Hc{J9B$qRai;=;dA%!`GrG+7drI{&;dt*)q
zkI*gt<oulcqWHA@qTIxilGLJGli7F;8GlW-;9bIKGx;uW8l%Hx7rqZb(v*KCki5_D
pz{67{47A}ES8-9Y-eh$FRX$!2tE4D3HNGgdplGtUfCCc`69AQuDpvpi

delta 120
zcmZ3Y*sI8!$ji&c00c*-ZccqMkyn<{X`;5FKoqAtLke>WOAA8^OEXgx*T#Yl9-%1y
z<oulcqWHA@qTIxilGLK8$y~gKjJGD+@GfDrn*5MAjnQth2j2%EY01A5NIvFw;NUJ2
U2AUqlHd$Xlb#k(RJrf5L05MP@`v3p{

diff --git a/src/__pycache__/tree_find.cpython-39.pyc b/src/__pycache__/tree_find.cpython-39.pyc
index aca21788ed7cde9347630a7e538f2224282de578..81280eeb91a2db2d3fc52c395e02a0326a057c5b 100644
GIT binary patch
delta 1232
zcmZWp&2QX96rUN7y<_j%X-TqlA)7QMEww<~qC)sk6p$#9xCB*29IS#_b|$UUtameZ
zD1@<)8YCo6twsom(}o)tP}Kt*IdI{GIO6}H2QIzBfqAp#qp78N`n@;r{pQVkZ(Mi(
z>W2L=43PQ0_~r-k)$Q=d_WMQ=D8U>hgyY1?P=>b1o!>|JWeScDKESS7ubWjoYXrv%
zrRXX$5@#TZa*T9Jy6x&u$RhUi*JNd(_P4T}r=xLMNM^!nmZn*rm1!DN{VQqv@|dxH
zi{z}nN1mBKZj`23g-sHB*drdk`?Rx;AAby@X+Gj2@j<48(fDK8)`~9A&6+)F#T=ln
z5wrRZ-CqTi_}j%OPfIC;jEo**30zbEe}f(*3d}2a`)-9_X&a}vH)0|e45&kUkHNKU
zhf9&i4a<>GU`@iT6vMRtL6MI8<u#MjiAMKmPNxjXS^Z|?9eiF-8uM2;*~9XbLfe?*
z4$294Dxz@F%g|+{h<s1si|E}mm7|;q;q(xXH?E?JmX4waSUKEXgW5ZsuA$x43{|f6
z|GDy%$2}dgrB53jR5ewTv+G2Dr(8MN-B&^7nkd&s?W=}j+&_VgUrV8hr(D(IK?fa@
zE2Ii5s=}f!8@zddZ(`-c4sY@BGo<K6bbvoZ^1fkYCZomy2A=EW7M6@h_Ab#G72U$c
zqM{R~f<ugvaR%&7@$Qy#FQa%y|HhWjZf-~@w(09X^{uI`<h-%&lis;|*Z{e#JN|3E
zWq{DOnAD_qzQ%*2oeb7?vZ5S}hU4{t{@j06oP-dVvuP`x9g)|*$oGaq_Ddl<umk<6
z{Wm@uh;f;Xa`~ju<C*ECz|_do;NVG7?Cq4(bequpv?%4?pp=XHxnQ;T@F3R5?9lwr
zZQYO*Yepc4kwyJwaQo7dQDQnx&mI&SgPZ}YGaa%p7dKKnVtIck5(cV$P)AO4O>&|f
z>Kn~Jn)bHO1JeaPYIUEo6K|PYV0<`IzF=&H{=W6+5+J$ys7p=ZDmV2n;pR*5NbMVw
zCjmj_<OLA+Jj?JRsQ(>e(VTuOnqP<*JI*L;vvbUgW~6T%H{&1)3OVR$6|J037(n}4
O;fG;89;Nyzdg5=Xyb%Tf

delta 1267
zcmZWo&u`pB6rN{~y|#C~ZIVsZq->hdD%H{|AXGwymH?&nz@bQhkgzCLvg2)?ZoHeZ
z6Hpl|KsazgN;MJ^5{E1Yq;jYPz4Q;@fW)1HxpCvbk2@TAZ)j1wSeob0GvE8>`Th92
zUA4pYFbo(zZ@0cn-ntduYrTsf35!6m$Py6QDeJ?xKM&z`n`rkR!Va#NaplWs1Sl4>
zYYZ)<WUa#9<ccX!masmvfeH2o>)2N+b{i@_92I%T(O+4o&q@#7D=&`IEG^==V^=&+
z2K~s-i*;Fy){E=1fsETsPHV8w&K$*$I<B1j7`WhCcVDvC0H*YZd=_T)=X|L(jYra$
z8A6Ytnbtq@o2!H(e={FuaUqjLHu1Xya*9}wI#qr@%|x6HMRMxy2Tlolvq*aLa>tS<
z5LBT#?p;p@LMB<KW;7~Js!^C0$yU6+k;f)}a;zLft(^XFClit1f8aTBtz1I#8C>~s
zG|X*403I}~Cg}HS7x&Hq!|WciF)K0Sn{bFV8iQOQEMSghnC;S1kUx=PzLg~-1((>>
z<<eF*a_upUE?#3LFKos4LD_<LscN>cyKMXEl$DO*e_y)F70w(hYXw)eebTt|j#cWe
z*W7b)yRPb@cEgh2C`W#|y{m%KK^e!$?5djbgm;`FuYtk3%25sB&#|$E5@E?znEzDN
zM6eGZgYxM_)J5YHruZec54%kMj*1?Jh!>O24eJ0T7oqvN=CW!YKt8Ydp{If|07Eii
z_g}odp&abArvAfQI&)c4?jt{YknM__inMU|rnU0qub56I7xe}I#g#>Z)B_!>vR0m|
zpf^#PD(DW<yyy+LMr#}TYyX9*lgPf`C)CmMg#OcSwW)BXjFtb&`@LjTq{B?klkgEa
zrI&(ZGbX4ywMv{9a;I0wwth8OU76^>qo&0kqCJlyc^3_{J!@Xy32rSckS>?!@MW4=
zhJ=pLu8p@{t+$U&s5&yqlVYF`>c7>E+cV^GUVqwXpE7e`6lmrq{$H-rp8lcn*GYq(
z{XeDR6!J&<m+;DKG^l3sB&EqDM4l&P?lTG(9#t@bnt}E4faA((%LDE<YI?6Z3$Fg6
jdHOl|IQh~1OCmlL=6?#6*WDV5ok6lj70yX(5vS-M7o8CS

diff --git a/src/__pycache__/tree_repr.cpython-39.pyc b/src/__pycache__/tree_repr.cpython-39.pyc
index dea2951389da53fc96ba286624d65ae948a80226..c2fe51429b3d7a2178cfc6ab3c29cd4624fcaf76 100644
GIT binary patch
delta 329
zcmdnXd76Vak(ZZ?0SHp(ZBN}ikykdt4aiAhNMX!jh+;@#iegM*PGuBfh+;}%iDFJ+
zjbce*i(*Y-k77&VNN0><PvK1AYGH`tNa0T50n(f)yeWJ@nk$7dm_d{OCCFsImkWU8
zuZd?iFmg_=WprcYo_vJSM}Pw;Si}h@xF#zwDKhd-c4B(Z#tjnXoovqRD}IYJvm`Y)
zJ})sh6-DuEW^)xupfXMFTTJ=IQDUyX0U?p`AwjOL@jjk@K*H6}J;b9(V6qI0eF6_q
z;Vl8r5Le&e_z?ei*9d2yP#4!Ch&3R)Kqes>4|WwORBmzD<mRW8=A_y&0(r$EK!SsX
Tg^`1ag^7g`gxQ!lSUA`KM5;sq

delta 444
zcmY+8u}T9$5QcB}?lzm4OD;xYr6^XZEo?;5xW>pO#^i!#nnRsagsg#!Ha3Fz0^wf3
z2e43kpTOQ?-ynt9TDq$sy0G8R4D-+YulyzJ9eG~P(yv;6j-STf+|8*j&_Xh>z$hIM
zNr6fmZ0Ud_U2r7>CNq#B7TUnkiTR8LS#%+X45lyxdF0?>_KZLQb2xQpK@s!lY4x#y
zMXe<yU90L(%;i><DZ3-K6zhv1SlT;KZoMN!OF6avhUR9_6N<HaopvB5_u8Pww~wTl
zE|YCiO0P-a7j!cz{=!e+$Ra7HpX7s;Ot+lA)8;H>&o5&cT^z~y-|)V@vOBNGs?&-N
zM~a;c2g78PP~)@>90S*Y8Dun|BAV^6AB8$-6f{~IqE!!fl1lnuuOCnA5#={R(d<Uy
xUL+1{L2pYWKVi%cbMH@x@9e%Y@l`p*t3kX*bTz7IaJoPp!l|QaKBL^>><d!tU%UVS

diff --git a/src/cmdline_parser.py b/src/cmdline_parser.py
new file mode 100644
index 0000000..c2384eb
--- /dev/null
+++ b/src/cmdline_parser.py
@@ -0,0 +1,24 @@
+import argparse
+from pathlib import Path
+
+cmd_parser = argparse.ArgumentParser(
+    prog='vfind',
+    description='A visual version of unix find command in which the results are shown in a tree',
+    epilog="Thanks for using it! :)",
+    )
+
+cmd_parser.add_argument("path")
+cmd_parser.add_argument("pattern")
+# if -s is found in the command line cmd_args.show_children_nodes is True
+cmd_parser.add_argument("-s", "--show-children-nodes", action="store_true")
+
+# start the command line parsing
+cmd_args = cmd_parser.parse_args()
+
+p = cmd_args.path
+path = Path(p)
+
+if not path.exists():
+  print(f'Path: {p} dont\' exists')
+  raise SystemExit(1)
+
diff --git a/src/tree.py b/src/tree.py
index 51bde54..aa02caa 100644
--- a/src/tree.py
+++ b/src/tree.py
@@ -1,7 +1,7 @@
 import os
-from colors import RED, YELLOW, GREEN, CYAN , BLUE, PURPLE
+from src.colors import RED, YELLOW, GREEN, CYAN , BLUE, PURPLE
 
-import tree_repr as _tree_repr
+import src.tree_repr as _tree_repr
 
 class NerdTree():
   '''
diff --git a/src/tree_find.py b/src/tree_find.py
index 167be25..3e31048 100644
--- a/src/tree_find.py
+++ b/src/tree_find.py
@@ -1,5 +1,5 @@
-from colors import RED, YELLOW, GREEN, CYAN , BLUE, PURPLE
-from tree import NerdTree
+from src.colors import RED, YELLOW, GREEN, CYAN , BLUE, PURPLE
+from src.tree import NerdTree
 
 class NerdTreeFind(NerdTree):
   '''
@@ -8,11 +8,12 @@ class NerdTreeFind(NerdTree):
     @param          opts                   dict: the opts for the tree representation
   '''
 
-  def __init__(self, startpath, opts={}):
+  def __init__(self, startpath, opts={}, find_opts={}):
     self.item_name = ''
     self.results = []
     ## computed tree for find
     self.json_tree_find = {}
+    self.find_opts = find_opts
     NerdTree.__init__(self, startpath, opts=opts)
 
   def find_node_recursively(self, node):
@@ -63,7 +64,7 @@ class NerdTreeFind(NerdTree):
     '''
     results = self.find_node()
 
-    dont_show_children_nodes = self.find_opts['dont_show_children_nodes'] if self.find_opts.get('dont_show_children_nodes') else False
+    dont_show_children_nodes = self.find_opts['dont_show_children_nodes'] if self.find_opts.get('dont_show_children_nodes') is not None else False
 
     if not results:
       return {}
diff --git a/src/tree_repr.py b/src/tree_repr.py
index e7253f5..94877f8 100644
--- a/src/tree_repr.py
+++ b/src/tree_repr.py
@@ -1,6 +1,6 @@
 import os
 
-from colors import RED, YELLOW, GREEN, CYAN , BLUE, PURPLE
+# from colors import RED, YELLOW, GREEN, CYAN , BLUE, PURPLE
 
 EMPTY_TREE_LINE_LENGTH = 160
 EMPTY_TREE_LINE = ' ' * EMPTY_TREE_LINE_LENGTH
diff --git a/vfind.py b/vfind.py
new file mode 100755
index 0000000..e6ac9de
--- /dev/null
+++ b/vfind.py
@@ -0,0 +1,18 @@
+#!/usr/bin/env python
+
+from src.cmdline_parser import cmd_args
+from src.tree_find import NerdTreeFind
+
+if __name__ == '__main__':
+  print(cmd_args.path)
+  print(cmd_args.show_children_nodes)
+  path = cmd_args.path
+  pattern = cmd_args.pattern
+  show_children_nodes = cmd_args.show_children_nodes
+
+  find_opts = {
+      'dont_show_children_nodes': not show_children_nodes
+      }
+
+  ntree = NerdTreeFind(path)
+  ntree.find(pattern, opts=find_opts)