Skip to content

Commit a098587

Browse files
committed
Removed assumption that company data files exist. Fixes ericduran#27 Fixes ericduran#41
1 parent 0aadd1d commit a098587

File tree

1 file changed

+10
-2
lines changed

1 file changed

+10
-2
lines changed

app/bin/companies.rb

Lines changed: 10 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -15,8 +15,16 @@
1515
UPDATE_ALL=2
1616

1717
name_variants = Hash.new(0)
18-
$companies_info = YAML::load_file('../data/company_infos.yml') || Hash.new(0)
19-
company_mapping = YAML::load_file('../data/company_mapping.yml') || Hash.new(0)
18+
if File.exists? ('../data/company_infos.yml')
19+
$companies_info = YAML::load_file('../data/company_infos.yml')
20+
else
21+
$companies_info = Hash.new(0)
22+
end
23+
if File.exists? ('../data/company_mapping.yml')
24+
company_mapping = YAML::load_file('../data/company_mapping.yml') || Hash.new(0)
25+
else
26+
company_mapping = Hash.new(0)
27+
end
2028
update=UPDATE_NONE
2129
if ARGV.length == 1
2230
if ARGV[0] == '--update-all'

0 commit comments

Comments
 (0)