File I/O & Term replacement in Ruby - 파일 내용 치환 및 저장에 관한 예제

Reference Item

---

• Argument 사용
• String class method : gsub() , include? , chop
• Dir 및 File 클래스 사용법
• Random shuffle in Array

Senario

---

지정된 경로의 파일을 모두 읽어와서, 지정된 글자가 발견되면 그에 대응하는 다른 문자로 치환하여 지정된 갯수만큼 파일에 저장함.

 

 Source

---

#!/usr/bin/ruby ###################################################################### # Global variables and functions ###################################################################### $replacedTag = { "<title>" => "<Title>" , "<content>" => "<Content>" } # replacedTag $usageText = <<USAGE_TEXT ==================================================================================== SCD Creator ==================================================================================== < Usage > scd-creator [sourceDir] [targetSCD] [numberOfDocuments] < P.S > Before using this script code, please modify the replacedTag variable. ==================================================================================== USAGE_TEXT def shuffleArray( arr ) n = arr.length for i in 0...n r = Kernel.rand(n-i) + i arr[r] , arr[i] = arr[i] , arr[r] end end # shuffleArray() # ----------------------------- [ Parameter check if ARGV.size != 3 puts $usageText exit -1 end # if ARGV sourceDIR = ARGV[0] targetSCD = ARGV[1] numOfDocs = Integer(ARGV[2]) # ----------------------------- [ Get all the file names scriptFile = __FILE__ scriptFile = scriptFile.gsub("./" , "") sourceDIR = sourceDIR.chop if (sourceDIR[sourceDIR.length() -1] == 47 ) # 47 == / currentDir = Dir.new(sourceDIR) scdFileList = Array.new currentDir.each { |filename| filename = sourceDIR + "/" + filename scdFileList.push( filename ) unless filename.include? scriptFile } # end - d.each # ----------------------------- [ open target file targetFile = File.open( targetSCD , "w" ) currentDocNum = -1 # ----------------------------- [ shuffle the scdFileList shuffleArray( scdFileList ) # ----------------------------- [ replace the tags scdFileList.each { |currentFile| begin next if File.directory? currentFile puts "Current File is : " + currentFile File.open( currentFile ) do |file| file.each_line do |line| if ( line.include? "<DOCID>" ) currentDocNum += 1 exit if ( currentDocNum == numOfDocs ) end $replacedTag.each { |key , value| line = line.gsub(key , value) } # end - $replacedTag.each targetFile.puts line end # file.each_line end # File.open() end # begin } # end - scdFileList puts "Warning : Need more documents : " + currentDocNum.to_s if ( currentDocNum < numOfDocs )