Try this:
class Employee:
def __init__(self,emp_name,emp_grade):
self.emp_name = emp_name
self.emp_grade = emp_grade
def emp_info(self):
return self.emp_name + "(" + str(self.emp_grade) + ")"
class Department:
def __init__(self,dept):
self.department = dept
self.__budget = 1000
self.employees = []
def addEmployee(self, employee):
if employee.emp_grade >= 5:
self.__budget += 1000
else:
self.__budget += 500
self.employees.append(employee)
def describe(self):
print("Dept Name :", self.department)
print("Budget :", self.__budget)
print("\nEmployees :")
for i in self.employees:
print(" ", i.emp_info())
it_dep = Department("IT")
it_dep.addEmployee(Employee("john",3))
it_dep.addEmployee(Employee("Krishna",4))
it_dep.addEmployee(Employee("Pawan",5))
sales_dep = Department("Sales")
sales_dep.addEmployee(Employee("Rahim",6))
sales_dep.addEmployee(Employee("Jessy",7))
sales_dep.addEmployee(Employee("Charan",2))
it_dep.describe()
print("\n\n")
sales_dep.describe()