fix: 修复自动安装Platformio权限问题

fishros · Jul 10, 2024 · 2ef483f · 2ef483f
1 parent b3b86f3
commit 2ef483f
Showing 1 changed file with 32 additions and 20 deletions.
diff --git a/tools/tool_install_micros_fishbot_env.py b/tools/tool_install_micros_fishbot_env.py
@@ -4,6 +4,8 @@
 from .base import osversion, osarch
 from .base import run_tool_file
 
+
+
 class Tool(BaseTool):
     def __init__(self):
         self.name = "安装PlateformIO MicroROS开发环境(支持Fishbot)"
@@ -13,27 +15,37 @@ def __init__(self):
     def install_nodejs(self):
         CmdTask('sudo apt update && sudo apt install git python3-venv -y', os_command=True).run()
         PrintUtils.print_warn("注意:运行本指令前需要在VS Code 中安装 PlatformIO 插件后再运行本命令")
-        user_home = FileUtils.getusershome()[0]
-        PrintUtils.print_info("开始安装Platform IO~")
-        print(CmdTask('export HOME={} && $HOME.platformio/penv/bin/pip3 install -i https://pypi.tuna.tsinghua.edu.cn/simple platformio'.format(user_home),os_command=True).run())
-        PrintUtils.print_info("开始下载ESP32开发依赖库~")
-        PrintUtils.print_warn("下载不使用代理会很慢（大约4小时左右），建议运行一键安装14开启代理后，导出终端代理可10分钟装好")
-        PrintUtils.print_info("当前进度: 1/7 ")
-        CmdTask('export HOME={} && $HOME.platformio/penv/bin/pio pkg install --global --platform "platformio/espressif32@^6.4.0"'.format(user_home),os_command=True).run()
-        PrintUtils.print_info("当前进度: 2/7 ")
-        CmdTask('export HOME={} && $HOME.platformio/penv/bin/pio pkg install --global --tool "platformio/contrib-piohome"'.format(user_home), os_command=True).run()
-        PrintUtils.print_info("当前进度: 3/7 ")
-        CmdTask('export HOME={} && $HOME.platformio/penv/bin/pio pkg install --global --tool "platformio/framework-arduinoespressif32"'.format(user_home), os_command=True).run()
-        PrintUtils.print_info("当前进度: 4/7 ")
-        CmdTask('export HOME={} && $HOME.platformio/penv/bin/pio pkg install --global --tool "platformio/tool-scons"'.format(user_home), os_command=True).run()
-        PrintUtils.print_info("当前进度: 5/7 ")
-        CmdTask('export HOME={} && $HOME.platformio/penv/bin/pio pkg install --global --tool "platformio/tool-mkfatfs"'.format(user_home), os_command=True).run()
-        PrintUtils.print_info("当前进度: 6/7 ")
-        CmdTask('export HOME={} && $HOME.platformio/penv/bin/pio pkg install --global --tool "platformio/tool-mkspiffs"'.format(user_home), os_command=True).run()
-        PrintUtils.print_info("当前进度: 7/7 ")
-        CmdTask('export HOME={} && $HOME.platformio/penv/bin/pio pkg install --global --tool "platformio/tool-mklittlefs"'.format(user_home), os_command=True).run()
+        user_homes = FileUtils.getusershome()
+        user = FileUtils.getusers()[0]
+        user_home = user_homes[0]
+        # if len(user_homes)>1:
+        #     dic = {}
+        #     for i in range(len(user_homes)):
+        #         dic[i+1] = user_homes[i]
+        #     code,result = ChooseTask(dic,"请选择用户目录",array=True).run()    
 
-        PrintUtils.print_info("安装,接下来你可以到vscode里新建工程了~!")
+        PrintUtils.print_info("开始生成安装脚本~")
+        install_sh = """export HOME={} && $HOME.platformio/penv/bin/pip3 install -i https://pypi.tuna.tsinghua.edu.cn/simple platformio
+export http_proxy=http://fishros.org:8899 && export https_proxy=http://fishros.org:8899
+echo "开始下载ESP32开发依赖库~"
+echo "当前进度: 1/7"
+$HOME.platformio/penv/bin/pio pkg install --global --platform "platformio/espressif32"
+echo "当前进度: 2/7"
+$HOME.platformio/penv/bin/pio pkg install --global --tool "platformio/contrib-piohome"
+echo "当前进度: 3/7"
+$HOME.platformio/penv/bin/pio pkg install --global --tool "platformio/framework-arduinoespressif32"
+echo "当前进度: 4/7"
+$HOME.platformio/penv/bin/pio pkg install --global --tool "platformio/tool-scons"
+echo "当前进度: 5/7"
+$HOME.platformio/penv/bin/pio pkg install --global --tool "platformio/tool-mkfatfs"
+echo "当前进度: 6/7"
+$HOME.platformio/penv/bin/pio pkg install --global --tool "platformio/tool-mkspiffs"
+echo "当前进度: 7/7"
+$HOME.platformio/penv/bin/pio pkg install --global --tool "platformio/tool-mklittlefs"
+""".format(user_home)
+        FileUtils.new('/tmp/', 'install_pio.sh',install_sh)
+        CmdTask('sudo -u {} bash /tmp/install_pio.sh'.format(user), os_command=True).run()
+        PrintUtils.print_info("安装完成,接下来你可以到vscode里新建工程了~!")
 
     def run(self):
         self.install_nodejs()