Question

我想在Ruby中处理命令行输入：

> cat input.txt | myprog.rb
> myprog.rb < input.txt
> myprog.rb arg1 arg2 arg3 ...

最好的方法是什么？特别是我想处理空白STDIN，我希望有一个优雅的解决方案。

#!/usr/bin/env ruby

STDIN.read.split("\n").each do |a|
   puts a
end

ARGV.each do |b|
    puts b
end

Answer 1

以下是我在晦涩的Ruby系列中找到的一些东西。

因此，在Ruby中，Unix命令cat的一个简单的no-bells实现将是：

#!/usr/bin/env ruby
puts ARGF.read

在输入方面，

ARGF是你的朋友;它是一个虚拟文件，从命名文件或STDIN中获取所有输入。

ARGF.each_with_index do |line, idx|
    print ARGF.filename, ":", idx, ";", line
end

# print all the lines in every file passed via command line that contains login
ARGF.each do |line|
    puts line if line =~ /login/
end

谢天谢地，我们没有在Ruby中获得菱形运算符，但我们确实得到ARGF作为替代。虽然模糊不清，但它实际上证明是有用的。考虑一下这个程序，该程序会在命令行中提到的每个文件上就地插入版权标题（感谢另一个Perlism，-i）：

#!/usr/bin/env ruby -i

Header = DATA.read

ARGF.each_line do |e|
  puts Header if ARGF.pos - e.length == 0
  puts e
end

__END__
#--
# Copyright (C) 2007 Fancypants, Inc.
#++

感谢：

Answer 2

Ruby提供了另一种处理STDIN的方法：-n标志。它将整个程序视为在STDIN上的循环中（包括作为命令行args传递的文件）。参见例如以下1行脚本：

#!/usr/bin/env ruby -n

#example.rb

puts "hello: #{$_}" #prepend 'hello:' to each line from STDIN

#these will all work:
# ./example.rb < input.txt
# cat input.txt | ./example.rb
# ./example.rb input.txt

Answer 3

我不太确定你需要什么，但我会用这样的东西：

#!/usr/bin/env ruby

until ARGV.empty? do
  puts "From arguments: #{ARGV.shift}"
end

while a = gets
  puts "From stdin: #{a}"
end

请注意，因为ARGV数组在第一个gets之前是空的，所以Ruby不会尝试将参数解释为从中读取的文本文件（从Perl继承的行为）。

如果stdin为空或没有参数，则不打印任何内容。

几个测试用例：

$ cat input.txt | ./myprog.rb
From stdin: line 1
From stdin: line 2

$ ./myprog.rb arg1 arg2 arg3
From arguments: arg1
From arguments: arg2
From arguments: arg3
hi!
From stdin: hi!

Answer 4

或许这样的事情？

#/usr/bin/env ruby

if $stdin.tty?
  ARGV.each do |file|
    puts "do something with this file: #{file}"
  end
else
  $stdin.each_line do |line|
    puts "do something with this line: #{line}"
  end
end

示例：

> cat input.txt | ./myprog.rb
do something with this line: this
do something with this line: is
do something with this line: a
do something with this line: test
> ./myprog.rb < input.txt 
do something with this line: this
do something with this line: is
do something with this line: a
do something with this line: test
> ./myprog.rb arg1 arg2 arg3
do something with this file: arg1
do something with this file: arg2
do something with this file: arg3

Answer 5

while STDIN.gets
  puts $_
end

while ARGF.gets
  puts $_
end

这是受Perl启发的：

while(<STDIN>){
  print "$_\n"
}

Answer 6

我要补充一点，为了将ARGF与参数一起使用，您需要在调用ARGV之前清除ARGF.each。这是因为ARGF会将ARGV中的任何内容视为文件名并首先从那里读取行。

以下是'tee'实施示例：

File.open(ARGV[0], 'w') do |file|
  ARGV.clear

  ARGF.each do |line|
    puts line
    file.write(line)
  end
end

Answer 7

我这样做：

all_lines = ""
ARGV.each do |line|
  all_lines << line + "\n"
end
puts all_lines

Answer 8

快速而简单：

STDIN.gets.chomp == 'YES'

Answer 9

您还可以使用STDIN.each_line和STDIN.each_line.to_a将其作为数组获取。

例如

STDIN.each_line do |line|
  puts line
end

Answer 10

似乎大多数答案都假设参数是包含内容的文件名，而不是stdin。以下所有内容都被视为参数。如果STDIN来自TTY，则忽略它。

$ cat tstarg.rb

while a=(ARGV.shift or (!STDIN.tty? and STDIN.gets) )
  puts a
end

参数或stdin可以为空或有数据。

$ cat numbers 
1
2
3
4
5
$ ./tstarg.rb a b c < numbers
a
b
c
1
2
3
4
5

Ruby中STDIN的最佳实践？

10 个答案: