แนวทางปฏิบัติที่ดีที่สุดกับ STDIN ใน Ruby?

ฉันต้องการจัดการกับอินพุตบรรทัดคำสั่งใน 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

ต่อไปนี้เป็นบางสิ่งที่ฉันพบในคอลเล็กชันทับทิมที่ไม่ชัดเจน

ดังนั้นใน Ruby การใช้คำสั่ง Unix แบบไม่มีเบลล์อย่างง่ายcatจะเป็น:

#!/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

ขอบคุณพระเจ้าที่เราไม่ได้รับผู้ประกอบการเพชรในทับทิม แต่เราได้รับ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.
#++

เครดิตไปที่:

• http://www.oreillynet.com/ruby/blog/2007/04/trivial_scripting_with_ruby.html#comment-565558
• http://blog.nicksieger.com/articles/2007/10/06/obscure-and-ugly-perlisms-in-ruby

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

ฉันไม่แน่ใจว่าคุณต้องการอะไร แต่ฉันจะใช้สิ่งนี้:

#!/usr/bin/env ruby

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

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

โปรดทราบว่าเนื่องจากอาร์เรย์ ARGV ว่างเปล่าก่อนหน้าgetsนี้ทับทิมจะไม่พยายามตีความอาร์กิวเมนต์เป็นไฟล์ข้อความที่จะอ่าน (พฤติกรรมที่สืบทอดมาจาก 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!

บางทีสิ่งนี้บางที

#/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

while STDIN.gets
  puts $_
end

while ARGF.gets
  puts $_
end

นี่คือแรงบันดาลใจจาก Perl:

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

ง่ายและรวดเร็ว:

STDIN.gets.chomp == 'YES'

ฉันจะเพิ่มว่าเพื่อที่จะใช้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

ฉันทำอะไรแบบนี้

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

ดูเหมือนว่าคำตอบส่วนใหญ่สมมติว่าอาร์กิวเมนต์เป็นชื่อไฟล์ที่มีเนื้อหาที่จะ catd ไปยัง 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