如何将本地 LoCyanPureFrp 导入到 NyaLCF 中

1
from tqdm import tqdm
2

3
import io
4
import aiofiles
5
import zipfile
6
import aiohttp
7
import asyncio
8
import shutil
9
import os
10
import sys
11
import time
12

13

14
async def download_file(url, file_name):
15
    """
16
    异步下载文件到指定路径。
17

18
    :param url: 文件的URL地址
19
    :param file_name: 保存文件的本地路径及名称
20
    """
21
    async with aiohttp.ClientSession() as session:
22
        async with session.get(url) as response:
23
            if response.status == 200:
24
                # 获取文件总大小
25
                total_size = int(response.headers.get("content-length", 0))
26
                # 初始化tqdm进度条，total为文件总大小
27
                with tqdm(
28
                        total=total_size, unit="B", unit_scale=True, desc=file_name
29
                ) as pbar:
30
                    # 使用异步写入方式将文件内容写入本地
31
                    with open(file_name, "wb") as f:
32
                        downloaded = 0
33
                        while True:
34
                            chunk = await response.content.read(1024)  # 每次读取1KB数据
35
                            if not chunk:
36
                                break
37
                            f.write(chunk)
38
                            downloaded += len(chunk)  # 累加已下载字节数
39
                            pbar.update(len(chunk))
40
                        print(f"文件 {file_name} 下载完成。")
41

42

43
async def main():
44
    url = "https://api.ymbit.cn/public/files/locyanfrp/frp_LoCyanFrp-0.51.3_windows_amd64.zip"
45
    file_name = "frp_LoCyanFrp-0.51.3_windows_amd64.zip"
46
    await download_file(url, file_name)
47
    await unzip_file(file_name)
48

49

50
async def unzip_file(file_name):
51
    async with aiofiles.open(file_name, "rb") as f:
52
        zip_data = await f.read()
53
    zip_ref = zipfile.ZipFile(io.BytesIO(zip_data), "r")
54
    for name in zip_ref.namelist():
55
        zip_ref.extract(name)
56
    zip_ref.close()  # 关闭 zip 文件对象
57

58

59
if __name__ == "__main__":
60
    asyncio.run(main())
61

62

63
def set_path():
64
    src_file = ".\\frp_LoCyanFrp-0.51.3_windows_amd64\\frpc.exe"
65
    path_set = os.system(f'set NYA_LCF_FRPC_PATH = "{src_file}"')
66
    if path_set == 0:
67
        print("环境变量 NYA_LCF_FRPC_PATH 设置成功")
68
    else:
69
        print("环境变量 NYA_LCF_FRPC_PATH 设置失败，使用文件移动方式更换版本，确认你的先前版本为 v0.51.3")
70
        time.sleep(3)
71

72
        def move_file(src_path, dst_path):
73
            try:
74
                shutil.move(src_path, dst_path)
75
                print(f"文件 {src_path} 成功移动到 {dst_path}")
76
                sys.exit(1)
77
            except shutil.Error as e:
78
                print(f"移动文件时出错: {str(e)}")
79
                sys.exit(1)
80
            except FileNotFoundError as fnf_error:
81
                print(f"源文件 {src_path} 未找到: {str(fnf_error)}")
82
                sys.exit(1)
83
            except PermissionError as perm_error:
84
                print(f"无权限移动文件: {str(perm_error)}")
85
                sys.exit(1)
86
            except Exception as generic_error:
87
                print(f"其他错误: {str(generic_error)}")
88
                sys.exit(1)
89

90
        src_file = ".\\frp_LoCyanFrp-0.51.3_windows_amd64\\frpc.exe"
91
        dst_dir = "%appdata%\\moe.xmcn.nyanana\\nyanana\\frpc\\0.51.3-2"
92
        dst_file = dst_dir + "frpc.exe"  # 目标文件路径   ，假设目标目录已存在
93

94
        # 确保目标目录存在
95
        os.makedirs(dst_dir, exist_ok=True)
96
        move_file(src_file, dst_file)