2010年12月31日 星期五

使用指令行參數

使用方法:ruby MyScript.rb InputFile.csv SortedOutputFile.csv
執行後:產生test.rb檔案

MyScript.rb程式碼


#除非取得兩個參數,否則結束程式
unless ARGV.length == 2
puts "Dude, not the right number of arguments."
puts "Usage: ruby MyScript.rb InputFile.csv SortedOutputFile.csv\n"
exit
end

#輸入的檔案當作第一個參數
input_file = ARGV[0]
print input_file #print aa
#輸出的檔案當作第二個參數
output_file = ARGV[1]

#寫入檔案test.rb
File.open('test.rb', 'w') do |f2|
# use "\n" for two lines of text
f2.puts input_file+"\n"
f2.puts output_file+"\n"
f2.puts "Created by Satish\nThank God!"
end
#讀取檔案test.rb
File.open('test.rb', 'r') do |f1|
while line = f1.gets
puts line
end
end

沒有留言: