Ruby

Returning value from a method

class CapitalCityFinder

  def get_city(country)
    data = { France: "Paris", India: "New Delhi", England: "London" }
    city = data[country.to_sym]
    return city
  end

  def info(country)
    city = get_city(country)
    puts "The capital of #{country} is #{city}"
  end
end

finder = CapitalCityFinder.new
finder.info("India")

class CapitalCityFinder

  def get_city(country)
    data = { France: "Paris", India: "New Delhi", England: "London" }
    
    return nil if country == nil
    city = data[country.to_sym]
    return city
  end

  def info(country)
    city = get_city(country)
    if city == nil
      puts "Sorry we don't have info about the country #{country}"
    else
      puts "The capital of #{country} is #{city}"
    end
  end
end

finder = CapitalCityFinder.new
finder.info("Japan")