我写了一个简单的 bash 脚本,它从文件或 STDIN 中转储随机行:
#!/bin/bash
if [ $# -ne 1 ]
then
echo "Syntax: $0 FILE (or \'-\' for STDIN)"
echo $0 - display a random line from FILE
exit 1
fi
RAND=`cat /proc/sys/kernel/random/uuid | cut -c1-4 | od -d | head -1 | cut -d' ' -f2`
if [ $1 != "-" ]
then
LINES=`cat "$1" | wc -l`
LINE=`expr $RAND % $LINES + 1`
head -$LINE $1 | tail -1
else
piped=`cat -`
LINES=`echo "$piped" | wc -l`
LINE=`expr $RAND % $LINES + 1`
echo "$piped" | head -$LINE | tail -1
fi
但是,如果没有传递任何选项,我希望它也处理 STDIN(但如果没有来自管道的 STDIN,仍然会失败并获得帮助)。
也就是说,我想说:
echo "foo\nbar\nbaz" | randline
...而不是...
echo "foo\n\bar\nbaz" | randline -
如何做到这一点?
编辑:
感谢杜恩!
#!/bin/bash
if [ "$( tty )" == 'not a tty' ]
then
STDIN_DATA_PRESENT=1
else
STDIN_DATA_PRESENT=0
fi
if [[ $# -ne 1 && $STDIN_DATA_PRESENT -eq 0 ]]
then
echo "Syntax: $0 [FILE (or \'-\' for STDIN)]"
echo $0 - display a random line from FILE
echo -e "\nWill also process piped STDIN if no arguments are given."
exit 1
fi
RAND=`cat /proc/sys/kernel/random/uuid | cut -c1-4 | od -d | head -1 | cut -d' ' -f2`
if [[ $1 && $1 != "-" ]]
then
LINES=`cat "$1" | wc -l`
LINE=`expr $RAND % $LINES + 1`
head -$LINE $1 | tail -1
else
piped=`cat -`
LINES=`echo "$piped" | wc -l`
LINE=`expr $RAND % $LINES + 1`
echo "$piped" | head -$LINE | tail -1
fi
最佳答案
要从文件中获取随机行,您可以这样做:
awk 'BEGIN{srand();}{printf "%04d %s\n", int(rand()*10000), $0}' < $FILENAME | sort | cut -f2- -d' ' | head -1
我们这样做:
- 在每一行的开头添加随机数
- 排序输出
- 去掉每行开头的随机数
- 打印第一行(混合输出)
顺便说一句,如果你使用的是 BASH,你可以使用 $RANDOM(由 bash 提供)变量而不是你自己生成 $RAND。
关于来自管道的 Bash 输入,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/5710957/