根据 getopts 标志 bash 更改函数中的变量

编程问答 2022-05-26

问题描述

我有一个 bash 脚本，用于查找连接的设备并将串行端口输出记录到文件中。 我正在使用 getopts 来选择日志记录方法。 除了一行之外，两个函数是相同的。 我想将其更改为单个函数，其中包含不同的单行变量（我的尝试如下）但没有成功。我怎样才能用 getopts 做到这一点？ 非常感谢。

#!/bin/bash
#script to capture camera serial port logs via usb or UART.

file_name='DashcamLog'
#Date e.g: 20210204T120159 (ISO 8601)
current_date=$(date +%Y%m%dT%H%M%s)
BAUdratE='115200'
#change to reflect udev rules e.g /dev/h1*
BOARDS=(/dev/ttyUSB* /dev/ttyACM*)
screen="$(screen -Sdm $port_name -L -Logfile $new_file_name $i $BAUdratE)"
minicom="$(screen -Sdm ${port_name} minicom -b ${BAUdratE} -D ${i} -C ${new_file_name})"

usage() {
    echo "Usage:"
    echo "Use $0 -a to use GNU screen to automatically capture logs on USB plug in"
    echo "Use $0 -s to use GNU screen to capture logs WITHOUT automatic capture on USB plug in."
    echo "Use $0 -m to use minicom to capture logs WITHOUT automatic capture on USB plug in."

}

#detects current + new devices + automatically starts logging
autoStartLog() {

    while true;do

        current_date=$(date +%Y%m%dT%H%M%s)

        for i in $BOARDS; do

            port_name=${i#/dev/}

            if ! screen -ls | grep -o $port_name > /dev/null;then
                (
                    serial_no="$(udevadm info --attribute-walk $i | grep -m 1 ATTRS{serial})"
                    #file name e.g = DashcamLog_20210208T094013_peri_h1p1.log
                    new_file_name="${file_name}_${current_date}_${HOSTNAME}_${port_name}.log"
                    $screen
                    echo $port_name 'serial_no: ' $serial_no $new_file_name
                )
            fi

        done

    done

}

#detects current devices + starts logging using screen(-s)/minicom(-m)
startLog() {

    for i in $BOARDS; do
        (
            port_name=${i#/dev/}
            serial_no="$(udevadm info --attribute-walk $i | grep -m 1 ATTRS{serial})"
            new_file_name="${file_name}_${current_date}_${HOSTNAME}_${port_name}.log"
            $1
            echo $port_name 'serial_no: ' $serial_no $new_file_name
        )
    done

}


while getopts ":hmsa" opt; do
    case ${opt} in
        h)
          usage
          ;;

        a)
          autoStartLog
          ;;

        s)
          startLog $screen
          ;;

        m)
          startLog $minicom
          ;;

        \? )
          echo "Invalid Option: -$OPTARG" 1>&2
          usage
          exit 1
          ;;
    esac
done
shift $((OPTIND -1))

解决方法

我改变了一些东西，因为我意识到在评论中解释所有的小细节会太长。

代码如下：

#!/bin/bash

#script to capture camera serial port logs via usb or UART.

file_name='DashcamLog'
#Date e.g: 20210204T120159 (ISO 8601)
current_date=$(date +%Y%m%dT%H%M%S)
BAUDRATE='115200'
#change to reflect udev rules e.g /dev/h1*
BOARDS=(/dev/ttyUSB* /dev/ttyACM*)

usage()
{
    echo "Usage:"
    echo "Use $0 -a to use GNU screen to automatically capture logs on USB plug in"
    echo "Use $0 -s to use GNU screen to capture logs WITHOUT automatic capture on USB plug in."
    echo "Use $0 -m to use minicom to capture logs WITHOUT automatic capture on USB plug in."
    exit 1
}

#detects current devices + starts logging using screen(-s)/minicom(-m)
startLog()
{
    for i in "${BOARDS[@]}"
    do
        port_name="${i#/dev/}"
        serial_no=$(udevadm info --attribute-walk "$i" | grep -m 1 "ATTRS{serial}")
        new_file_name="${file_name}_${current_date}_${HOSTNAME}_${port_name}.log"
        case "$1" in
            "screen")
                echo "PUT THE SCREEN COMMAND HERE"
                ;;
            "minicom")
                echo "PUT THE MINICOM COMMAND HERE"
                ;;
            \?)
                echo "Invalid Option: -$OPTARG" 1>&2
                usage
                ;;
        esac
        echo "$port_name serial_no:  $serial_no $new_file_name"
    done
}

while getopts ":hms" opt
do
    case "$opt" in
        h)
            usage
            ;;
        s)
            startLog screen
            ;;
        m)
            startLog minicom
            ;;
        \?)
            echo "Invalid Option: -$OPTARG" 1>&2
            usage
            ;;
    esac
done
shift $((OPTIND -1))

详情：

  • 当您定义变量 screen 和 minicom 时，您有 "$( ... )"。 $() 中的命令会立即执行，而不是在您稍后放置 $1 时执行。这就是为什么我删除了那部分并将命令直接放在函数中的原因。
  • 在 startLog() 中，我重用了一个 case。也可以使用 if。
  • BOARDS 是一个数组。因此，当您执行 $BOARDS 时，不指定索引，您只会获得第一项。您必须执行 ${BOARDS[@]} 才能处理所有项目。
  • 如果您不知道，BOARDS=(/dev/ttyUSB* /dev/ttyACM*) 将扩展 *，因此 BOARDS 将被定义为所有文件 /dev/ttyUSB* 和所有文件 /dev/ttyACM*。它不会“留在”*。
  • 当您从另一个命令 (var=$(command)) 定义一个变量时，您不需要将 " " 放在它周围。这也令人困惑，因为您应该在 $() 内用双引号引用变量，因此您会在其他双引号内使用双引号。

我也把它放在我喜欢的代码风格中，但这是灵活的，受个人喜好的影响。

您可以将代码放在 https://www.shellcheck.net/ 中以验证语法。

bashgetoptsshshshshell