CODEWITHSUNDEEP
pythonpython-3.xclass
MC Abstract
MC Abstract

Reputation: 37

How do I change a superClass attribute that's inside of a subClass?

When I define the __init__ of ProductionWorker, I also need to set the attributes of EmployeeClass. I entered "Bob" and "001121" as a test and it works but I need to be able to change it in my main from the input of the user.

class ProductionWorker(EmployeeClass):
    SHIFT = {1: "day shift", 2: "night shift"}

    def __init__(self, shift=None, hourly_pay=None):
        EmployeeClass.__init__(self,  "Bob", "001121")
        self.__shift = shift
        self.set_shift = shift
        self.__hourly_pay = hourly_pay
        self.set_hourly_pay = hourly_pay
    # setters

    def set_shift(self, shift):
        if shift in ProductionWorker.SHIFT:
            self.__shift = shift
        else:
            self.__shift = None

    def set_hourly_pay(self, hourly_pay):
        self.__hourly_pay = hourly_pay
    # getters

    def get_shift(self):
        return self.__shift

    def get_hourly_pay(self):
        return self.__hourly_pay

    def __str__(self):
        summary = EmployeeClass.__str__(self)
        return summary + "They work on the " + ProductionWorker.SHIFT[self.__shift] + " and make " + "$" \
               + str(format(self.__hourly_pay, "0.2f")) + " an hour."

My main:

from Employee import EmployeeClass
from Employee import ProductionWorker

e_name = input("Enter the name of the employee: ")
e_number = input("Enter the ID number of the employee: ")
e_shift = int(input("Enter 1 if they work day shift or 2 if they work night shift: "))
e_hourly_pay = float(input("Enter how much they make hourly (numerical): "))

x = EmployeeClass(e_name, e_number)
z = ProductionWorker(e_shift, e_hourly_pay)
print(z)

This is the result I get:

Enter the name of the employee: Joe
Enter the ID number of the employee: 5555
Enter 1 if they work day shift or 2 if they work night shift: 2
Enter how much they make hourly (numerical): 30

The employee's name is Bob. Bob's ID number is: 001121. They work on the night shift and make $30.00 an hour.

Upvotes: 0

Views: 107

Answers (2)

Matt Eding
Matt Eding

Reputation: 1002

I would include the parameters of the EmployeeClass in the init method parameters of the ProductionWorker to pass along to the superclass.

For python 3 you can do super().__init___() rather than EmployeeClass.__init__().

Additionally you should consider using descriptors rather than implementing getters and setters as that is the pythonic way to do that.

class ProductionWorker(EmployeeClass):
    def __init__(self, name, number, shift=None, hourly_pay=None):
        super().__init__(name, number)
        self.__shift = shift
        self.__hourly_pay = hourly_pay

Upvotes: 0

Daniel
Daniel

Reputation: 42748

You have to use arguments as any with any other parameters:

class ProductionWorker(EmployeeClass):
    SHIFT = {1: "day shift", 2: "night shift"}

    def __init__(self, name, number, shift=None, hourly_pay=None):
        EmployeeClass.__init__(self,  name, number)
        self._shift = shift
        self.hourly_pay = hourly_pay

    @property
    def shift(self):
        return self._shift

    @shift.setter
    def shift(self, shift):
        if shift in ProductionWorker.SHIFT:
            self._shift = shift
        else:
            self._shift = None

    def __str__(self):
        summary = EmployeeClass.__str__(self)
        return summary + "They work on the {} and make ${:.2f} an hour.".format(
            ProductionWorker.SHIFT[self.shift], self.hourly_pay)

name = input("Enter the name of the employee: ")
number = input("Enter the ID number of the employee: ")
shift = int(input("Enter 1 if they work day shift or 2 if they work night shift: "))
hourly_pay = float(input("Enter how much they make hourly (numerical): "))

z = ProductionWorker(name, number, shift, hourly_pay)
print(z)

Upvotes: 1

Related Questions