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

bcmd/common/func.py

@@ -0,0 +1,19 @@
+import importlib.resources
+from contextlib import contextmanager
+from pathlib import Path
+
+from beni import btask
+
+
+def checkFileOrNotExists(file: Path):
+    btask.assertTrue(file.is_file() or not file.exists(), f'必须是文件 {file}')
+
+
+def checkPathOrNotExists(folder: Path):
+    btask.assertTrue(folder.is_dir() or not folder.exists(), f'必须是目录 {folder}')
+
+
+@contextmanager
+def useResources(name: str):
+    with importlib.resources.path('bcmdx.resources', name) as target:
+        yield target

bcmd/common/secret.py

@@ -0,0 +1,51 @@
+import getpass
+from typing import Any, TypedDict, cast
+
+from async_lru import alru_cache
+from beni import bcrypto, btask
+from beni.bcolor import printRed
+
+
+class PypiSecret(TypedDict):
+    username: str
+    password: str
+
+
+@alru_cache
+async def getPypi(value: str = '') -> PypiSecret:
+    return cast(
+        Any,
+        _getData(value or 'QbuF2mV/lqovtF5dskZGD7qHknYbNuF2QseWRtWxLZTPrC/jL1tcxV8JEKaRjLsu46PxJZ7zepJwggnUTIWnEAoV5VtgP2/hbuzxxHha8817kR5c65H9fXm8eOal7DYXsUoGPQMnm59UWNXUKjmIaP4sn9nySFlRYqa8sEZSbYQ4N0NL35Dpj1e3wyQxJ+7h2jwKAz50Hh8G4yAM3/js9+NUe4ymts+UXcwsP3ADIBMkzjnFc0lEYg2d+fw0A74XWCvoZPoGqHZR/THUOVNAYxoGgDzP4SPIk1XsmtpxvfO/DpJd/Cg/0fB3MYagGKI1+m6Bxqhvd1I/lf0YbM5y4E4='),
+    )
+
+
+class QiniuSecret(TypedDict):
+    bucket: str
+    baseUrl: str
+    ak: str
+    sk: str
+
+
+@alru_cache
+async def getQiniu(value: str = '') -> QiniuSecret:
+    return cast(
+        Any,
+        _getData(value or 'vNroFKeKklrdcJ89suFm+iyuJsq/cyUB5+QWoeeiMc/J0oSLF9cg5rqbK1IRxF0cCQ8KmkQQhdVa+PI6kuTBhoSH6IviVTylzAOrJywEccz9jWkJkW28Y9Vo4ePZmfWf/j7wdxNB144z234KD8IxJn4lR2A0L9JN5kk1o1/hpcydXL74FNtt03lYL/E3WVcvpUfw37mri2HMYOfUw81dRwW35/hMuQjtq1BBrKrIsSKTHH44tROMcgyvt+Qy292AtDBcsYiZxBKhQtBFPMq/vUs='),
+    )
+
+
+def _getData(content: str) -> dict[str, Any]:
+    index = content.find(' ')
+    if index > -1:
+        tips = f'请输入密码（{content[:index]}）：'
+    else:
+        tips = '请输入密码：'
+    while True:
+        try:
+            pwd = getpass.getpass(tips)
+            return bcrypto.decryptJson(content, pwd)
+        except KeyboardInterrupt:
+            print('')
+            btask.abort('用户操作取消')
+        except BaseException:
+            printRed('密码错误')

bcmd/resources/project/main.py

@@ -0,0 +1 @@
+print('OK')

bcmd/tasks/__init__.py

@@ -0,0 +1,16 @@
+# type: ignore
+from . import bin
+from . import code
+from . import crypto
+from . import debian
+from . import download
+from . import image
+from . import json
+from . import lib
+from . import math
+from . import mirror
+from . import pdf
+from . import proxy
+from . import time
+from . import upgrade
+from . import wasabi

bcmd/tasks/bin.py

@@ -0,0 +1,103 @@
+from datetime import datetime
+from pathlib import Path
+from typing import Final
+
+import typer
+from beni import bcolor, bfile, bpath, btask, bzip
+from beni.bfunc import syncCall, textToAry
+from beni.bqiniu import QiniuBucket
+from beni.btype import Null
+from prettytable import PrettyTable
+
+from ..common import secret
+
+app: Final = btask.newSubApp('bin 工具')
+
+_PREFIX = 'bin/'
+
+
+@app.command()
+@syncCall
+async def download(
+    names: list[str] = typer.Argument(None, help="支持多个"),
+    file: Path = typer.Option(None, '--file', '-f', help="文件形式指定参数，行为单位"),
+    output: Path = typer.Option(Path.cwd(), '--output', '-o', help="本地保存路径"),
+    secretValue: str = typer.Option('', '--secret', '-s', help='密钥信息'),
+):
+    '从七牛云下载执行文件'
+    bucket: QiniuBucket = Null
+    if file:
+        content = await bfile.readText(Path(file))
+        names.extend(
+            textToAry(content)
+        )
+    for target in names:
+        binFile = output / target
+        if binFile.exists():
+            bcolor.printYellow(f'已存在 {binFile}')
+        else:
+            key = f'bin/{target}.zip'
+            bucket = bucket or await _getBucket(secretValue)
+            await bucket.downloadPrivateFileUnzip(key, output)
+            bcolor.printGreen(f'added  {binFile}')
+
+
+@app.command('list')
+@syncCall
+async def getList(
+    secretValue: str = typer.Option('', '--secret', '-s', help='密钥信息'),
+):
+    '列出可下载的文件'
+    bucket = await _getBucket(secretValue)
+    datas = (await bucket.getFileList(_PREFIX, limit=1000))[0]
+    datas = [x for x in datas if x.key != _PREFIX and x.key.endswith('.zip')]
+    datas.sort(key=lambda x: x.time, reverse=True)
+    table = PrettyTable()
+    table.add_column(
+        bcolor.yellow('文件名称'),
+        [x.key[len(_PREFIX):-len('.zip')] for x in datas],
+        'l',
+    )
+    table.add_column(
+        bcolor.yellow('上传时间'),
+        [datetime.fromtimestamp(x.time / 10000000).strftime('%Y-%m-%d %H:%M:%S') for x in datas],
+    )
+    print()
+    print(table.get_string())
+
+
+@app.command()
+@syncCall
+async def upload(
+    file: Path = typer.Argument(..., help="本地文件路径"),
+    force: bool = typer.Option(False, '--force', '-f', help="强制覆盖"),
+    secretValue: str = typer.Option('', '--secret', '-s', help='密钥信息'),
+):
+    '上传'
+    bucket = await _getBucket(secretValue)
+    key = f'{_PREFIX}{file.name}.zip'
+    if not force:
+        if await bucket.getFileStatus(key):
+            btask.abort('云端文件已存在，可以使用 --force 强制覆盖')
+    with bpath.useTempFile() as f:
+        bzip.zipFile(f, file)
+        await bucket.uploadFile(key, f)
+    bcolor.printGreen('OK')
+
+
+@app.command()
+@syncCall
+async def remove(
+    key: str = typer.Argument(..., help="云端文件key"),
+    secretValue: str = typer.Option('', '--secret', '-s', help='密钥信息'),
+):
+    bucket = await _getBucket(secretValue)
+    await bucket.deleteFiles(f'{_PREFIX}{key}.zip')
+    bcolor.printGreen('OK')
+
+
+# ------------------------------------------------------------------------------------
+
+
+async def _getBucket(secretValue: str):
+    return QiniuBucket(**await secret.getQiniu())

bcmd/tasks/code.py

@@ -0,0 +1,132 @@
+import asyncio
+import os
+from pathlib import Path
+from typing import Final
+
+import typer
+from beni import bfile, bpath, btask
+from beni.bcolor import printGreen, printMagenta, printYellow
+from beni.bfunc import syncCall
+
+app: Final = btask.newSubApp('code 工具')
+
+
+@app.command()
+@syncCall
+async def tidy_tasks(
+    tasks_path: Path = typer.Argument(Path.cwd(), help="tasks 路径"),
+):
+    '整理 task 项目中的 tasks/__init__.py'
+
+    initFile = tasks_path / '__init__.py'
+    btask.assertTrue(initFile.is_file(), '文件不存在', initFile)
+    files = bpath.listFile(tasks_path)
+    files = [x for x in files if not x.name.startswith('_')]
+    contents = [f'from . import {x.stem}' for x in files]
+    contents.insert(0, '# type: ignore')
+    contents.append('')
+    content = '\n'.join(contents)
+    oldContent = await bfile.readText(initFile)
+    if oldContent != content:
+        await bfile.writeText(
+            initFile,
+            content,
+        )
+        printYellow(initFile)
+        printMagenta(content)
+    printGreen('OK')
+
+
+@app.command()
+@syncCall
+async def tidy_modules(
+    modules_path: Path = typer.Argument(Path.cwd(), help="modules_path 路径"),
+):
+    '整理 fastapi 项目中的 modules/__init__.py'
+
+    importContents: list[str] = []
+    managerContents: list[str] = []
+
+    xxdict: dict[str, set[Path]] = {}
+    for file in sorted(modules_path.glob('**/*Manager.py')):
+        if file.parent == modules_path:
+            subName = '.'
+        elif file.parent.parent == modules_path:
+            subName = f'.{file.parent.stem}'
+        else:
+            continue
+        xxdict.setdefault(subName, set()).add(file)
+    for subName in sorted(xxdict.keys()):
+        files = sorted(xxdict[subName])
+        importContents.append(f'from {subName} import {", ".join([x.stem for x in files])}')
+    managerContents.extend([f'    {x.stem},' for x in sorted([y for x in xxdict.values() for y in x])])
+
+    managerContents = [x for x in managerContents if x]
+    contents = [
+        '\n'.join(importContents),
+        'managers = [\n' + '\n'.join(managerContents) + '\n]',
+    ]
+    content = '\n\n'.join(contents) + '\n'
+    file = modules_path / '__init__.py'
+    printYellow(str(file))
+    printMagenta(content)
+    await bfile.writeText(file, content)
+    printGreen('OK')
+
+
+@app.command()
+@syncCall
+async def gen_init_py(
+    workspace_path: Path = typer.Argument(Path.cwd(), help='workspace 路径'),
+):
+    '递归生成 __init__.py 文件'
+
+    ignoreSubDirs = [
+        '.git',
+        'venv',
+        'node_modules',
+        '.pytest_cache',
+        '__pycache__',
+        '.vscode',
+    ]
+    folderList = bpath.listDir(workspace_path, True)
+    # 剔除子目录是这些的文件 .git venv ...
+    folderList = [x for x in folderList if not any([y in x.parts for y in ignoreSubDirs])]
+    for folder in folderList:
+        pyInitFile = folder / '__init__.py'
+        if not pyInitFile.exists():
+            printYellow(pyInitFile)
+            await bfile.writeText(pyInitFile, '')
+    printGreen('OK')
+
+
+@app.command()
+@syncCall
+async def to_lf(
+    path: Path = typer.Option(None, '--path', help='指定目录或具体图片文件，默认当前目录'),
+):
+    '将所有文件转换为 LF 格式'
+    ignoreSubDirs = [
+        '.git',
+        'venv',
+        'node_modules',
+        '.pytest_cache',
+        '__pycache__',
+    ]
+    path = path or Path(os.getcwd())
+    files = bpath.listFile(path, True)
+    # 剔除子目录是这些的文件 .git venv ...
+    files = [x for x in files if not any([y in x.parts for y in ignoreSubDirs])]
+
+    async def convertFile(file: Path):
+        try:
+            content = await bfile.readText(file)
+            if '\r\n' in content:
+                content = content.replace('\r\n', '\n')
+                await bfile.writeText(file, content)
+                printYellow(file)
+        except:
+            pass
+
+    await asyncio.gather(*[convertFile(file) for file in files])
+    printGreen('OK')

bcmd/tasks/crypto.py

@@ -0,0 +1,105 @@
+import getpass
+import json
+from pathlib import Path
+from typing import Final
+
+import pyperclip
+import typer
+from beni import bcolor, bcrypto, btask
+from beni.bfunc import syncCall
+from beni.binput import genPassword
+from rich.console import Console
+
+app: Final = btask.newSubApp('加密（v2）')
+
+
+@app.command()
+@syncCall
+async def encrypt_text():
+    '加密文本（使用剪贴板内容）'
+    content = pyperclip.paste()
+    assert content, '剪贴板内容不能为空'
+    bcolor.printGreen(content)
+    password = genPassword()
+    result = bcrypto.encryptText(content, password)
+    pyperclip.copy(result)
+    print('密文已复制到剪贴板')
+    bcolor.printYellow(result)
+
+
+@app.command()
+@syncCall
+async def decrypt_text():
+    '解密文本（使用剪贴板内容）'
+    content = pyperclip.paste().strip()
+    bcolor.printYellow(content)
+    while True:
+        try:
+            password = getpass.getpass('输入密码：')
+            result = bcrypto.decryptText(content, password)
+            print('解密成功')
+            bcolor.printGreen(result)
+            return
+        except KeyboardInterrupt:
+            break
+        except BaseException:
+            pass
+
+
+@app.command()
+@syncCall
+async def encrypt_json():
+    '生成JSON密文（使用剪贴板内容）'
+    content = pyperclip.paste()
+    try:
+        data = json.loads(content)
+    except:
+        return btask.abort('错误：剪贴板内容必须是JSON格式', content)
+    Console().print_json(data=data, indent=4, ensure_ascii=False, sort_keys=True)
+    password = genPassword()
+    result = bcrypto.encryptJson(data, password)
+    pyperclip.copy(result)
+    print('密文已复制到剪贴板')
+    bcolor.printYellow(result)
+
+
+@app.command()
+@syncCall
+async def decrypt_json():
+    '还原JSON密文内容（使用剪贴板内容）'
+    content = pyperclip.paste().strip()
+    bcolor.printYellow(content)
+    while True:
+        try:
+            password = getpass.getpass('输入密码：')
+            data = bcrypto.decryptJson(content, password)
+            Console().print_json(data=data, indent=4, ensure_ascii=False, sort_keys=True)
+            return
+        except KeyboardInterrupt:
+            break
+        except BaseException:
+            pass
+
+
+@app.command()
+@syncCall
+async def encrypt_file(
+    file: Path = typer.Argument(..., help='指定需要加密的文件')
+):
+    '加密文件（文件路径使用剪贴板内容）'
+    assert file.is_file(), '文件不存在'
+    password = genPassword()
+    await bcrypto.encryptFile(file, password)
+    bcolor.printGreen('OK')
+
+
+@app.command()
+@syncCall
+async def decrypt_file(
+    file: Path = typer.Argument(..., help='指定需要解密的文件')
+):
+    '解密文件（文件路径使用剪贴板内容）'
+    assert file.is_file(), '文件不存在'
+    password = getpass.getpass('输入密码：')
+    await bcrypto.decryptFile(file, password)
+    bcolor.printGreen('OK')

bcmd/tasks/debian.py

@@ -0,0 +1,78 @@
+import os
+from typing import Final
+
+from beni import bcolor, bfile, bpath, btask, binput
+from beni.bfunc import syncCall
+
+app: Final = btask.newSubApp('Debian 系统管理')
+
+
+@app.command()
+@syncCall
+async def install_python():
+    '安装python脚本'
+    pythonZipFile = ''
+    pythonName = ''
+    pythonVersion = ''
+    bcolor.printYellow(f'官网地址：https://www.python.org/downloads/source/')
+    bcolor.printYellow(f'先从官网上下载 python 源码压缩包（.tar.xz）')
+    while True:
+        pythonZipFile = input('输入python压缩文件名（如：Python-3.12.1.tar.xz）：').strip()
+        if pythonZipFile.endswith('.tar.xz'):
+            pythonName = pythonZipFile.replace('.tar.xz', '')
+            pythonVersion = '.'.join(pythonName.split('-')[-1].split('.')[:-1])
+            break
+        else:
+            print('输入有误，请重新输入')
+    cmdList: list[str] = [
+        f'apt update',
+        f'apt install build-essential zlib1g-dev libncurses5-dev libgdbm-dev libnss3-dev libssl-dev libsqlite3-dev libreadline-dev libffi-dev libbz2-dev -y',
+        f'tar -xf {pythonZipFile}',
+        f'cd {pythonName}',
+        f'./configure --enable-optimizations',  # 配置编译选项
+        f'make -j `nproc`',                     # 编译源代码，`nproc` 会利用所有可用的 CPU 核心加快编译速度
+        f'make altinstall',                     # 安装 Python，altinstall 避免替换默认的 python 命令
+        f'cd ..',
+        f'rm -rf {pythonName}',
+        f'rm -rf /usr/local/bin/python',
+        f'ln -s /usr/local/bin/python{pythonVersion} /usr/local/bin/python',
+        f'rm -rf /usr/local/bin/pip',
+        f'ln -s /usr/local/bin/pip{pythonVersion} /usr/local/bin/pip',
+        f'python --version',
+        f'pip --version',
+        f'pip install pipx',
+        f'pipx install bcmd',
+    ]
+    shFile = bpath.desktop(f'install-python-{pythonName}.sh')
+    await bfile.writeText(shFile, '\n'.join(cmdList))
+    bcolor.printGreen(f'将以下两个文件拷贝到服务器上，然后执行 sh {shFile.name} 即可安装 python')
+    print(pythonZipFile)
+    print(shFile)
+
+
+@app.command()
+@syncCall
+async def install_software():
+    '安装常用软件'
+    await binput.confirm('即将安装这些软件（7z nginx mariadb），是否确认？')
+    cmdList: list[str] = [
+        f'echo 更新软件包列表',
+        f'apt update',
+        f'echo 安装 7z',
+        f'apt install p7zip-full -y',
+        f'echo 安装 nginx',
+        f'apt install nginx -y',
+        f'systemctl start nginx',
+        f'systemctl enable nginx',
+        f'systemctl status nginx',
+        f'echo 安装 MariaDB',
+        f'apt install mariadb-server -y',
+        f'systemctl start mariadb',
+        f'systemctl enable mariadb',
+        f"sed -i 's/bind-address/# bind-address/' /etc/mysql/mariadb.conf.d/50-server.cnf"
+        f'systemctl restart mariadb',
+        f'systemctl status mariadb',
+
+    ]
+    for cmd in cmdList:
+        os.system(cmd)

bcmd/tasks/download.py

@@ -0,0 +1,74 @@
+import asyncio
+import os
+from pathlib import Path
+from typing import Final
+from urllib.parse import urlparse
+from uuid import uuid4
+
+import pyperclip
+import typer
+from beni import bcolor, bfile, bhttp, binput, bpath, btask
+from beni.bfunc import syncCall, textToAry
+
+app: Final = btask.app
+
+
+@app.command()
+@syncCall
+async def download(
+    url_file: Path = typer.Option(None, '--file', '-f', help='需要下载的url文件路径，默认使用剪贴板内容'),
+    save_path: Path = typer.Option(None, '--path', '-p', help='下载存放目录，默认当前目录'),
+    keep_directory: bool = typer.Option(False, '--keep', '-k', help='保持原始目录结构，默认不保持'),
+):
+    '下载资源资源文件'
+    save_path = save_path or Path(os.getcwd())
+
+    if url_file:
+        if not url_file.exists():
+            btask.abort('指定文件不存在', url_file)
+        content = await bfile.readText(url_file)
+    else:
+        content = pyperclip.paste()
+    urlSet = set(textToAry(content))
+
+    for i, url in enumerate(urlSet):
+        print(f'{i + 1}. {url}')
+    print(f'输出目录：{save_path}')
+    await binput.confirm('是否确认？')
+
+    fileSet: set[Path] = set()
+    retryUrlSet: set[str] = set()
+
+    async def download(url: str):
+        urlPath = urlparse(url).path
+        if keep_directory:
+            file = bpath.get(save_path, '/'.join([x for x in urlPath.split('/') if x]))
+        else:
+            file = save_path / Path(urlPath).name
+        if file in fileSet:
+            file = file.with_stem(f'{file.stem}--{uuid4()}')
+        fileSet.add(file)
+        try:
+            bcolor.printGreen(url)
+            await bhttp.download(url, file)
+        except:
+            retryUrlSet.add(url)
+            bcolor.printRed(url)
+
+    await asyncio.gather(*[download(x) for x in urlSet])
+
+    for i in range(4):
+        if i > 0:
+            print(f'等待重试第 {i} 次')
+            await asyncio.sleep(3)
+        await asyncio.gather(*[download(x) for x in urlSet])
+        if not retryUrlSet:
+            break
+        urlSet = set(retryUrlSet)
+        retryUrlSet.clear()
+
+    if retryUrlSet:
+        pyperclip.copy('\n'.join(retryUrlSet))
+        bcolor.printYellow('部分下载失败，失败部分已复制到剪贴板')
+    else:
+        bcolor.printGreen('OK')