root/lang/ruby/bakagaikuonrails/cron.rb @ 38969

module BakaReception
  def wget_entry(bakaid)
    require 'net/http'
    Net::HTTP.version_1_2
    Net::HTTP.start(BakaAddress, BakaPort) do |http|
      path = BakaPath + bakaid
      http.request_get( path )
    end
  end

  def create_or_update_entry(bakaid)
    print bakaid
    response = wget_entry(bakaid)
    print " => #{response.code}"
    unless response.code == "200"
      puts
      return nil
    end
    entry = Entry.find(:first, :conditions => ['bakaid = ?', bakaid])
    if entry
      entry.body = response.body
      entry.save
      puts " => update"
    else
      entry = entries.create(:bakaid => bakaid, :body => response.body)
      puts " => create"
    end
    entry
  end

  def wget_entries(max = 30)
    if last_entry = Entry.find(:first, :order => "bakaid DESC")
      last_entry.bakaid =~ /^(\d\d\d\d)(\d\d)(\d\d)/
      first_year, first_month, first_day = $1.to_i, $2.to_i, $3.to_i
    else
      first_year, first_month, first_day = BeginningDate
    end
    t = Time.now
    last_year, last_month, last_day = t.year, t.mon, t.day

    counter = 0
    (first_year..last_year).each do |y|
      (1..12).each do |m|
        next if y == first_year && m < first_month
        break if y == last_year && m > last_month
        (1..31).each do |d|
          next if y == first_year && m == first_month && d < first_day
          break if y == last_year && m == last_month && d > last_day
          (0..100).each do |n|
            return 0 if counter >= max
            bakaid = sprintf "%04d%02d%02d", y, m, d
            bakaid += n.to_s if n > 0
            #next if  Entry.find(:first, :conditions => ['bakaid = ?', bakaid])
            counter += 1
            break unless create_or_update_entry(bakaid)
          end
        end
      end
    end
  end
end

def baka_reader(n = nil)
  @reception = Reception.create
  @reception.extend BakaReception
  @reception.wget_entries(n)
  @reception.destroy if 0 == @reception.entries.count
end

def srcdir
  File.dirname(File.expand_path(__FILE__)).untaint
end

$LOAD_PATH.unshift srcdir
require 'myutil'
setup_environment

n = ARGV[0].to_i
n = 20 if n <= 0

ActiveRecord::Base.transaction do
  baka_reader n
end