Sha256: c14ad36f4d900b913dbddb6e0da47cd959bc7621d341778d2e878691abaaff63
Contents?: true
Size: 1.45 KB
Versions: 7
Compression:
Stored size: 1.45 KB
Contents
module Sites class Base attr_reader :job_type, :doc, :url def initialize(job_type: :programming) @job_type = job_type @url = build_url @doc = Nokogiri::HTML(open_page(@url)) @current_time = Time.new @timestamp = @current_time.strftime("%Y%m%d%H%M%S") @count = get_count end def open_page(url) sleep(rand(0..2.0)) unless ENV['RAILS_ENV'] == 'test' # less mechanical behaviour if ENV['RAILS_ENV'] == 'test' open(url) else open(url, 'User-Agent' => user_agent) end end private def user_agent Support::UserAgent::LIST.sample end def build_url case job_type when :programming then "#{self.class::HOST}#{self.class::PROGRAMMING}" when :devops then "#{self.class::HOST}#{self.class::DEVOPS}" else raise "Error" end end def filepath return test_filepath if ENV["RAILS_ENV"] == 'test' "#{self.class::STORE_DIR}/#{@timestamp}.csv" end def test_filepath "spec/fixtures/data/#{underscore(self.class.name.split('::').last)}/#{@timestamp}.csv" end # https://stackoverflow.com/a/5622585 def underscore(camel_cased_word) word = camel_cased_word.dup word.gsub!(/::/, '/') word.gsub!(/([A-Z]+)([A-Z][a-z])/,'\1_\2') word.gsub!(/([a-z\d])([A-Z])/,'\1_\2') word.tr!("-", "_") word.downcase! word end end end
Version data entries
7 entries across 7 versions & 1 rubygems