Skip to content

Instantly share code, notes, and snippets.

@schacon

schacon/dem.rb

Created February 3, 2009 23:02
Show Gist options

  • Select an option

    Learn more about clone URLs
  • Save schacon/57815 to your computer and use it in GitHub Desktop.

Select an option

Learn more about clone URLs
Save schacon/57815 to your computer and use it in GitHub Desktop.
Download ZIP

Revisions

  1. schacon revised this gist Feb 4, 2009. 1 changed file with 110 additions and 97 deletions.
    207 changes: 110 additions & 97 deletions dem.rb
    View file
    Original file line number Diff line number Diff line change
    @@ -1,97 +1,110 @@
    require "rubygems"
    require "activerecord"
    require '../../munger/lib/munger'

    ActiveRecord::Base.establish_connection(
    :adapter => "postgresql",
    :host => "localhost",
    :database => "testing",
    :username => "postgres",
    :password => "postgres"
    )

    class Participant < ActiveRecord::Base
    end

    class Demographic < ActiveRecord::Base
    end

    # original code

    class DemographicReport
    include Munger

    def initialize(program_ids, start_date, end_date)
    date_range = "assessments.created_at #{(start_date..end_date).to_s(:db)}"
    participant_result = Participant.find(:all,
    :conditions => "program_id in (#{program_ids}) AND #{date_range}",
    :joins => :assessments )
    @participant_data = Munger::Data.load_data(participant_result)
    demographic_result = Demographic.find(:all,
    :conditions => "program_id in (#{program_ids}) AND #{date_range}",
    :joins => :assessment )
    @demographic_data = Munger::Data.load_data(demographic_result)
    @age_data = build_age_data(program_ids, date_range)
    end

    def build_age_data(program_ids, date_range)
    age_data = Munger::Data.new
    AgeRange::AGE_RANGES.each do |name, range|
    result = Demographic.find(:all,
    :select => "count(age_when_taken) as count",
    :conditions => "age_when_taken between #{range.first} and
    #{range.last} and program_id in (#{program_ids}) AND #{date_range}",
    :joins => :assessment )
    data = Munger::Data.load_data(result)
    data.add_column(:age, :default => name)
    age_data << data.data
    end
    age_data
    end

    def gender_report
    data = munge_data(:gender, @participant_data)
    data = translate_column(data, :gender, Gender::GENDER['en-US'])
    report = munge_report(data, :gender)
    puts Munger::Render.to_text(report)
    end

    def race_report
    data = munge_data(:race, @participant_data)
    data = translate_column(data, :race, Race::RACES['en-US'])
    report = munge_report(data, :race)
    puts Munger::Render.to_text(report)
    end

    def age_report
    report = Munger::Report.from_data(@age_data)
    report.process
    puts Munger::Render.to_text(report)
    end

    private

    def munge_data(field, data)
    rows = data.size
    data.group(field, :count => :id)
    data.add_column('percent') do |r|
    (r.count_id.to_f / rows.to_f) * 100
    end
    data
    end

    def translate_column(data, column_name, translation_hash)
    data.transform_column(column_name) do |r|
    eval("translation_hash[r.#{column_name}]")
    end
    end

    def munge_report(data, column_name)
    report = Munger::Report.from_data(data)
    report.columns(:count_id => 'count')
    report.columns([column_name, :count_id, :percent])
    report.aggregate(:sum => [:count_id, :percent])
    report.process
    end

    end
    require "rubygems"
    require "activerecord"
    require '../../munger/lib/munger'

    ActiveRecord::Base.establish_connection(
    :adapter => "postgresql",
    :host => "localhost",
    :database => "testing",
    :username => "postgres",
    :password => "postgres"
    )

    class Participant < ActiveRecord::Base
    has_many :demographics
    has_many :assessments
    end

    class Demographic < ActiveRecord::Base
    belongs_to :participant
    has_one :assessment
    end

    class Assessment < ActiveRecord::Base
    belongs_to :participant
    belongs_to :demographic
    end

    # original code

    class DemographicReport
    include Munger

    def initialize(program_ids, start_date, end_date)
    date_range = "assessments.created_at #{(start_date..end_date).to_s(:db)}"
    participant_result = Participant.find(:all,
    :conditions => "program_id in (#{program_ids}) AND #{date_range}",
    :joins => :assessments )
    @participant_data = Munger::Data.load_data(participant_result)
    demographic_result = Demographic.find(:all,
    :conditions => "program_id in (#{program_ids}) AND #{date_range}",
    :joins => :assessment )
    @demographic_data = Munger::Data.load_data(demographic_result)
    end

    def gender_report
    data = munge_data(:gender, @participant_data)
    #data = translate_column(data, :gender, Gender::GENDER['en-US'])
    report = munge_report(data, :gender)
    puts Munger::Render.to_text(report)
    end

    def race_report
    data = munge_data(:race, @participant_data)
    #data = translate_column(data, :race, Race::RACES['en-US'])
    report = munge_report(data, :race)
    puts Munger::Render.to_text(report)
    end

    AGE_RANGES = [
    ['<18', [0, 18]],
    ['>18', [19, 100]],
    ]

    def age_report
    demographic_result = Demographic.find(:all, :joins => :assessment)
    data = Munger::Data.load_data(demographic_result)
    data.add_column('age') do |r|
    AGE_RANGES.select { |a| (a[1][0] <= r.age_when_taken.to_i) && (a[1][1] >= r.age_when_taken.to_i) }.first[0]
    end
    munge_data(:age, data)
    report = munge_report(data, :age)
    puts Munger::Render.to_text(report)
    end

    private

    def munge_data(field, data)
    rows = data.size
    data.group(field, :count => :id)
    data.add_column('percent') do |r|
    (r.count_id.to_f / rows.to_f) * 100
    end
    data
    end

    def translate_column(data, column_name, translation_hash)
    data.transform_column(column_name) do |r|
    eval("translation_hash[r.#{column_name}]")
    end
    end

    def munge_report(data, column_name)
    report = Munger::Report.from_data(data)
    report.columns(:count_id => 'count')
    report.columns([column_name, :count_id, :percent])
    report.aggregate(:sum => [:count_id, :percent])
    report.process
    end

    end

    dr = DemographicReport.new('1,15,13,27', 1.year.ago, 1.day.ago)

    dr.gender_report
    puts

    dr.race_report
    puts

    dr.age_report
  2. schacon renamed this gist Feb 4, 2009. 1 changed file with 20 additions and 1 deletion.
    21 changes: 20 additions & 1 deletion gistfile1.txt → dem.rb
    View file
    Original file line number Diff line number Diff line change
    @@ -1,4 +1,23 @@
    require 'munger'
    require "rubygems"
    require "activerecord"
    require '../../munger/lib/munger'

    ActiveRecord::Base.establish_connection(
    :adapter => "postgresql",
    :host => "localhost",
    :database => "testing",
    :username => "postgres",
    :password => "postgres"
    )

    class Participant < ActiveRecord::Base
    end

    class Demographic < ActiveRecord::Base
    end

    # original code

    class DemographicReport
    include Munger

  3. @invalid-email-address Anonymous created this gist Feb 3, 2009.
    78 changes: 78 additions & 0 deletions gistfile1.txt
    View file
    Original file line number Diff line number Diff line change
    @@ -0,0 +1,78 @@
    require 'munger'
    class DemographicReport
    include Munger

    def initialize(program_ids, start_date, end_date)
    date_range = "assessments.created_at #{(start_date..end_date).to_s(:db)}"
    participant_result = Participant.find(:all,
    :conditions => "program_id in (#{program_ids}) AND #{date_range}",
    :joins => :assessments )
    @participant_data = Munger::Data.load_data(participant_result)
    demographic_result = Demographic.find(:all,
    :conditions => "program_id in (#{program_ids}) AND #{date_range}",
    :joins => :assessment )
    @demographic_data = Munger::Data.load_data(demographic_result)
    @age_data = build_age_data(program_ids, date_range)
    end

    def build_age_data(program_ids, date_range)
    age_data = Munger::Data.new
    AgeRange::AGE_RANGES.each do |name, range|
    result = Demographic.find(:all,
    :select => "count(age_when_taken) as count",
    :conditions => "age_when_taken between #{range.first} and
    #{range.last} and program_id in (#{program_ids}) AND #{date_range}",
    :joins => :assessment )
    data = Munger::Data.load_data(result)
    data.add_column(:age, :default => name)
    age_data << data.data
    end
    age_data
    end

    def gender_report
    data = munge_data(:gender, @participant_data)
    data = translate_column(data, :gender, Gender::GENDER['en-US'])
    report = munge_report(data, :gender)
    puts Munger::Render.to_text(report)
    end

    def race_report
    data = munge_data(:race, @participant_data)
    data = translate_column(data, :race, Race::RACES['en-US'])
    report = munge_report(data, :race)
    puts Munger::Render.to_text(report)
    end

    def age_report
    report = Munger::Report.from_data(@age_data)
    report.process
    puts Munger::Render.to_text(report)
    end

    private

    def munge_data(field, data)
    rows = data.size
    data.group(field, :count => :id)
    data.add_column('percent') do |r|
    (r.count_id.to_f / rows.to_f) * 100
    end
    data
    end

    def translate_column(data, column_name, translation_hash)
    data.transform_column(column_name) do |r|
    eval("translation_hash[r.#{column_name}]")
    end
    end

    def munge_report(data, column_name)
    report = Munger::Report.from_data(data)
    report.columns(:count_id => 'count')
    report.columns([column_name, :count_id, :percent])
    report.aggregate(:sum => [:count_id, :percent])
    report.process
    end

    end