bcmd 0.6.9__py3-none-any.whl → 0.6.10__py3-none-any.whl

bcmd/common/func.py

@@ -15,5 +15,5 @@ def checkPathOrNotExists(folder: Path):
 
 @contextmanager
 def useResources(name: str):
-    with importlib.resources.path('bcmdx.resources', name) as target:
+    with importlib.resources.path('bcmd.resources', name) as target:
         yield target

bcmd/tasks/__init__.py

@@ -3,6 +3,7 @@ from . import bin
 from . import code
 from . import crypto
 from . import debian
+from . import docs
 from . import download
 from . import image
 from . import json

bcmd/tasks/docs.py

@@ -0,0 +1,185 @@
+import asyncio
+import datetime
+import pickle
+import tkinter as tk
+from contextlib import asynccontextmanager
+from pathlib import Path
+from tkinter import messagebox
+from typing import Any, Final
+from uuid import uuid4
+
+import paramiko
+from beni import bcolor, bcrypto, bfile, bpath, brun, btask, bzip
+from beni.bform import BForm
+from beni.bfunc import syncCall
+from typer import Argument
+
+from bcmd.utils.utils import decryptFileIgnoreError
+
+app: Final = btask.newSubApp('Vitepress 网站相关')
+conf: dict[str, Any] = {}
+isUpload: bool = False
+password: str = ''
+now = datetime.datetime.now().strftime("%Y%m%d_%H%M%S")
+
+projectPath = bpath.get('')
+docsPath = bpath.get('')
+vitepressDistPath = bpath.get('')
+distPath = bpath.get('')
+deployPath = bpath.get('')
+zipFile = bpath.get('')
+
+
+@app.command()
+@syncCall
+async def build(
+    path: Path = Argument(Path.cwd(), help='workspace 路径'),
+):
+    '发布'
+    with bpath.useTempPath(True) as tempPath:
+        await init(path)
+        await userInput()
+        await vitepressBuild(tempPath / 'site')
+        await makeDeploy(tempPath / 'deploy')
+        for file in bpath.listPath(tempPath):
+            await bzip.sevenZip(zipFile, file)
+        if isUpload:
+            await upload()
+    bcolor.printGreen('OK')
+
+
+async def init(path: Path):
+    global projectPath, docsPath, vitepressDistPath, distPath, deployPath, conf, zipFile
+
+    # 初始化目录路径
+    projectPath = path
+    docsPath = projectPath / 'docs'
+    vitepressDistPath = docsPath / '.vitepress/dist'
+    distPath = projectPath / 'dist'
+    deployPath = projectPath / 'deploy'
+
+    # 清空打包用到的目录
+    bpath.remove(distPath)
+    bpath.make(distPath)
+    bpath.remove(vitepressDistPath)
+
+    # 更新配置
+    infoTomlFile = deployPath / 'info.toml'
+    if not infoTomlFile.exists():
+        btask.abort('部署文件不存在', infoTomlFile)
+    conf.update(await bfile.readToml(infoTomlFile))
+
+    # 整理特殊的字段
+    zipFile = distPath / f'{conf['domain']}_{now}.7z'
+    conf['upload_file_name'] = zipFile.name
+    conf['temp_path'] += f'/{uuid4()}'
+
+
+async def userInput():
+    global isUpload, password
+
+    class BuildForm(BForm):
+        def __init__(self):
+            super().__init__(title='发布网站')
+            self.varIsUpload = tk.BooleanVar(value=True)
+            self.varPassword = tk.StringVar(value='')
+            self.addLabel('网站', conf['domain'])
+            self.addCheckBox('上传服务器', '', self.varIsUpload)
+            self.addEntry('请输入密码', self.varPassword, width=20, command=self.onBtn, password=True, focus=True)
+
+        def onBtn(self):
+            password = self.varPassword.get()
+            try:
+                conf['server_info'] = bcrypto.decryptJson(conf['server_info'], password)
+                self.destroy()
+            except:
+                messagebox.showerror('密码错误', '密码错误，请重新输入')
+
+        def getResult(self):
+            return self.varIsUpload.get(), self.varPassword.get()
+
+    isUpload, password = BuildForm().run()
+
+    # 配置里面每个字段都尝试解密（实际效率较低但不影响使用）
+    for k, v in conf.items():
+        try:
+            conf[k] = bcrypto.decryptText(v, password)
+        except:
+            pass
+
+
+async def vitepressBuild(outputPath: Path):
+    with bpath.changePath(projectPath):
+        await brun.run('pnpm install')
+        await brun.run('pnpm docs:build')
+        bpath.copy(vitepressDistPath, outputPath)
+
+
+async def makeDeploy(outputPath: Path):
+    bpath.copy(projectPath / 'deploy', outputPath)
+    bpath.remove(outputPath / 'info.toml')
+
+    # 删除配置里面加密和删除非字符串的配置，剩下的内容用于替换文件名和文件内容
+    dataDict: dict[str, Any] = pickle.loads(pickle.dumps(conf))
+    for k in list(dataDict.keys()):
+        v = dataDict[k]
+        if not isinstance(v, str):
+            del dataDict[k]
+
+    # 文件名以及内容调整
+    fileList = bpath.listFile(outputPath, True)
+    await asyncio.gather(*[decryptFileIgnoreError(x, password) for x in fileList])
+    for file in fileList:
+
+        # 替换文件名
+        toFile = str(file)
+        for k, v in dataDict.items():
+            toFile = toFile.replace(f'{{{k}}}', str(v))
+        if toFile != str(file):
+            file.rename(toFile)
+            file = bpath.get(toFile)
+
+        # 替换文件内容
+        content = await bfile.readText(file)
+        oldContent = content
+        for k, v in dataDict.items():
+            content = content.replace(f'{{{k}}}', str(v))
+        if oldContent != content:
+            await bfile.writeText(file, content)
+
+    # 将 Scrips 里面的文件都抽离到 dist 目录
+    scriptsPath = outputPath / 'Scripts'
+    for file in bpath.listFile(scriptsPath):
+        bpath.move(file, distPath / f'{conf['domain']}_{now}_{file.stem}{file.suffix}')
+    bpath.remove(scriptsPath)
+
+
+@asynccontextmanager
+async def sshClient():
+    client = paramiko.SSHClient()
+    client.set_missing_host_key_policy(paramiko.AutoAddPolicy())
+    with bpath.useTempFile() as tempFile:
+        await bfile.writeText(tempFile, conf['server_key'])
+        client.connect(
+            **conf['server_info'],
+            key_filename=str(tempFile),
+        )
+        yield client
+        client.close()
+
+
+async def upload():
+    async with sshClient() as client:
+
+        def executeCmd(cmd: str):
+            _, stdout, _ = client.exec_command(f"bash -lc '{cmd}'")
+            print(stdout.read().decode())
+
+        sftp = client.open_sftp()
+        for file in bpath.listFile(distPath):
+            executeCmd(f'mkdir -p {conf['temp_path']}')
+            sftp.put(str(file), f'{conf['temp_path']}/{file.name}')
+        shFile = list(distPath.glob('*.sh'))[0]
+        executeCmd(f'sh {conf['temp_path']}/{shFile.name}')
+        executeCmd(f'rm -rf {conf['temp_path']}')
+        sftp.close()

bcmd/tasks/upgrade.py

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

bcmd/utils/utils.py

@@ -0,0 +1,9 @@
+from beni import bcrypto
+from beni.btype import XPath
+
+
+async def decryptFileIgnoreError(file: XPath, password: str):
+    try:
+        await bcrypto.decryptFile(file, password)
+    except:
+        pass

{bcmd-0.6.9.dist-info → bcmd-0.6.10.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: bcmd
-Version: 0.6.9
+Version: 0.6.10
 Summary: Commands for Beni
 Author-email: Beni Mang <benimang@126.com>
 Maintainer-email: Beni Mang <benimang@126.com>
@@ -11,6 +11,7 @@ Requires-Dist: async-lru==2.0.5
 Requires-Dist: benimang==0.8.6
 Requires-Dist: cryptography==45.0.4
 Requires-Dist: nest-asyncio==1.6.0
+Requires-Dist: paramiko==3.5.1
 Requires-Dist: pillow==11.2.1
 Requires-Dist: prettytable==3.16.0
 Requires-Dist: psutil==7.0.0

{bcmd-0.6.9.dist-info → bcmd-0.6.10.dist-info}/RECORD

@@ -1,13 +1,14 @@
 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/func.py,sha256=MehbfuWFHTOsihhYxVFj0_U6-hjMTfLh3n-oMrpyyKo,508
 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/__init__.py,sha256=_G8hO9P0UEKJRreXv8114XeTR8uFG8gRSjvA1WMv8f8,333
 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/docs.py,sha256=pVs9iy9fjo4DrWX-YfrxjnxLOO4VAW7w5NUqIZHxnVQ,5927
 bcmd/tasks/download.py,sha256=XdZYKi8zQTNYWEgUxeTNDqPgP7IGYJkMmlDDC9u93Vk,2315
 bcmd/tasks/image.py,sha256=_ck-WVfUlyQ2fZTpVPcpcurWSud7AkANKUuFjMW7MwA,14283
 bcmd/tasks/json.py,sha256=WWOyvcZPYaqQgp-Tkm-uIJschNMBKPKtZN3yXz_SC5s,635
@@ -17,16 +18,17 @@ bcmd/tasks/mirror.py,sha256=nAe8NYftMKzht16MFBj7RqXwvVhR6Jh2uuAyJLh87og,1098
 bcmd/tasks/pdf.py,sha256=fkHRgxqzrRxdb4_-9pL9wp2roqAHJPS_dVqAGJvRUsM,1504
 bcmd/tasks/proxy.py,sha256=KDOXtNRoY4n45tBrAgK-UM7GmvCCx3Ho545yxMc8PKQ,1602
 bcmd/tasks/time.py,sha256=ZiqA1jdgl-TBtFSOxxP51nwv4g9iZItmkFKpf9MKelk,2453
-bcmd/tasks/upgrade.py,sha256=z9Ein8U_Co7fsijPxyDFM8tXBFCp4bWATImwwwvC5ho,536
+bcmd/tasks/upgrade.py,sha256=nyhcl5oGAUnOOR8JJZW2jfepcVJ6O9gufK8VgxUeil0,543
 bcmd/tasks/wasabi.py,sha256=xWFAxprSIlBqDDMGaNXZFb-SahnW1d_R9XxSKRYIhnM,3110
 bcmd/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+bcmd/utils/utils.py,sha256=Oyf5ubxaCD_T2GOz3zJkR4AJzfVcfBqtmL8Gk8_gOJs,202
 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.9.dist-info/METADATA,sha256=dUP3hFm6hT4tvf13ubEiaPvjyA5tloVlls7RlQQiNTE,812
-bcmd-0.6.9.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-bcmd-0.6.9.dist-info/entry_points.txt,sha256=mriCeYh3wksKcqq3-LtzyFkSCIdN1uZc1IJwom-SW1s,34
-bcmd-0.6.9.dist-info/top_level.txt,sha256=fYY6tRrJ_G7tn24RXAG0M5ZKbcuaQznodfX1toFPSKs,10
-bcmd-0.6.9.dist-info/RECORD,,
+bcmd-0.6.10.dist-info/METADATA,sha256=x2T8YTvQ8vQxS0aiegxZNmPoMgqp76gyfeBpRYd79l4,845
+bcmd-0.6.10.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+bcmd-0.6.10.dist-info/entry_points.txt,sha256=mriCeYh3wksKcqq3-LtzyFkSCIdN1uZc1IJwom-SW1s,34
+bcmd-0.6.10.dist-info/top_level.txt,sha256=fYY6tRrJ_G7tn24RXAG0M5ZKbcuaQznodfX1toFPSKs,10
+bcmd-0.6.10.dist-info/RECORD,,