recruitment/hr_pro.py at master · nancyelsh/recruitment · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
from datetime import date
today = date.today()

class Employee:
  def __init__(self, name, age, salary, employment_date):
    self.name = name
    self.age = age
    self.salary = salary
    # Enter date in (year,month,date) format
    self.employment_date = employment_date
    #year,month,day = employment_date(',')
  def get_working_years(self):
    return today.year - employment_date
    #return today.year - year - ((today.month, today.day) < (month, day))
  def __str__(self):
    return "Name: {self.name}, Age: {self.age}, Salary: {self.salary}, Working years: {get_working_years(self)}".format(self = self)

class Manager:
  def __init__(self, name, age, salary, employment_date, bonus_percentage):
    self.name = name
    self.age = age
    self.salary = salary
    self.bonus_percentage = bonus_percentage
    # Enter date in (year,month,date) format
    self.employment_date = employment_date
    #year,month,day = employment_date(',')
  def get_working_years(self):
    return today.year - employment_date
    #return today.year - year - ((today.month, today.day) < (month, day))
  def get_bonus(self):
    return bonus_percentage*salary
  def __str__(self):
    return "Name: {self.name}, Age: {self.age}, Salary: {self.salary}, Working years: {get_working_years(self), Bonus: {get_bonus(self)}".format(self = self)

employees = []
managers = []

print("""Welcome to HR Pro 2019
  Options:
    1. Show Employees
    2. Show Managers
    3. Add an Employee
    4. Add a Manager
    5. Exit\n""")
option = input("What would you like to do? ")
while option != '5':
  if option == '1':
    print("-----------------")
    print("Employees\n")
    for employee in employees:
      print(employee)
  elif option == '2':
    print("-----------------")
    print("Managers\n")
    for manager in managers:
      print(manageryy)
  elif option == '3':
    print("-----------------")
    name = input("Name: ")
    age = input("Age: ")
    salary = input("Salary: ")
    year = input("Employment year: ")
    employees.append(Employee(name, int(age), int(salary), int(year)))
#    employees.append(new_employee)
    print("Employee added successfully")
    print("""Welcome to HR Pro 2019
      Options:
        1. Show Employees
        2. Show Managers
        3. Add an Employee
        4. Add a Manager
        5. Exit\n""")
    option = input("What would you like to do? ")
  elif option == '4':
    print("-----------------")
    name = input("Name: ")
    age = input("Age: ")
    salary = input("Salary: ")
    bonus = input("Bonus percentage: ")
    year = input("Employment year: ")
    managers.append(Manager(name, int(age), int(salary), float(bonus), int(year)))
#    managers.append(new_manager)
    print("Manager added successfully")
    print("""Welcome to HR Pro 2019
      Options:
        1. Show Employees
        2. Show Managers
        3. Add an Employee
        4. Add a Manager
        5. Exit\n""")
    option = input("What would you like to do? ")

#    new_employee = Employee(input("Add an employee in the format (name, age, salary, employment date),\nex: ('Nancy', 25, 600, (2019,2,5))"))
#      employees.append(new_employee)
#  elif option == '4':
#    new_manager = Manager(input("Add a manager in the format (name, age, salary, bonus percentage, employment date), \nex: ('Nancy', 25, 600, 25, (2019,2,5))"))
#      managers.append(new_manager)