bcmd 0.5.14__tar.gz → 0.5.15__tar.gz

{bcmd-0.5.14 → bcmd-0.5.15}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: bcmd
-Version: 0.5.14
+Version: 0.5.15
 Summary: Commands for Beni
 Author-email: Beni Mang <benimang@126.com>
 Maintainer-email: Beni Mang <benimang@126.com>

{bcmd-0.5.14 → bcmd-0.5.15}/bcmd/tasks/wasabi.py

@@ -1,4 +1,5 @@
 import getpass
+import stat
 from pathlib import Path
 from typing import Final
 
@@ -16,16 +17,14 @@ MAX_ENCRYPT_SIZE = 199 * 1024
 @app.command()
 @syncCall
 async def unzip(
-    file: Path = typer.Argument(Path.cwd(), help='加密文件'),
+    target: Path = typer.Argument(Path.cwd(), help='加密文件'),
     password: str = typer.Option('', '--password', '-p', help='密码'),
 ):
     '解压缩加密文件成目录'
-    assert file.is_file(), f'不是文件 {file}'
-    workspace = file.with_suffix('')
-    assert not workspace.exists(), f'目录已存在 {workspace}'
+    assert target.is_file(), f'不是文件 {target}'
     password = password or getpass.getpass('请输入密码: ')
     with bpath.useTempFile() as tempFile:
-        data = await bfile.readBytes(file)
+        data = await bfile.readBytes(target)
         if SEP not in data:
             data = bcrypto.decrypt(data, password)
         else:
@@ -34,28 +33,31 @@ async def unzip(
             data = partA + partB
         data = shuffleSequence(data)
         await bfile.writeBytes(tempFile, data)
-        await bzip.sevenUnzip(tempFile, workspace)
-        await bpath.removeSecure(file)
-        bcolor.printGreen(workspace)
+        tempPath = target.with_suffix('.tmp')
+        await bzip.sevenUnzip(tempFile, tempPath)
+
+        # 调整文件权限，完全擦除
+        target.chmod(stat.S_IWRITE)
+        await bpath.removeSecure(target)
+
+        bpath.move(tempPath, target)
         bcolor.printGreen('OK')
 
 
 @app.command()
 @syncCall
 async def zip(
-    workspace: Path = typer.Argument(Path.cwd(), help='输出目录'),
+    target: Path = typer.Argument(Path.cwd(), help='输出目录'),
     password: str = typer.Option('', '--password', '-p', help='密码'),
 ):
     '将目录压缩成加密文件'
-    workspace = workspace.absolute()
-    assert workspace.is_dir(), f'不是目录 {workspace}'
-    zipFile = workspace.with_suffix('.dat')
-    assert not zipFile.exists(), f'文件已存在 {zipFile}'
+    target = target.absolute()
+    assert target.is_dir(), f'不是目录 {target}'
     password = password or genPassword()
     with bpath.useTempFile() as tempFile:
-        await bzip.sevenZipFolder(tempFile, workspace)
+        await bzip.sevenZipFolder(tempFile, target)
         data = await bfile.readBytes(tempFile)
-        bpath.remove(tempFile)
+        bpath.remove(tempFile)  # 为了安全所以立即删除
         data = shuffleSequence(data)
         if len(data) < MAX_ENCRYPT_SIZE:
             data = bcrypto.encrypt(data, password)
@@ -63,9 +65,17 @@ async def zip(
             partA, partB = data[:MAX_ENCRYPT_SIZE], data[MAX_ENCRYPT_SIZE:]
             partA = bcrypto.encrypt(partA, password)
             data = partA + SEP + partB
-        await bfile.writeBytes(zipFile, data)
-        await bpath.removeSecure(workspace)
-    bcolor.printGreen(zipFile)
+        tempZipFile = target.with_suffix('.tmp')
+        await bfile.writeBytes(tempZipFile, data)
+
+        # 调整目录权限，完全擦除
+        target.chmod(stat.S_IWRITE)
+        for file in target.glob('**/*'):
+            file.chmod(stat.S_IWRITE)
+        await bpath.removeSecure(target)
+
+        bpath.move(tempZipFile, target)
+
     bcolor.printGreen('OK')
 
 
@@ -77,9 +87,8 @@ async def change_pass(
     new_password: str = typer.Option('', '--new-password', '-n', help='新密码'),
 ):
     with bpath.useTempPath() as tempPath:
-        tempFile = tempPath / file.name
-        bpath.copy(file, tempFile)
-        unzip(tempFile, password)
-        workspace = tempFile.with_suffix('')
-        zip(workspace, new_password)
-        bpath.copy(tempFile, file)
+        target = tempPath / file.name
+        bpath.copy(file, target)
+        unzip(target, password)
+        zip(target, new_password)
+        bpath.copy(target, file)

{bcmd-0.5.14 → bcmd-0.5.15}/bcmd.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: bcmd
-Version: 0.5.14
+Version: 0.5.15
 Summary: Commands for Beni
 Author-email: Beni Mang <benimang@126.com>
 Maintainer-email: Beni Mang <benimang@126.com>

{bcmd-0.5.14 → bcmd-0.5.15}/pyproject.toml

@@ -3,7 +3,7 @@
 
 [project]
 name = 'bcmd'
-version = '0.5.14'
+version = '0.5.15'
 description = 'Commands for Beni'
 requires-python = '>=3.10'
 keywords = ['benimang', 'beni', 'bcmd']