bcmd 0.6.6__py3-none-any.whl → 0.6.7__py3-none-any.whl

bcmd/tasks/proxy.py

@@ -0,0 +1,55 @@
+from __future__ import annotations
+
+from typing import Final
+
+import psutil
+import pyperclip
+import typer
+from beni import bcolor, btask
+from beni.bfunc import syncCall, textToAry
+
+app: Final = btask.app
+
+
+@app.command()
+@syncCall
+async def proxy(
+    port: int = typer.Argument(15236, help="代理服务器端口"),
+):
+    '生成终端设置代理服务器的命令'
+    processNameAry: list[str] = []
+    process = psutil.Process().parent()
+    while process:
+        processNameAry.append(process.name())
+        process = process.parent()
+    template = ''
+
+    # 针对不同的终端使用不同的模板
+    if 'cmd.exe' in processNameAry:
+        template = cmdTemplate
+    elif set(['powershell.exe', 'pwsh.exe']) & set(processNameAry):
+        template = powerShellTemplate
+
+    btask.assertTrue(template, f'不支持当前终端（{processNameAry}）')
+    lineAry = textToAry(template.format(port))
+    msg = '\n'.join(lineAry)
+    bcolor.printMagenta('\r\n' + msg)
+    msg += '\n' # 多增加一个换行，直接粘贴的时候相当于最后一行也执行完
+    pyperclip.copy(msg)
+    bcolor.printYellow('已复制，可直接粘贴使用')
+
+
+# ------------------------------------------------------------------------------------
+
+
+cmdTemplate = '''
+    set http_proxy=http://localhost:{0}
+    set https_proxy=http://localhost:{0}
+    set all_proxy=http://localhost:{0}
+'''
+
+powerShellTemplate = '''
+    $env:http_proxy="http://localhost:{0}"
+    $env:https_proxy="http://localhost:{0}"
+    $env:all_proxy="http://localhost:{0}"
+'''

bcmd/tasks/time.py

@@ -0,0 +1,81 @@
+import time
+from datetime import datetime as Datetime
+from datetime import timezone
+from typing import Final
+from zoneinfo import ZoneInfo
+
+import typer
+from beni import bcolor, btask
+from beni.bfunc import syncCall, textToAry
+from beni.btype import Null
+
+app: Final = btask.app
+
+
+@app.command('time')
+@syncCall
+async def showtime(
+    args: list[str] = typer.Argument(None),
+):
+    '''
+    格式化时间戳\n
+    beni time\n
+    beni time 1632412740\n
+    beni time 1632412740.1234\n
+    beni time 2021-9-23\n
+    beni time 2021-9-23 09:47:00\n
+    '''
+    args = args or []
+    btask.assertTrue(len(args) <= 2, '参数过多')
+    value1: str | None = args[0] if len(args) >= 1 else None
+    value2: str | None = args[1] if len(args) >= 2 else None
+    timestamp: float = Null
+    if not value1:
+        timestamp = time.time()
+    else:
+        try:
+            timestamp = float(value1)
+        except:
+            try:
+                if value2:
+                    timestamp = Datetime.strptime(f'{value1} {value2}', '%Y-%m-%d %H:%M:%S').timestamp()
+                else:
+                    timestamp = Datetime.strptime(f'{value1}', '%Y-%m-%d').timestamp()
+            except:
+                pass
+    if not timestamp:
+        bcolor.printRed('参数无效\n')
+        bcolor.printRed('使用示例：')
+        msgAry = textToAry(str(showtime.__doc__))[1:]
+        bcolor.printRed('\n'.join(msgAry))
+        return
+    print()
+    bcolor.printMagenta(timestamp)
+    print()
+    # localtime = time.localtime(timestamp)
+    # tzname = time.tzname[(time.daylight and localtime.tm_isdst) and 1 or 0]
+    # bcolor.printx(time.strftime('%Y-%m-%d %H:%M:%S %z', localtime), tzname, colors=[Fore.YELLOW])
+    # print()
+    datetime_utc = Datetime.fromtimestamp(timestamp, tz=timezone.utc)
+    tzname_list = [
+        'Australia/Sydney',
+        'Asia/Tokyo',
+        'Asia/Shanghai',
+        'Asia/Kolkata',
+        'Africa/Cairo',
+        'Europe/London',
+        'America/Sao_Paulo',
+        'America/New_York',
+        'America/Chicago',
+        'America/Los_Angeles',
+    ]
+    for tzname in tzname_list:
+        datetime_tz = datetime_utc.astimezone(ZoneInfo(tzname))
+        dstStr = ''
+        dst = datetime_tz.dst()
+        if dst:
+            dstStr = f'(DST+{dst})'
+        if tzname == 'Asia/Shanghai':
+            bcolor.printYellow(f'{datetime_tz} {tzname} {dstStr}')
+        else:
+            print(f'{datetime_tz} {tzname} {dstStr}')

bcmd/tasks/upgrade.py

@@ -0,0 +1,22 @@
+from typing import Final
+
+import pyperclip
+import typer
+from beni import bcolor, btask
+from beni.bfunc import syncCall
+
+app: Final = btask.app
+
+
+@app.command()
+@syncCall
+async def upgrade(
+    name: str = typer.Argument('bcmdx', help='要更新的包名'),
+):
+    '使用 pipx 官方源更新指定包到最新版本'
+
+    cmd = f'pipx upgrade {name} -i https://pypi.org/simple'
+    pyperclip.copy(cmd + '\n')
+    bcolor.printGreen(cmd)
+    bcolor.printGreen('已复制到剪贴板（需要手动执行）')
+    bcolor.printGreen('OK')

bcmd/tasks/wasabi.py

@@ -0,0 +1,94 @@
+import getpass
+import stat
+from pathlib import Path
+from typing import Final
+
+import typer
+from beni import bcolor, bcrypto, bfile, bpath, btask, bzip
+from beni.bfunc import shuffleSequence, syncCall
+from beni.binput import genPassword
+
+app: Final = btask.newSubApp('Wasabi 工具')
+
+SEP = f'{chr(852)}{chr(322)}{chr(470)}'.encode()
+MAX_ENCRYPT_SIZE = 199 * 1024
+
+
+@app.command()
+@syncCall
+async def unzip(
+    target: Path = typer.Argument(Path.cwd(), help='加密文件'),
+    password: str = typer.Option('', '--password', '-p', help='密码'),
+):
+    '解压缩加密文件成目录'
+    assert target.is_file(), f'不是文件 {target}'
+    password = password or getpass.getpass('请输入密码: ')
+    with bpath.useTempFile() as tempFile:
+        data = await bfile.readBytes(target)
+        if SEP not in data:
+            data = bcrypto.decrypt(data, password)
+        else:
+            partA, partB = data.split(SEP)
+            partA = bcrypto.decrypt(partA, password)
+            data = partA + partB
+        data = shuffleSequence(data)
+        await bfile.writeBytes(tempFile, data)
+        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(
+    target: Path = typer.Argument(Path.cwd(), help='输出目录'),
+    password: str = typer.Option('', '--password', '-p', help='密码'),
+):
+    '将目录压缩成加密文件'
+    target = target.absolute()
+    assert target.is_dir(), f'不是目录 {target}'
+    password = password or genPassword()
+    with bpath.useTempFile() as tempFile:
+        await bzip.sevenZipFolder(tempFile, target)
+        data = await bfile.readBytes(tempFile)
+        bpath.remove(tempFile)  # 为了安全所以立即删除
+        data = shuffleSequence(data)
+        if len(data) < MAX_ENCRYPT_SIZE:
+            data = bcrypto.encrypt(data, password)
+        else:
+            partA, partB = data[:MAX_ENCRYPT_SIZE], data[MAX_ENCRYPT_SIZE:]
+            partA = bcrypto.encrypt(partA, password)
+            data = partA + SEP + partB
+        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')
+
+
+@app.command()
+@syncCall
+async def change_pass(
+    file: Path = typer.Argument(Path.cwd(), help='加密文件'),
+    password: str = typer.Option('', '--password', '-p', help='密码'),
+    new_password: str = typer.Option('', '--new-password', '-n', help='新密码'),
+):
+    with bpath.useTempPath() as tempPath:
+        target = tempPath / file.name
+        bpath.copy(file, target)
+        unzip(target, password)
+        zip(target, new_password)
+        bpath.copy(target, file)

bcmd/utils/tkUtil.py

@@ -0,0 +1,243 @@
+import tkinter as tk
+from tkinter import filedialog
+from tkinter.scrolledtext import ScrolledText
+from typing import Any, Callable, Literal, TypeVar, Union
+from uuid import uuid4
+
+RADIO_NOTHING = uuid4().hex
+TkVar = TypeVar('TkVar', bound=Union[tk.StringVar, tk.IntVar, tk.DoubleVar, tk.BooleanVar])
+
+
+class TkForm(tk.Tk):
+
+    _rowIndex = -1
+    _initList: list[Callable[..., None]] = []
+    _varList: list[tk.Variable] = []  # 用来存储 var 变量，避免传出去，外面没有接收导致界面异常
+
+    def __init__(self):
+        super().__init__()
+        self.resizable(False, False)
+        self.bind("<Map>", self._onInit)
+
+    def _onInit(self, evt: tk.Event):
+        if evt.widget == self:
+            for callback in self._initList:
+                callback()
+            self._initList.clear()
+
+    def addInitHandler(self, handler: Callable[..., None]):
+        self._initList.append(handler)
+
+    def _initVar(self, var: TkVar) -> TkVar:
+        self._varList.append(var)
+        return var
+
+    def run(self):
+        self.center()
+        self.mainloop()
+
+    def center(self):
+        self.withdraw()  # 先隐藏窗口，避免闪动
+        self.update_idletasks()  # 确保获取正确的窗口尺寸
+        width = self.winfo_width()  # 获取窗口宽度
+        height = self.winfo_height()  # 获取窗口高度
+        screen_width = self.winfo_screenwidth()  # 屏幕宽度
+        screen_height = self.winfo_screenheight()  # 屏幕高度
+        x = (screen_width - width) // 2  # 水平居中
+        y = (screen_height - height) // 2  # 垂直居中
+        self.geometry(f"+{x}+{y}")  # 设置窗口位置
+        self.deiconify()  # 恢复显示窗口
+
+    def addRow(self, desc: str, widget: tk.Widget):
+        self._rowIndex += 1
+        tk.Label(text=desc).grid(row=self._rowIndex, column=0, padx=10, pady=5, sticky='e')
+        widget.grid(row=self._rowIndex, column=1, padx=10, pady=5, sticky='w')
+
+    def addRowFrame(self):
+        self._rowIndex += 1
+        frame = tk.Frame(self)
+        frame.grid(row=self._rowIndex, column=0, columnspan=2, padx=10, pady=5)
+        return frame
+
+    def addRowFrameWithDesc(self, desc: str):
+        self._rowIndex += 1
+        tk.Label(text=desc).grid(row=self._rowIndex, column=0, padx=10, pady=5, sticky='e')
+        frame = tk.Frame(self)
+        frame.grid(row=self._rowIndex, column=1, padx=10, pady=5, sticky='w')
+        return frame
+
+    def addLabel(
+        self,
+        desc: str,
+        text: str
+    ):
+        self.addRow(desc, tk.Label(text=text))
+
+    def addBtn(
+        self,
+        label: str,
+        command: Callable[..., None],
+        *,
+        width: int = 20,
+        focus: bool = False
+    ):
+        frame = self.addRowFrame()
+        btn = tk.Button(frame, text=label, width=width, command=command)
+        btn.pack(side="left", expand=True, padx=15)
+        if focus:
+            self._initFocus(btn)
+
+    def addRadioBtnList(
+        self,
+        desc: str,
+        selectionList: list[str],
+        *,
+        selectedIndex: int | None = None,
+        focusIndex: int | None = None,
+        onChanged: Callable[[str], None] | None = None,
+    ):
+        frame = tk.Frame()
+        self.addRow(desc, frame)
+        var = tk.StringVar(value=selectionList[selectedIndex] if selectedIndex is not None else RADIO_NOTHING)
+        radioBtnList: list[tk.Radiobutton] = []
+        for version in selectionList:
+            radioBtn = tk.Radiobutton(frame, text=version, variable=var, value=version)
+            radioBtn.pack(side="left", padx=(0, 15))
+            setWidgetClickFocus(radioBtn)
+            radioBtnList.append(radioBtn)
+        if focusIndex is not None:
+            self._initFocus(radioBtnList[focusIndex])
+        if onChanged:
+            var.trace_add('write', lambda *args: onChanged(var.get()))  # type: ignore
+            self.addInitHandler(
+                lambda: onChanged(var.get())
+            )
+        return var
+
+    def addEntry(
+        self,
+        desc: str,
+        var: tk.StringVar,
+        *,
+        width: int = 60,
+        focus: bool = False,
+        justify: Literal['left', 'right', 'center'] = tk.LEFT,
+        password: bool = False,
+        command: Callable[..., Any] | None = None,
+    ):
+        self._initVar(var)
+        entry = tk.Entry(self, width=width, justify=justify, textvariable=var)
+        entry.icursor(tk.END)
+        self.addRow(desc, entry)
+        if password:
+            entry.config(show='*')
+        if focus:
+            self._initFocus(entry)
+        if command:
+            entry.bind('<Return>', lambda event: command())
+        return entry
+
+    def addChoisePath(
+        self,
+        desc: str,
+        var: tk.StringVar,
+        *,
+        width: int = 47,
+        focus: bool = False,
+        isDir: bool = False,
+    ):
+        self._initVar(var)
+        frame = self.addRowFrameWithDesc(desc)
+        entry = tk.Entry(frame, width=width, textvariable=var)
+        entry.icursor(tk.END)
+        entry.pack(side="left")
+        btn = tk.Button(frame, text=f'选择{'目录' if isDir else '文件'} ...', width=10, command=lambda: onBtn())
+        btn.pack(side="left", padx=(10, 0))
+        if focus:
+            self._initFocus(btn)
+
+        def onBtn():
+            if isDir:
+                var.set(filedialog.askdirectory())
+            else:
+                var.set(filedialog.askopenfilename())
+
+    def addScrolledText(
+        self,
+        desc: str,
+        var: tk.StringVar,
+        *,
+        width: int = 60,
+        height: int = 3,
+        focus: bool = False,
+    ):
+        self._initVar(var)
+        scrolledText = ScrolledText(self, width=width, height=height)
+        scrolledText.insert(tk.END, var.get())
+        self.addRow(desc, scrolledText)
+
+        def on_text_change(*args: Any):
+            new_value = scrolledText.get("1.0", tk.END)
+            if new_value != var.get():
+                var.set(new_value)
+
+        scrolledText.bind("<KeyRelease>", on_text_change)
+
+        def on_tab(event: tk.Event):
+            widget = event.widget.tk_focusNext()
+            assert widget
+            widget.focus_set()
+            return "break"
+
+        scrolledText.bind("<Tab>", on_tab)
+
+        if focus:
+            self._initFocus(scrolledText)
+        return scrolledText
+
+    def addCheckBox(
+        self,
+        desc: str,
+        text: str,
+        value: bool = False,
+    ):
+        var = tk.BooleanVar(value=value)
+        self._initVar(var)
+        check_btn = tk.Checkbutton(text=text, variable=var)
+        self.addRow(desc, check_btn)
+        self.addInitHandler(
+            lambda: var.set(var.get())
+        )
+        setWidgetClickFocus(check_btn)
+        return var
+
+    def addCheckBoxList(
+        self,
+        desc: str,
+        dataList: list[tuple[str, bool]],
+    ):
+        varDict: dict[str, tk.BooleanVar] = {}
+        frame = tk.Frame(self)
+        self.addRow(desc, frame)
+        for label, value in dataList:
+            varDict[label] = tk.BooleanVar(value=value)
+            checkbox = tk.Checkbutton(frame, text=label, variable=varDict[label])
+            checkbox.pack(side="left", expand=True, padx=(0, 15))
+            setWidgetClickFocus(checkbox)
+            self.addInitHandler(
+                lambda: varDict[label].set(varDict[label].get())
+            )
+        return varDict
+
+    def _initFocus(self, widget: tk.Widget):
+        self.addInitHandler(
+            lambda: widget.focus_set()
+        )
+
+
+def setWidgetEnabled(widget: tk.Widget, value: bool):
+    widget['state'] = tk.NORMAL if value else tk.DISABLED
+
+
+def setWidgetClickFocus(widget: tk.Widget):
+    widget.bind("<Button-1>", lambda args: args.widget.focus_set())

{bcmd-0.6.6.dist-info → bcmd-0.6.7.dist-info}/METADATA

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

bcmd-0.6.7.dist-info/RECORD

@@ -0,0 +1,33 @@
+bcmd/__init__.py,sha256=GP_60-6vImXqdMfC5vc4xlscWajx4OYmnlNXASWn19w,147
+bcmd/common/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+bcmd/common/func.py,sha256=MB_ChA4aRzEjhMhyFmZ-h8aOZePzk0poYcTYRJUQxmI,509
+bcmd/common/secret.py,sha256=clemEVTCl3shhRBgjKZfrtTdyODz5U-wOyHfUXPegG4,1692
+bcmd/resources/project/main.py,sha256=xdskz_sf05fYA1SRMFCIxDjx8SnegxTbCmHpW86ItLs,11
+bcmd/tasks/__init__.py,sha256=lcVrTgdEph3tKBwwCPVyEOHp-1A3zdFA_6OWLrOo19M,314
+bcmd/tasks/bin.py,sha256=DufZGRX7b2zclSaZM-zUPGwOiycN9RsV8KxF8tfSyqs,3240
+bcmd/tasks/code.py,sha256=IUs_ClZuSsBk2gavlitC8mkRrQQX9rvNDgR8cFxduBA,3992
+bcmd/tasks/crypto.py,sha256=LKvgsMPLvsi1wlt66TinYiN-oV2IPAfaN9y7hWaVpHs,2951
+bcmd/tasks/debian.py,sha256=B9aMIIct3vNqMJr5hTr1GegXVf20H49C27FMvRRGIzI,3004
+bcmd/tasks/download.py,sha256=XdZYKi8zQTNYWEgUxeTNDqPgP7IGYJkMmlDDC9u93Vk,2315
+bcmd/tasks/image.py,sha256=OQxrtTkPnJ_FZ6uenxdNlynhxwrAVy1P_68QAnSBYB0,14238
+bcmd/tasks/json.py,sha256=WWOyvcZPYaqQgp-Tkm-uIJschNMBKPKtZN3yXz_SC5s,635
+bcmd/tasks/lib.py,sha256=ODF21sTjQ687X5AXGOvJjjw6dGNSnflwSz4wDlIbYVM,2861
+bcmd/tasks/math.py,sha256=xbl5UdaDMyAjiLodDPleP4Cutrk2S3NOAgurzAgOEAE,2862
+bcmd/tasks/mirror.py,sha256=nAe8NYftMKzht16MFBj7RqXwvVhR6Jh2uuAyJLh87og,1098
+bcmd/tasks/pdf.py,sha256=fkHRgxqzrRxdb4_-9pL9wp2roqAHJPS_dVqAGJvRUsM,1504
+bcmd/tasks/proxy.py,sha256=xvxN5PClUnc5LQpmq2Wug7_LUVpJboMWLXBvL9lX7EM,1552
+bcmd/tasks/time.py,sha256=ZiqA1jdgl-TBtFSOxxP51nwv4g9iZItmkFKpf9MKelk,2453
+bcmd/tasks/upgrade.py,sha256=z9Ein8U_Co7fsijPxyDFM8tXBFCp4bWATImwwwvC5ho,536
+bcmd/tasks/wasabi.py,sha256=xWFAxprSIlBqDDMGaNXZFb-SahnW1d_R9XxSKRYIhnM,3110
+bcmd/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+bcmd/utils/tkUtil.py,sha256=zhOFQB0SN-ZPSXgJ0-pyHeVldf_u8VNnODz7pQRbLkw,7709
+test/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+test/conftest.py,sha256=grlPunlsvrkt_8QPckmF4POiKUPVxIxm2TPAh_ZB-zs,405
+test/test_pdf.py,sha256=7yYlfydyhy2dmVYdTA5Vir2AI8TUdzEi55fL-AqJmio,1533
+test/test_proxy.py,sha256=UMF2hFFGUEbJR1jT2mO_wdo-7Rfp0NDqIdTRnOmwtjY,164
+test/test_wasabi.py,sha256=qqXG1Kb9hKH6t624R173j6LagkgmejN0CFYt7kL0nNs,1066
+bcmd-0.6.7.dist-info/METADATA,sha256=XVbdxqxA2VLOsMucxKpgGgVgslDi1nBM4QtLoCSDYvU,812
+bcmd-0.6.7.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+bcmd-0.6.7.dist-info/entry_points.txt,sha256=mriCeYh3wksKcqq3-LtzyFkSCIdN1uZc1IJwom-SW1s,34
+bcmd-0.6.7.dist-info/top_level.txt,sha256=fYY6tRrJ_G7tn24RXAG0M5ZKbcuaQznodfX1toFPSKs,10
+bcmd-0.6.7.dist-info/RECORD,,

{bcmd-0.6.6.dist-info → bcmd-0.6.7.dist-info}/top_level.txt

@@ -1 +1,2 @@
 bcmd
+test

test/conftest.py

@@ -0,0 +1,17 @@
+# '''
+# 快捷键（默认）
+# CTRL+; A        执行全部单元测试
+# CTRL+; E        只执行上次出错的用例
+# CTRL+; C        清除结果
+# CTRL+; CTRL+A   调试全部单元测试
+# CTRL+; CTRL+E   只调试上次出错的用例
+# '''
+
+import pytest_asyncio
+
+from bcmd.tasks import *
+
+
+@pytest_asyncio.fixture(scope='session', autouse=True)  # type: ignore
+def prepareSession():
+    yield

test/test_pdf.py

@@ -0,0 +1,47 @@
+from contextlib import asynccontextmanager
+from pathlib import Path
+from typing import Literal
+
+import img2pdf
+import pytest
+from beni import bfile, bpath, btask
+from PIL import Image
+
+
+@pytest.mark.asyncio
+async def test_pdf_output_images():
+    async with _createTempPdfFile() as pdfFile:
+        result = btask.testCall('pdf', 'output-images', '--target', pdfFile.as_posix())
+        assert result.exit_code == 0
+        outputImagesPath = pdfFile.parent / f'{pdfFile.stem}-PDF图片文件'
+        assert outputImagesPath.is_dir()
+        assert len(list(outputImagesPath.glob('*.png'))) == 1
+        assert len(list(outputImagesPath.glob('*.jpeg'))) == 1
+
+
+@asynccontextmanager
+async def _createTempPdfFile():
+
+    def create_color_block_image(
+            file: Path,
+            color: tuple[int, int, int],
+            size: tuple[int, int] = (500, 500),
+            image_format: Literal['JPEG', 'PNG'] = 'JPEG'
+    ):
+        image = Image.new('RGB', size, color)
+        image.save(file, format=image_format)
+        image.close()
+        return file
+
+    with bpath.useTempPath(True) as tempPath:
+        # 创建色块图片
+        fileList = [
+            create_color_block_image(tempPath / 'blue.png', (0, 0, 255), image_format='PNG'),
+            create_color_block_image(tempPath / 'red.jpeg', (255, 0, 0), image_format='JPEG'),
+        ]
+
+        # 生成PDF文件
+        pdfFile = tempPath / 'output.pdf'
+        await bfile.writeBytes(pdfFile, img2pdf.convert(fileList))  # type: ignore
+
+        yield pdfFile

test/test_proxy.py

@@ -0,0 +1,8 @@
+import pytest
+from beni import btask
+
+
+@pytest.mark.asyncio
+async def test_proxy():
+    result = btask.testCall('proxy', '--help')
+    assert result.exit_code == 0

test/test_wasabi.py

@@ -0,0 +1,28 @@
+from uuid import uuid4
+
+import pytest
+from beni import bfile, bpath, btask
+
+
+@pytest.mark.asyncio
+async def test_wasabi():
+    with bpath.useTempPath() as tempPath:
+        target = tempPath / uuid4().hex
+        file = target / uuid4().hex
+        content = uuid4().hex
+        password = uuid4().hex
+        await bfile.writeText(file, content)
+
+        result = btask.testCall('wasabi', 'zip', target.as_posix(), '--password', password)
+        assert result.exit_code == 0
+        assert target.is_file(), '生成加密文件失败'
+
+        newPassword = uuid4().hex
+        result = btask.testCall('wasabi', 'change-pass', target.as_posix(), '--password', password, '--new-password', newPassword)
+        assert result.exit_code == 0
+        assert target.is_file(), '修改密码失败'
+
+        result = btask.testCall('wasabi', 'unzip', target.as_posix(), '--password', newPassword)
+        assert result.exit_code == 0
+        assert target.is_dir(), '解密文件失败'
+        assert await bfile.readText(file) == content, '解密文件内容不一致'

bcmd-0.6.6.dist-info/RECORD

@@ -1,6 +0,0 @@
-bcmd/__init__.py,sha256=GP_60-6vImXqdMfC5vc4xlscWajx4OYmnlNXASWn19w,147
-bcmd-0.6.6.dist-info/METADATA,sha256=4bllONvfMpWoucDX_cdkz3rqND0yPbWBrtIupX99lHI,812
-bcmd-0.6.6.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-bcmd-0.6.6.dist-info/entry_points.txt,sha256=mriCeYh3wksKcqq3-LtzyFkSCIdN1uZc1IJwom-SW1s,34
-bcmd-0.6.6.dist-info/top_level.txt,sha256=-KrvhhtBcYsm4XhcjQvEcFbBB3VXeep7d3NIfDTrXKQ,5
-bcmd-0.6.6.dist-info/RECORD,,