Importing A Political Database

importing_political_db.rb

require 'csv'
require 'sqlite3'

class PoliticianDB
  
  @@politicians_stats =[]

  def self.setup
    system('rm politicians.db')
    $db = SQLite3::Database.new "politicians.db"
    $db.execute(
      <<-SQL
        CREATE TABLE politicians (
          id INTEGER PRIMARY KEY AUTOINCREMENT,
          name VARCHAR(64) NOT NULL,
          party VARCHAR(64) NOT NULL,
          location_1 VARCHAR(2) NOT NULL,
          grade_level_since_1996 INTEGER (64) NOT NULL,
          grade_level_112th_congress  INTEGER (64) NOT NUll,
          years_in_congress INTEGER (2) NOT NULL,
          dw1_score INTEGER (6) NOT NUll

        );
      SQL
    )
  end

  def self.seed
    CSV.foreach('politician_data.csv', :headers => true, :header_converters => :symbol ) do |row|
      @@politicians_stats << row
    end

    @@politicians_stats.each do |row|
    $db.execute(
      <<-SQL
        INSERT INTO politicians 
          (name, party, location_1, grade_level_since_1996, grade_level_112th_congress, years_in_congress, dw1_score)
        VALUES
          ("#{row[:name]}", "#{row[:party]}", "#{row[:location_1]}", "#{row[:grade_level_since_1996]}", "#{row[:grade_level_112th_congress]}", "#{row[:years_in_congress]}", "#{row[:dw1_score]}");
      SQL
    )
    end 
  end
  

  def self.where(col_name,param)
    $db.execute("SELECT * FROM politicians WHERE #{col_name} like '#{param}'")    
  end  

  # delete all based on column name and parameter
  def self.delete(col_name,param)
    $db.execute("DELETE FROM politicians WHERE #{col_name} like '#{param}'")    
  end  

  # delete primary keys / single row
  def self.delete_at_id(id)
    $db.execute("DELETE FROM politicians WHERE id = #{id}")
  end

  def self.insert(row={})
    $db.execute(
      <<-SQL
        INSERT INTO politicians 
          (name, party, location_1, grade_level_since_1996, grade_level_112th_congress, years_in_congress, dw1_score)
        VALUES
          ("#{row[:name]}", "#{row[:party]}", "#{row[:location_1]}", "#{row[:grade_level_since_1996]}", "#{row[:grade_level_112th_congress]}", "#{row[:years_in_congress]}", "#{row[:dw1_score]}");
      SQL
    )
  end

  def self.update(id,num,row={})
    persons_data_pre_update = self.where(id,num)
    $db.execute(
      <<-SQL
        UPDATE politicians 
        SET
          name ="#{row[:name] || persons_data_pre_update[0][1]}", 
          party="#{row[:party] || persons_data_pre_update[0][2]}", 
          location_1="#{row[:location_1] || persons_data_pre_update[0][3]}", 
          grade_level_since_1996="#{row[:grade_level_since_1996] || persons_data_pre_update[0][4]}",
          grade_level_112th_congress="#{row[:grade_level_112th_congress] || persons_data_pre_update[0][5]}",
          years_in_congress="#{row[:years_in_congress] || persons_data_pre_update[0][6]}", 
          dw1_score="#{row[:dw1_score] || persons_data_pre_update[0][7]}"
        WHERE "#{id}" = "#{num}"
      SQL
    )
  end

  def self.count(col_name,param)
    $db.execute(
      <<-SQL
        select "#{col_name}",count(*) from politicians where "#{col_name}"= "#{param}";
      SQL
    )
  end


end

#PoliticianDB.new
PoliticianDB.setup
PoliticianDB.seed
# puts PoliticianDB.where('party','R')
# puts PoliticianDB.where('name','Rep.%')
# puts PoliticianDB.where('id','10')
# puts PoliticianDB.delete(10)
# puts PoliticianDB.where('id','10')

puts PoliticianDB.insert(name:"Dave Hoover", party:"DBC", years_in_congress:5)
p PoliticianDB.where('id','532')
# 532 is Dave Hoover
puts "** UPDATE **"
PoliticianDB.update('id','532',name:"Super Dave",party:"RastaFarians")
puts "** UPDATE **"
puts PoliticianDB.where('id','532')

# puts PoliticianDB.count('party','R') # 287
# PoliticianDB.delete('party','R')
# puts PoliticianDB.count('party','R') # 0



# parser = PoliticianDB.new

politician_data.csv

queries_and_statistics.txt

-- Enter your query results and statistical insights here!  --