SteveL_votersim.rb

##votersim.rb

class Person
  attr_accessor :voters, :politicians

  def initialize
    @@voters = []
    @@politicians = []
  end

  def select_option
    option = ""
    while option == ""
      print "Please select (c)reate, (l)ist, (u)pdate, (v)ote, or (q)uit: "
      option = gets.chomp.downcase
      if option == "q"
        puts "Thank you.  Exiting program"
        exit
      else
        case option
        when "c"
          create_option
        when "l"
          list_option
        when "u"
          update_option
        when "v"
          vote_option
        else
          option = ""
          puts "Invalid option selected."
        end
      end
    end
  end

  def create_option
    puts "CREATE voter or politician:"
    create_option = ""
    until create_option == "quit"
      print "Please select (v)oter or (p)olitician, or (q)uit for main menu: "
      create_option = gets.chomp
      case create_option
      when "v"
        create_voter
      when "p"
        create_politician
      when "q"
        select_option
      else
        puts "Invalid option selected."
      end
    end
  end

  def list_option
    puts "LIST voters and politicians:"
    if @@voters.count >= 1
      @@voters.each do |voter|
        puts "* Voter, #{voter.voter_name }, #{voter.politics}"
      end
    else
      puts "There are no voters to list"
    end
    if @@politicians.count >= 1
      @@politicians.each do |politician|
        puts "* Politician, #{politician.politician_name}, #{politician.party}"
      end
    else
      puts "There are no politicians to list"
    end
    select_option
  end

  def update_option
    puts "UPDATE voter or politician: "
    update_option = ""
    until update_option == "quit"
      print "Please select (v)oter or (p)olitician, or (q)uit for main menu: "
      update_option = gets.chomp.downcase
      case update_option
      when "v"
        update_voter
      when "p"
        update_politician
      when "q"
        select_option
      else
        puts "Invalid option selected."
      end
    end
  end

  def vote_option
    puts "VOTING:"
    @dem_count = 0
    @rep_count = 0
    @@voters.each do | voter |
      @@politicians.each do | pol |
        case pol.party
        when "democrat"
          case voter.politics
          when "socialist", "liberal", "neutral"
            @dem_count += 1
          end
        when "republican"
          case voter.politics
          when "tea party", "conservative"
            @rep_count += 1
          end
        end
      end
    end

    politician_republican_vote = 1
    @rep_count += politician_republican_vote
    politician_democratic_vote = 1
    @dem_count += politician_democratic_vote

    puts "Voting Results:"
    puts "#{@dem_count} votes for the Democrat..."
    puts "#{@rep_count} votes for the Republican..."
    if @dem_count > @rep_count
      puts "The democrat wins!!!!"
    else
      puts "The republican wins!!!!"
    end
    select_option
  end

  def create_voter
    puts "CREATE new_voter:"
    name = ""
    until name != ""
      print "Voter name? "
      name = gets.chomp
    end
    politics = ""
    until politics != ""
      print "Politics?  Liberal, Conservative, Tea party, Socialist, or Neutral: "
      politics_entry = gets.chomp.downcase
      case politics_entry
      when "liberal", "conservative", "tea party", "socialist", "neutral"
        politics = politics_entry
      else
        puts "Invalid selection:"
      end
    end
    new_voter = Voter.new(name, politics)
    @@voters << new_voter
    puts "Voter #{new_voter.voter_name}, #{new_voter.politics}, has been created"
  end

  def update_voter
    puts "UPDATE voter:"
    name = ""
    until name != ""
      print "Voter name? "
      name = gets.chomp
    end
    politics = ""
    until politics != ""
      print "Politics?  Liberal, Conservative, Tea party, Socialist, or Neutral: "
      politics_entry = gets.chomp.downcase
      case politics_entry
      when "liberal", "conservative", "tea party", "socialist", "neutral"
        politics = politics_entry
      else
        puts "Invalid selection:"
      end
    end
    @found = false
    @@voters.each do |voter|
      if voter.voter_name == name
        @found = true
        update_voter = voter
        update_voter.politics = politics
      end
    end
    if !@found then
      puts "no match found"
    end
  end

  def create_politician
    puts "in create_politician"
    puts "Create new_politician:"
    name = ""
    until name != ""
      print "Politician name? "
      name = gets.chomp
    end
    party = ""
    until party != ""
      print "Party?  Democrat or Republican: "
      party_entry = gets.chomp.downcase
      case party_entry
      when "democrat", "republican"
        party = party_entry
      else
        puts "Invalid selection:"
      end
    end
    new_politician = Politician.new(name, party)
    @@politicians << new_politician
    puts "Politician #{new_politician.politician_name}, #{new_politician.party}, has been created"
  end

  def update_politician
    puts "in update_politician"
    puts "Update politician:"
    print "Politician name? "
    name = gets.chomp
    party = ""
    until party != ""
      print "Party?  Democrat or Republican: "
      party_entry = gets.chomp.downcase
      case party_entry
      when "democrat", "republican"
        party = party_entry
      else
        puts "Invalid selection:"
      end
    end
    @found = false
    @@politicians.each do |politician|
      if politician.politician_name == name
        @found = true
        update_politician = politician
        update_politician.party = party
      end
    end
    if !@found
      puts "Politician #{name} not found"
    end
  end
end

class Voter < Person
  attr_accessor :voter_name, :politics

  def initialize(name, politics)
    @voter_name = name
    @politics = politics
  end
end

class Politician < Person
  attr_accessor :politician_name, :party

  def initialize(name, party)
    @politician_name = name
    @party = party
  end
end

def test_voter_sim
  person = Person.new
  person.select_option
  puts "returned to test_voter_sim"
end

test_voter_sim