16
votes

I've just learned the basics of Ruby after being very happy with Python for several years (I'm still using Python for some things), but I'd like to know if there's an idiom or hack to solve this particular problem.

I have a Ruby script which I'd like to be able to do require script_name with, but I'd also like to be able to run ruby script_name.rb from the terminal and have it run as a command line script. In Python this would be done by having the following structure at the bottom of the script:

if __name__ == '__main__':
    # do something here

However, I can't seem to find an equivalent in Ruby. Is there a way of detecting whether or not the current script is being run from the command-line? Maybe some Kernel:: method or something? Ideally what I'd like is something like this at the bottom of the script:

if from_command_line?
  # do something here
end
3
You should review later answers to your question as they also include code that replicate the python class testing idiom.QueueHammer

3 Answers

24
votes

You want to use:

if __FILE__ == $0
  # do stuff
end

__FILE__ is the source file name and $0 is the name of the script currently being executed.

7
votes

You can find a similar functionality in ruby.

__FILE__ the current source file name.

$0 Contains the name of the script being executed. May be assignable.

source: Ruby Quick Ref

5
votes

While

if __FILE__ == $0
  Foo.run
end

is the common approach, I'm currently using

if File.identical?(__FILE__, $0)
  Foo.run
end

because programs like ruby-prof can make $0 not equal __FILE__ even when you use --replace-progname.