4 - 带bash脚本的python项目调试（stable diffsuion）

4 - 带bash脚本的python项目调试（stable diffsuion）

• 项目中使用bash脚本运行python代码，如何调试。以stable diffsuion webui forge为例，使用vscode。

4.1 安装包debugpy包

• 注意你使用的是本地环境还是虚拟环境

pip install debugpy

4.2 修改配置文件

• vscode中ctrl + p，输入>lanuch.json回车，打开配置文件，添加如下代码：

  {
  	"version": "0.2.0",
  	"configurations": [
  
  		{
  			"name": "Python: Attach",
  			"type": "debugpy",
  			"request": "attach",
  			"connect": {
  				"host": "localhost",
  				"port": xxxx //此处为空闲端口号
  			}
  		}
  	]
  } 

4.3 修改bash启动脚本

• 将启动脚本的运行语句改成：

  python -m debugpy --listen localhost:xxxx --wait-for-client \
          main.py \ # 要运行的文件
          --参数等等 \

  • 以stable diffusion webui forge为例，修改webui.sh:

    • 开头加上：

      #!/usr/bin/env bash
      #################################################
      # Please do not make any changes to this file,  #
      # change the variables in webui-user.sh instead #
          #################################################
      #-----------------------------------------------
      
      ifdebug=false
      debug_port=29000
      
      # 解析命令行参数并删除指定参数
      args=("$@")
      filtered_args=()
      for arg in "${args[@]}"; do
      case "$arg" in
              -ifdebug=*)
                  ifdebug="${arg#*=}"
                  ;;
              -debug_port=*)
                  debug_port="${arg#*=}"
                  ;;
              *)
                  filtered_args+=("$arg")
                  ;;
          esac
      done
      # 更新 $@ 以只包含 filtered_args 中的值
      set -- "${filtered_args[@]}"
      echo "ifdebug: $ifdebug"
      echo "debug_port: $debug_port"
      echo "remaining args: $@"
      #-----------------------------------------------

    • 翻到最后，将"${python_cmd}" -u "${LAUNCH_SCRIPT}" "$@"改成：

      if [[ "${ifdebug}" = "true" ]]; then
          "${python_cmd}" -m debugpy --listen "${debug_port}" --wait-for-client "${LAUNCH_SCRIPT}" "$@"
      else
          "${python_cmd}" -u "${LAUNCH_SCRIPT}" "$@"
      fi

4.4 运行

• 在vscode中打开终端，运行bash脚本

• 运行后点击调试，python: Remote Attach

  

  

赏

谢谢你请我吃糖果