From b0779c5b07e74f22c84a497a7927e4a551e14b15 Mon Sep 17 00:00:00 2001
From: wangchuanli <wangchuanli_@hotmail.com>
Date: Wed, 2 Jul 2025 16:22:31 +0800
Subject: [PATCH] =?UTF-8?q?=E6=B7=BB=E5=8A=A0=E6=96=87=E4=BB=B6=E5=8A=A0?=
 =?UTF-8?q?=E5=AF=86=E5=92=8C=E8=A7=A3=E5=AF=86=E5=8A=9F=E8=83=BD=EF=BC=8C?=
 =?UTF-8?q?=E6=94=AF=E6=8C=81=E5=8E=8B=E7=BC=A9=E5=92=8C=E5=88=86=E5=8D=B7?=
 =?UTF-8?q?=E5=AD=98=E5=82=A8?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .gitignore          |   1 +
 data/photoprism.png | Bin 0 -> 5885 bytes
 main.py             | 142 ++++++++++++++++++++++++++++++++++++++++++++
 3 files changed, 143 insertions(+)
 create mode 100644 .gitignore
 create mode 100644 data/photoprism.png
 create mode 100644 main.py

diff --git a/.gitignore b/.gitignore
new file mode 100644
index 0000000..1d17dae
--- /dev/null
+++ b/.gitignore
@@ -0,0 +1 @@
+.venv
diff --git a/data/photoprism.png b/data/photoprism.png
new file mode 100644
index 0000000000000000000000000000000000000000..a33d9133409c3a7c6f2d5e1119b1d4dd04ef1bba
GIT binary patch
literal 5885
zcmV<Z76R#sP)<h;3K|Lk000e1NJLTq002M$002M;1^@s6s%dfF00001b5ch_0Itp)
z=>Px#L}ge>W=%~1DgXcg2mk?xX#fNO00031000^Q000001E2u_0{{R30RRC20H6W@
z1ONa40RR91K%fHv1ONa40RR91KmY&$07g+lumAuSXGugsRCoc+TM4{P<=%ewK4+hi
zgCjbro0N)<9LZEtu1Llvj>@Hpq$HHOH%%%Hlr*W3u1Z3EH?Ap3nhcToq$^F|*Pw(P
z&OEI9Jnvp>dG~hqafp83%i8O|-r+wy@BjV3!&)bRKPk<c<-|42Kzd>V>XeqEwo`%{
z<qi@hJ{UkzAb>sbLF`NbcC6cuUD?@zvOhz$z|j)2W{nd{E=Oi@Iobvsv~ftI94Am-
zO&T1E3mU`0GL)C|qPe_E5Ww$L>Ide35f{J;Cy3?v1v#17f#Rc~>k$WF?J6gWOncL=
z?ld?vDFN}0jHH2Oya)t;_FqN;Ju`_WE)X~dAku|JD5ZDg1RczC%J5d^MS=CqK4R&J
z0N^-IVEr0&DG%Ug_FPB0d17e^pvyOwq<Y&V-XmybdZ3WB%LGV#T=*&}MiJ{RDRc0`
z+n-|L0}liQ=O0Rc7=YF5oc8RUd&uZQDOy|<?q$Q+HOi-{s%=~ZEi+<(;MEhK#8B9i
zIBb?WrFgi>#eomh^l+2W?Qj}@@`+O~DHWrs^mU<l#1+wS8Q>dh(=0MZ$Uu)4p*n7!
zG2SaXNIf5jQj~E>{j(6bulYrRpTzP|(_sOyVy)AYy*+`w(V&phHLh(#{wxqACyDb&
zX)(SnE5|U#hq*@;08M2lo`(Avphhv0$CYr5M;LDP=AQ+EJSW7V$9q{>DaPdF;=x0i
zLaXcn8#Xwpg#o;jkchs;Zm$~`7UZ_!Htm%$&j*Bg-f#$UJS)fv$Y?K8Q?H&+ix960
zWN@ZH68Ui`POl9N%IF-(Q}u(TDgp4`8mC6}WK8Fj)~z6)EMq8PgvBaTzA8w0UI|oo
z?!0LzDA<QfyAMQCLV^LpHcr(K1qAU;NirZT$PeJ^7M%in#G|rwPyjY;cT)3rViw0*
zR~c_B7(q*lj8o#h!ZPKvSjMm4unKpL>cM%w0IkmMghAImh(@P2Q4fZ_!R4dd;ef#P
z!-7%=*;xnigh~P6{2sUb3rtH+Lf=C5yi&Arw$W~Rh4tJr#tsw}=3&I0UAY9-00`he
zTR%t5nssnhuMxP6-Vh%b0>`uOF-A$rab-n-TakR`$rB0+@J41X`UfgmR!f7J>C+tC
zmz0dYWCX{9iuahbI9c^<+UUu$5EuJz%$$Jlzu$rpw>^)+10O?jay1ZaygK7C+&iW(
zcJ2NJK?OD?$#d@>zk|wWn|yjwax$*YNyB}HPmG_K0DQc~=^mGeQD%O3jfyGQEss!^
z0)53@17!UA4WD85TTkK2J|l7Z>1Uu>vsUQQZGchlw1#Kl<K=TO`<;mfs0ev3k1B(}
z`UTuQ#D{SArgxq0MnDBWp6U@rpMC3OaM`(%(_MYDoMq{ujzclDU6mUf6w>wYxV<al
z;^Ohd*m;Zv$;O_-f;`59%kb5<^{89t1O~$*JUi}v)Ty0e_7656pRYkTQifg67{)r9
zBDqHR9>r*r*)6bL?d?>d2e8-2BqpJL3CCds;x1efJ>x;hyWitho+nJbKXc~O`0l$c
zxOwn6q@*N++hNltE<PS1Rwybiz?n_jqO`08i<izoJlhzrisEPFyLkuGrGy(W?oH~K
z$6<`8i+w);fDgAi=Q9TNXA}(!S8ZFEO}k|gN_h2IPk^Y5YX`diYRsBR*T3oxG-=u#
zMcOE_Ou)Q8;|YBI%|@Je?j=~YdJ&GPU5@}v$Ijh*;Ks>_xFZpV0z#pNLOib7xY)Ts
zZS52RfTBKi8Xb&Byn4qZ*O0>nAxH`P^78S*^CNLe!=~t+eXE)8#g9`zR<HR4GiQxQ
z?<<DkzPl&m*t#d;hab1$=k4F&qfh6UI=&uk<I%cT6$F{WQ5VO(;{y&?vxbNJ69K@7
zU*jSMjrPSldXO+npuxj^uBu?kP!G>~ldk{m7Z`lYSR|(=a|c6(sK&0}cjMXTh9fhx
z4SMyu1xXwhg9bi=0GD;T7A~Gf@aT%3mXvQ~i}4}qB9Egk7krV9PHOHYkF}<u0DX2S
zd*4sCp%NGYf@;}HxX-qNA-ev?4Qnua#&~33bvw??I@|C400Edf^+D|1xgA4>Jc%T(
zv^4CRHEWH_F1eY(ugnaKm1`CoFR=OAw8Ywa+GP<m>v3ugaLGN?*s6w1J^&y6fF|sf
z4(j~~LuI6EXaoq+fY^0N%k}f}^YP-dBhlcLGtv91TT!evKg$?kv258~ESNtP0|q{X
zQ%`M#VmIJwUyz+W6pb6TBv{3mzvvaVE2BrahDz%wT3JR%<|{=Q?&hi7jXM{9xdctZ
zH7x*m8rLTw83~^8-rZ+tG4jip=hIPw6~Sk}@eIEGdNXbsIu<FZDJJSO7~#hse>5?n
zZM#d*t$Tk)e>Qe`DfU#WmWqJ`9_GCUHf{bC>o=}2L&?`ggzCr8DONq{O7Ig(%h5-5
zl#B<=%5h|PPfdHiK@}Ni2{f(sTtPhrw@n*9$L#4(qIX}qezO*)_tlH#^ud=V{+-PN
zxasCc8RdgKI&_nG@sg)xWwpfRmkvRG!CuT;__F5ma4+)~VL0oXq_!gT?B-QeuT+Px
zLG;X-<xoo(fa7YQG2+lz>u)u-8TmAe#LzHx?|zw;Z=!yF0bY7~Bu+Y|33~M%>W}*n
zU4Q=EX;}X8T--Ed9O@l^Jm-8@CV7}7UL-UWdiNfJtfuE<#i|AP&sUqYs;ksEPj^p8
z@rwpQ@~C5D=NQ_|WITW_aUNYG-t>eNag{Jg+O%4D&o-?#Oj2Bd@9pVy{cW2t=(cf4
zt(Ib<zV`mxwtk7JlkP>A9yj2E4%`WPvyB>yuggcYW{Ok}8_fzz^A=1o`vV_z=Gpaa
zTJ_Ddr<cW7HZtDE^GxwP)iVHaA99Yz*xXgZMqXY7?UpIeB&(i`-?W~t|N4`-qVEWt
z*_^IVBej?*<nnUjgxgX3n0mN&;Jsm&CIerFn2DFgS3#>O7fso{hvB{VXJ8$-&{|a4
zZS49sEq0ZGz~XaMCpe3XAKo){W^YoI6236YP}Q@UOK}-lK-Zr%;ZD>)`3&^#cbjoz
z)sxZ+h+k}8iCcz0g_`NL_K)c(zR@6PcIe)H0MgQG;l)Y+z}|iP!k)l(UMJXw+jO*^
zUYX(v|7U7Z6wKI;bSi(mGQA4XN>l}eu<JjIZ?<m4jl;RtuO5#2W*PkXD$IQKaT5@m
zw{B~^!pM+@N|k^hy^cM$4mx)1i_JO9F=N&Q?i<3*jJ^VbDVFYjnN?3bW%IgfNDt;0
zqb`R)O})MGD~pg)ZleeVmG72&^<{j{`n7m#+7r0qZ^O{^?3Ttz?dkXI$;HGchofPm
z=IGOZxbGEe8&gJ9nKm??JNHAKI>+OkIWJ(%x)sK&RPAs;p#4fU^`0^q3ZQm?o3Uu4
zOqD8tps1fu*PlH8PMmmhV_b2~Ff)zGIIh3(+N3eq{?m6DeEWE_)HQpIXuf>*<%zJL
zNo#o4KdAxEJ+}w)^7dfz%l9C6?_U4Vq^ckozLHH5?3_?(JZcB|@-a<{Dzhq#e4t!L
z>ivJc{v5vkay<r&7>8=rQ_X@Wfm7b6%NAn(+b^O2O%LGo%qErhzHM{`Ajm_zbiEGM
ztEc0uuh!y?H^)a#i&mQm&yB%X-pQFDEl9?RW-f3m9tsFy*I$RXUKxkW`qA}Uwlb?a
z0s70&+wszqcc5k4i_yK;4gP9Rdd5^RhO%m-MOvdqS!jNC2QG_)`1icYSpE4Y;h~^*
zy!_6oOoo`Tt~{6fkwo6C<SI}p2(B=JR!~rgDNo#q6B;x^uj|6&TdO^(GWF?E?C~ND
z95I$dA&#4^Dteg5pLZGL=^|PisBYb^M~Lfe@t!<o6n<xnkukn|d+oipyc=+gUM_8w
z1*ve6CrEe0$gAKsis&)bVN!H|=QX<imUS2~k_(=+RI_-I>o1=B20nW4Eesew7AKr^
zqUpVZ@{Ll^Gsp;Yjne8ct65t#KBE;+l+w((!SpxAfd@DT_I^dWTQbaP6m1Wd1(9pk
z0Ira#fDljPx2#`}w_YBHOZwl2rmb3=sIR5!x7)Vim1pio#~%IAzRTt2oZs|-EU}fc
zX}9bEprdq>ZV8ET=t?h;2Q<sN0Q2Tg!Rpl?nGUIn?w&_!Dh!q*H^@&_cJiUbxUMTu
z`ovOJ3!Xx{{>$U&`X`-+Ue~*>&wOoPCyl)WX*JSu&8?$NFf_beepOa5x<2*t+G?ZJ
z`kc<l$f(all{hqMaxR{KVI+sbo{E>s6#y;UWHUQ|XD}rJzf#^FZid21!rVp`fuO)Q
z=hcbWma`T&+)3B3QOy`?@?U=eo7Q}U8}E7w>2+(H>B~dhFc)tTE`PhsmKi~ZwK8qg
zR&#i2^(1ubF#zk<EyIN!uflG=M|kbE$NX90U^3K+)Zusj?g(zrz%D-gplKmOa0MX9
z$S>D#z}u4_N6-F4(X>q)<N9VcSo=BNd}$0W?Kc!H+O<D0>RbFG6r>grfVj_<>7rP0
ze*4S0S*pP)aUHI`?taXf`yy7Z`oNzRB4kLm1-g#on0`EqA7cfx1N<6`-{5HKQi>(m
zbN#80--&uBor)_4jxfPZ!0pT3he?kP$B7Lyk$vMx@GGz|Fa27AJz95PL7QgvSn}<B
zjo<XzHPEHoby)oF>t<?f-n;`QP8>;Z*lo@mBV>4dU0vMH&e1C~IfI0*Fv{7=szA!Q
z3M%T)okrK+v<BDzoeLhiz7)~P&zmnkh9AD&ia~cj#XSIL1No_Vxor*2M`MKYmFzN`
zW|3`dEL!V=j-9XO6d1(Pr89BsZBJp>uI-ri>Z35{nvxUE*OhCAfJ2*^3;>)GKH+Y)
z^neAC+aT=v8}YBnPvFu4x1;&_=a_MyoC>U3x&#Ynyoi3o9z>JY%}mb<6irRK>r`0x
z2%2CjUwr+$O}k|}SrYK77eVt`9$UY`amEW4%$v#`=CQbG@Z(sx@D(gyu_QbcT>Wit
z6DT}<E#>VK6&uIT{S8}c9)m4Z@=Shi67Ddz68d_Zo5wBaw6Xs{wVJh%)wToUz-GR}
zc2Ja`hkritchqZe8X7ifj!mDfgTs>yp1QIz?`Gjons;S|S3~>D!}+SKBJ<<7F>Sp(
z@e(b59!)aHud}Li#>vgyiD|Dq$m8`^NKenel*#wtv{SRp8)toZp`_-T>2xN^81hB(
z&)A}TW`)zczd8>mr(hIc-<jmS3l#cmmwtdp242dDUz0-P_$62&dmx_Q{_)F3eioL%
z<0OX*>vBpFL#i8Q67{Qdz)Dj@arhy;vQ(>x!tf)-=slxSQBgh;6H{nl2z&SL#=ZAV
zN4s_xo3GYPOR2<*uO7I0!<d%m1nyD0aBC7AH#6xPBRM`2gqnI^4fS%gG1vH=uDk(@
zXG}rcuD#IX+JRU$=YOzx<`n#G_(M27tGW9=UncUOyQm1dN}CrQ;S$N@NHZhP)N@;y
ze9>B(N^<yLn9^@Je}9R}l|uZ!_WN(P;*kgY<LZ8Qp?2-#Fn-)%9K)@%aX?n{dS9%0
z0$s$;ooO2L#|NNeEo@%=8<ub>+m-Jq4O%7u5w!#k_8T4;hXj5s@ZOs*Bjdyd*tmQt
z+V|{>?3;&}wYdFe);Iv$8TS*ILSL@v<>}eIjB?g9OAdS!t#T!=9!bwb1;g<tG(fYn
zF2H9i-{m5y3#Z340#Gj;jMdHsA(7t94^rN5c`i09TgeB2z2v;N(|LxT+=YgRJ&ytf
z0`r&=himR0gCv46`#HY8;=A+Iw3-+*xIgo%@!UTRDb=bYxf<`HQ%UnuSu)!sCZ*6t
z`Qm}YCB$(R&wvx6yT&odXj)RN(WT^r>47K1MXOLcxt9TzSwRNeo|g~CBcuCa$FKi~
zV>3=dbsnLadgN*N@y^-Fp^RrN3Ht#M@7?KGlJ*-uq>k<71qKBPlVZFJh(5#bMx7Jt
zV;kT0=I_fz{@z^d-tmhGh(-K}Nbx~&p_F5%oagM}KuBJkt~6fb8TJzx+T~3GB{?OP
zR~Re-mrBsW0EU6o*CXL8m9NiSAnB$FJYyHHkfs}&adzv!q7j4O>J^J{QJ4M%I!HZO
zMBS-6Kf*F!u`X-T3QPFtC*K2<qeT&e;r)zl?=u#}%Y~ys(8j6Lm3?s$FO};fXnw@Z
z7*K{XhSyRa1D9}4E#?=L#r&|Sh!LkSe;*3+_MtG3fN@4C$lFWc_Jjd6d3&+@x9#}p
zf4A}bA+{+jpz?)`Kg9%6AhPTDPq8GZSJw={A%RFB1fqbcUOf#vcl^XoD(Cx#lg-xI
zgb7+N8>n6>``~`^ptk!{1OUqJTsQF1qF<d?3H<e%5zH8MJwToychdL3O1`TyFUa+!
zHYW*5Nk*;-bE?mJqNRjrU8YOCy9XlgWhMODk)Bg(fFy|YrecC2&(c>UVIT<}!ORzM
z20ZspQ0pH17t~6xi?dpE2=7BAg97=H7f-A)t#ymQ$I7yk(Z|jZJ^xoHgWvzIq&@Yu
zE|9{u;>K|1DcK;{v|o0RdJ%jre#-YiGytm1yF^$znXxWUB#0*PvP@p7q<Nns@%umt
z+O%pF*dET^|7+y$f80+k+U0a(<9EtS!b8Cps9Lc=sA@eUSm0QOM9>maNm(rUHeUr*
z#_Cc6aB1ta0`tW(IvJ3oGxdCNdSE_1aSY!W`_N0@uq#kENix!0+ACvTRqFY5?D{tC
z@-y{p+Lme4b{&^yFxQ4{Of2J7EhYde+?$Svc=j=!G3MxjAVId_Htm*qKu6SLS*%V_
zK3BCU;CI7qn&Ba_iIpk^K(j)y7=svx790);MyTCOHmyP#>*hrmr+l|uH`~yNdQoK_
zFB#8CXhDjDLD5lP{OnXI0IJwIBap{RS2G~bKU5Grg`~8SZHQeKL93$bA0`N#6}8)2
z5MP9=v$78U#rQ!1kUpK#1ACkz^x-n{b?!IJi_ppxj4;SG{%C`sJ-&?BkWDE?c8eAV
z|59AGRS5u9lqYnmg=+}HXs(LOw7>uV0)ljrPVD*7Kjz??s`R{UQu6vI{8Yu?1-qOc
z2|-NYO7vvzLe1Q37`Y)k<`m_H^h2oU^JN>KmY#gpG^Y)Y`IDJq3_CCL(Entn${t{2
z(=|QtZ!Xx*<BE5Rkk?ds!1y-Wy7_9%XM1QU>wq>D)H7I#m)g7KqPYszP%Puhd&=H2
zoOd|Jd+c;S?6hcy^H=V1Mp35=wdyMI{+xy!V|juskxgNXV%PKQxb^j{qcJXQv^#vC
z%l?Pkof=rir^8N0#$mU18}hg+@ZN58A+wvQSVt~E%pXqjn`g6)jWASg)Rk2PK`yF8
zA1=+gFEEE0`oqbI=bwy)+QN(7A4x|909%1YyPd31IeJmE?ldx!Jr<I&>NV{gp*6eI
z=4FIiw4SwT;_}rTZR?rSZz+%da#(F1UuDg1cXD6@vyWIZIvg=>7Na$e6WX-{ncRZ5
zrLk?;S7*>=|3bPZ4Gih!gfW<Hw1E*o+`tqwZx62@n4QBmD|zDeY2xw7X%XOd<|w42
z20*l~`kTXyc1SPgXDRf8+QBlU(P)M;<Kl%pzspSsU?(4U=xvpGVH3^dC_MicvQh3V
TxMBN400000NkvXXu0mjfUH*W0

literal 0
HcmV?d00001

diff --git a/main.py b/main.py
new file mode 100644
index 0000000..f51944a
--- /dev/null
+++ b/main.py
@@ -0,0 +1,142 @@
+import os
+import zlib
+from cryptography.hazmat.primitives.kdf.pbkdf2 import PBKDF2HMAC
+from cryptography.hazmat.primitives.hashes import SHA256
+from cryptography.hazmat.primitives.ciphers import Cipher, algorithms, modes
+from cryptography.hazmat.backends import default_backend
+
+
+# 辅助函数：使用AES加密数据
+def encrypt_data(key, data):
+    iv = os.urandom(16)  # 生成随机的初始化向量（IV）
+    cipher = Cipher(algorithms.AES(key), modes.CFB(iv), backend=default_backend())
+    encryptor = cipher.encryptor()
+    encrypted_data = iv + encryptor.update(data) + encryptor.finalize()
+    return encrypted_data
+
+# 辅助函数：使用AES解密数据
+def decrypt_data(key, data):
+    iv = data[:16]  # 提取初始化向量（IV）
+    encrypted_content = data[16:]
+    cipher = Cipher(algorithms.AES(key), modes.CFB(iv), backend=default_backend())
+    decryptor = cipher.decryptor()
+    decrypted_data = decryptor.update(encrypted_content) + decryptor.finalize()
+    return decrypted_data
+
+# 辅助函数：通过密码派生加密密钥
+def derive_key(password, salt):
+    kdf = PBKDF2HMAC(
+        algorithm=SHA256(),
+        length=32,
+        salt=salt,
+        iterations=100000,
+        backend=default_backend()
+    )
+    return kdf.derive(password.encode())
+
+# 压缩文件数据
+def compress_data(data):
+    return zlib.compress(data)
+
+# 解压文件数据
+def decompress_data(data):
+    return zlib.decompress(data)
+
+# 加密并压缩单个文件
+def encrypt_file(file_path, password, volume_size, output_dir):
+    # 读取文件内容
+    with open(file_path, 'rb') as f:
+        data = f.read()
+
+    # 压缩文件内容
+    compressed_data = compress_data(data)
+
+    # 加密压缩后的内容
+    salt = os.urandom(16)
+    key = derive_key(password, salt)
+    encrypted_data = encrypt_data(key, compressed_data)
+
+    # 如果数据大小超过分卷限制，分卷存储
+    file_name = os.path.basename(file_path)
+    encrypted_dir = os.path.join(output_dir, file_name)
+    os.makedirs(encrypted_dir, exist_ok=True)
+
+    if len(encrypted_data) > volume_size:
+        start = 0
+        part_number = 1
+        while start < len(encrypted_data):
+            end = start + volume_size
+            part_data = encrypted_data[start:end]
+            part_file_path = os.path.join(encrypted_dir, f"{file_name}.part{part_number}")
+            with open(part_file_path, 'wb') as part_file:
+                if part_number == 1:
+                    part_file.write(salt)
+                part_file.write(part_data)
+            start = end
+            part_number += 1
+    else:
+        # 如果数据不需要分卷，直接保存
+        output_path = os.path.join(encrypted_dir, f"{file_name}.enc")
+        with open(output_path, 'wb') as enc_file:
+            enc_file.write(salt + encrypted_data)
+
+    print(f"Encrypted and saved: {file_path} to {encrypted_dir}")
+
+# 解密并解压单个文件或分卷
+def decrypt_file(encrypted_dir, password, output_dir):
+    # 读取所有分卷或单个加密文件
+    parts = [os.path.join(encrypted_dir, f) for f in os.listdir(encrypted_dir) if f.rsplit('.')[-1].startswith("part") or f.endswith(".enc")]
+    parts = sorted(parts)  # 确保按顺序读取
+    print(f"Parts found: {parts}")
+    encrypted_data = b""
+    salt = None
+    for i, part in enumerate(parts):
+        with open(part, 'rb') as part_file:
+            if i == 0:
+                salt = part_file.read(16)
+            encrypted_data += part_file.read()
+
+    if salt is None:
+        raise ValueError("Salt not found in the first part.")
+
+    # 解密数据
+    key = derive_key(password, salt)
+    decrypted_data = decrypt_data(key, encrypted_data)
+
+    # 解压数据
+    decompressed_data = decompress_data(decrypted_data)
+    output_file = os.path.join(output_dir, os.path.basename(encrypted_dir))
+    with open(output_file, 'wb') as f:
+        f.write(decompressed_data)
+
+    print(f"Decrypted and extracted to: {output_file}")
+
+# 加密文件夹中的所有文件
+def encrypt_directory(directory, password, volume_size, output_dir):
+    for root, _, files in os.walk(directory):
+        for file in files:
+            file_path = os.path.join(root, file)
+            encrypt_file(file_path, password, volume_size, output_dir)
+
+# 解密文件夹中的所有加密文件
+def decrypt_directory(directory, password, output_dir):
+    for root, dirs, _ in os.walk(directory):
+        for dir in dirs:
+            encrypted_dir = os.path.join(root, dir)
+            decrypt_file(encrypted_dir, password, output_dir)
+
+if __name__ == "__main__":
+    action = "encrypt"  # 或 "decrypt"
+    # action = "decrypt"  # 或 "decrypt"
+    password = "123456"
+    volume_size = 10 * 1024 * 1024  # 每个分卷大小（10 MB）
+    source_dir = "./data"  # 原始数据文件夹
+    encrypted_dir = "./encrypted"  # 加密文件夹
+    decrypted_dir = "./decrypted"  # 解密文件夹
+
+    if action == "encrypt":
+        encrypt_directory(source_dir, password, volume_size, encrypted_dir)
+    elif action == "decrypt":
+        decrypt_directory(encrypted_dir, password, decrypted_dir)
+    else:
+        print("Invalid action. Use 'encrypt' or 'decrypt'.")
\ No newline at end of file