Skip to content

Instantly share code, notes, and snippets.

@vanmichael

vanmichael/calculate_payroll.rb

Created December 5, 2013 14:13
Show Gist options
  • Save vanmichael/7805740 to your computer and use it in GitHub Desktop.
Save vanmichael/7805740 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
calculate_payroll.rb
require 'pry'
require 'csv'
class EmployeeReader
attr_reader :employees
def initialize(file_name)
get_employees(file_name)
end
def get_employees(file_name)
@employees = []
CSV.foreach(file_name, headers: true) do |row|
case row["salary_type"]
when "commission" then @employees << CommissionSalesPerson.new(row["first_name"],row["last_name"],row["base_salary"],row["commission"])
when "quota" then @employees << QuotaSalesPerson.new(row["first_name"],row["last_name"],row["base_salary"],row["bonus"],row["sales_quota"])
when "owner" then @employees << Owner.new(row["first_name"],row["last_name"],row["base_salary"],row["bonus"],row["company_quota"])
else @employees << Employee.new(row["first_name"],row["last_name"],row["base_salary"])
end
end
@employees
binding.pry
end
end
class Employee
def initialize(first_name, last_name, base_salary)
@first_name = first_name
@last_name = last_name
@base_salary = base_salary.to_f
end
def gross_salary_per_month
(@base_salary.to_f/12).to_f
end
def net_pay_per_month
0.70 * gross_salary_per_month
end
def display_amount_owed
puts " "
puts "***#{@first_name} #{@last_name}***"
puts "Gross Salary: #{gross_salary_per_month.round(2)}"
puts "Net Pay: #{net_pay_per_month.round(2)}"
end
end
class Owner < Employee
def initialize(first_name, last_name, base_salary, bonus, company_quota)
super(first_name,last_name,base_salary)
@bonus = bonus.to_f
@company_quota = company_quota.to_f
get_sales
end
def get_sales
@total_sales = 0
employee_sales = Sale.new('sales.csv').gross_sale_values
employee_sales.each do |sale|
@total_sales += sale.values.join.to_f
end
end
def gross_salary_per_month
if @total_sales >= @company_quota
(@base_salary.to_f/12) + (@bonus)
else
(@base_salary.to_f/12)
end
end
def net_pay_per_month
super
end
def display_amount_owed
puts " "
puts "***#{@first_name} #{@last_name}***"
puts "Gross Salary: #{gross_salary_per_month.round(2)}"
puts "Net Pay: #{net_pay_per_month.round(2)}"
end
end
class CommissionSalesPerson < Employee
def initialize(first_name, last_name, base_salary, commission)
super(first_name,last_name,base_salary)
@commission = commission.to_f
get_sales(last_name)
end
def get_sales(last_name)
@total_sales = 0
employee_sales = Sale.new('sales.csv').gross_sale_values
employee_sales.each do |sale|
@total_sales += sale.values.join.to_f if sale.keys.join == last_name
end
end
def gross_salary_per_month
(@base_salary/12) + ((@commission * @total_sales))
end
def net_pay_per_month
super
end
def display_amount_owed
puts " "
puts "***#{@first_name} #{@last_name}***"
puts "Gross Salary: #{gross_salary_per_month.round(2)}"
puts "Commission: #{((@commission * @total_sales)*0.70).round(2)}"
puts "Net Pay: #{net_pay_per_month.round(2)}"
end
end
class QuotaSalesPerson < Employee
def initialize(first_name, last_name, base_salary, bonus, sales_quota)
super(first_name,last_name,base_salary)
@bonus = bonus.to_f
@sales_quota = sales_quota.to_f
get_sales(last_name)
end
def get_sales(last_name)
@total_sales = 0
employee_sales = Sale.new('sales.csv').gross_sale_values
employee_sales.each do |sale|
@total_sales += sale.values.join.to_f if sale.keys.join == last_name
end
end
def gross_salary_per_month
if @total_sales >= @sales_quota
(@base_salary/12) + (@bonus)
else
@base_salary/12
end
end
def net_pay_per_month
super
end
def display_amount_owed
puts " "
puts "***#{@first_name} #{@last_name}***"
puts "Gross Salary: #{gross_salary_per_month.round(2)}"
puts "Net Pay: #{net_pay_per_month.round(2)}"
end
end
class Sale
attr_reader :gross_sale_values
def initialize(file_name)
get_sales(file_name)
end
def get_sales(file_name)
@gross_sale_values = []
CSV.foreach(file_name, headers: true) do |row|
@gross_sale_values << Hash[row["last_name"],row["gross_sale_value"]]
end
@gross_sale_values
end
end
Raw
employees.csv
We can make this file beautiful and searchable if this error is corrected: It looks like row 2 should actually have 8 columns, instead of 4. in line 1.
salary_type,last_name,first_name,base_salary,commission,bonus,sales_quota,company_quota
base,Smith,Johnny,80000
base,McMahon,Jimmy,85000
commission,Lob,Bob,50000,0.005
commission,Bobby,Ricky,40000,0.015
base,Krabappel,Edna,87000
quota,Groundskeeper,Willie,75000,0,5000,60000
quota,Wiggum,Clancy,80000,0,10000,80000
owner,Burns,Charles,500000,0,1000,0,250000
Raw
payroll.rb
require 'pry'
require 'csv'
class EmployeeReader
attr_reader :employees
def initialize(file_name)
get_employees(file_name)
end
def get_employees(file_name)
@employees = []
CSV.foreach(file_name, headers: true) do |row|
case row["salary_type"]
when "commission" then @employees << CommissionSalesPerson.new(row["first_name"],row["last_name"],row["base_salary"],row["commission"])
when "quota" then @employees << QuotaSalesPerson.new(row["first_name"],row["last_name"],row["base_salary"],row["bonus"],row["sales_quota"])
when "owner" then @employees << Owner.new(row["first_name"],row["last_name"],row["base_salary"],row["bonus"],row["company_quota"])
else @employees << Employee.new(row["first_name"],row["last_name"],row["base_salary"])
end
end
@employees
binding.pry
end
end
class Employee
def initialize(first_name, last_name, base_salary)
@first_name = first_name
@last_name = last_name
@base_salary = base_salary.to_f
end
def gross_salary_per_month
(@base_salary.to_f/12).to_f
end
def net_pay_per_month
0.70 * gross_salary_per_month
end
def display_amount_owed
puts " "
puts "***#{@first_name} #{@last_name}***"
puts "Gross Salary: #{gross_salary_per_month.round(2)}"
puts "Net Pay: #{net_pay_per_month.round(2)}"
end
end
class Owner < Employee
def initialize(first_name, last_name, base_salary, bonus, company_quota)
super(first_name,last_name,base_salary)
@bonus = bonus.to_f
@company_quota = company_quota.to_f
get_sales
end
def get_sales
@total_sales = 0
employee_sales = Sale.new('sales.csv').gross_sale_values
employee_sales.each do |sale|
@total_sales += sale.values.join.to_f
end
end
def gross_salary_per_month
if @total_sales >= @company_quota
(@base_salary.to_f/12) + (@bonus)
else
(@base_salary.to_f/12)
end
end
def net_pay_per_month
super
end
def display_amount_owed
puts " "
puts "***#{@first_name} #{@last_name}***"
puts "Gross Salary: #{gross_salary_per_month.round(2)}"
puts "Net Pay: #{net_pay_per_month.round(2)}"
end
end
class CommissionSalesPerson < Employee
def initialize(first_name, last_name, base_salary, commission)
super(first_name,last_name,base_salary)
@commission = commission.to_f
get_sales(last_name)
end
def get_sales(last_name)
@total_sales = 0
employee_sales = Sale.new('sales.csv').gross_sale_values
employee_sales.each do |sale|
@total_sales += sale.values.join.to_f if sale.keys.join == last_name
end
end
def gross_salary_per_month
(@base_salary/12) + ((@commission * @total_sales))
end
def net_pay_per_month
super
end
def display_amount_owed
puts " "
puts "***#{@first_name} #{@last_name}***"
puts "Gross Salary: #{gross_salary_per_month.round(2)}"
puts "Commission: #{((@commission * @total_sales)*0.70).round(2)}"
puts "Net Pay: #{net_pay_per_month.round(2)}"
end
end
class QuotaSalesPerson < Employee
def initialize(first_name, last_name, base_salary, bonus, sales_quota)
super(first_name,last_name,base_salary)
@bonus = bonus.to_f
@sales_quota = sales_quota.to_f
get_sales(last_name)
end
def get_sales(last_name)
@total_sales = 0
employee_sales = Sale.new('sales.csv').gross_sale_values
employee_sales.each do |sale|
@total_sales += sale.values.join.to_f if sale.keys.join == last_name
end
end
def gross_salary_per_month
if @total_sales >= @sales_quota
(@base_salary/12) + (@bonus)
else
@base_salary/12
end
end
def net_pay_per_month
super
end
def display_amount_owed
puts " "
puts "***#{@first_name} #{@last_name}***"
puts "Gross Salary: #{gross_salary_per_month.round(2)}"
puts "Net Pay: #{net_pay_per_month.round(2)}"
end
end
class Sale
attr_reader :gross_sale_values
def initialize(file_name)
get_sales(file_name)
end
def get_sales(file_name)
@gross_sale_values = []
CSV.foreach(file_name, headers: true) do |row|
@gross_sale_values << Hash[row["last_name"],row["gross_sale_value"]]
end
@gross_sale_values
end
end
Raw
sales.csv
last_name gross_sale_value
Bobby 25000
Groundskeeper 20000
Bobby 20000
Bobby 40000
Wiggum 25000
Wiggum 10000
Lob 10000
Lob 10000
Groundskeeper 15000
Bobby 80000
Lob 5000
Groundskeeper 30000
Raw
user_story.txt
User Stories
As human resources
I want to calculate the net pay
So that I can pay the employees
As human resources
I want to calculate the gross salary
So that I can track the companies expenses
As human resources
I want to load a CSV of sales
So that I can calculate commisions
As human resources
I want to load a CSV of the employees
So that I know who gets paid
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment