Check for user-agents.txt before using it

2018-05-09 13:15:12 +01:00
parent 2c40913a64
commit 435fb34233
1 changed files with 5 additions and 0 deletions
--- a/lib/common/common_helper.rb
+++ b/lib/common/common_helper.rb
@@ -255,6 +255,11 @@ end
 # @return [ String ] A random user-agent from data/user-agents.txt
 def get_random_user_agent
  user_agents = []
+
+  unless File.exist?(USER_AGENTS_FILE)
+    raise('[ERROR] Missing user-agent data. Please re-run with --update.')
+  end
+
  f = File.open(USER_AGENTS_FILE, 'r')
  f.each_line do |line|
    # ignore comments