Posso accedere ai DATI da uno script richiesto in Ruby?

Question

È possibile accedere al testo dopo __END__ in un file rubino diverso dallo script "principale"?

Ad esempio:

# b.rb 
B_DATA = DATA.read 
__END__ 
bbb 

.

# a.rb 
require 'b' 
A_DATA = DATA.read 
puts 'A_DATA: ' + A_DATA 
puts 'B_DATA: ' + B_DATA 
__END__ 
aaa 

.

C:\Temp>ruby a.rb 
A_DATA: 
B_DATA: aaa 

Esiste un modo per arrivare al "BBB" da b.rb?

Answer 1

Sfortunatamente, la costante globale DATA viene impostata quando viene caricato lo script "principale". Alcune cose che potrebbero essere d'aiuto:

È possibile almeno A_DATA essere corretto. Basta invertire l'ordine delle prime due operazioni in a.rb:

# a.rb 
A_DATA = DATA.read 
require 'b' 
... 

È possibile ottenere il B_DATA essere corretto se si passa attraverso un po 'di manfrina:

# load_data_regardless_of_main_script.rb 
module LoadDataRegardlessOfMainScript 
    def self.from(file) 
    # the performance of this function could be 
    # greatly improved by using a StringIO buffer 
    # and only appending to it after seeing __END__. 
    File.read(file).sub(/\A.*\n__END__\n/m, '') 
    end 
end 

# b.rb: 
require 'load_data_regardless_of_main_script' 
B_DATA = LoadDataRegardlessOfMainScript.from(__FILE__) 

Answer 2

Implementare @ suggerimento di James da usare StringIO :

require 'stringio' 
module LoadDataRegardlessOfMainScript 
    def self.from(filename) 
    data = StringIO.new 
    File.open(filename) do |f| 
     begin 
     line = f.gets 
     end until line.match(/^__END__$/) 
     while line = f.gets 
     data << line 
     end 
    end 
    data.rewind 
    data 
    end 
end 

Poi b.rb diventa

require 'load_data_regardless_of_main_script' 
B_DATA = LoadDataRegardlessOfMainScript.from(__FILE__).read