我该如何接受用户的输入，如果它等于1，那就是白天，2就是晚上？

Question

我试图得到用户的输入，如果输入1的班次，它将打印出白班和2将等于夜班。我是用面向对象来做这个的，所以我不知道怎么做，因为我对它很陌生。另外，我在这里读到，教授教我设置类的方法并不是python的最佳方法，但教授就是这么想的，我对此无能为力。你知道吗

班级员工

class Employee:
    def __init__(self, name, number):
        self.__name = name
        self.__number = number

    def set_name(self, name):
        self.__name = name

    def set_number(self, number):
        self.__number = number

    def get_name(self):
        return self.__name

    def get_number(self):
        return self.__number

班级生产工人

import employee

class ProductionWorker(employee.Employee):
    def __init__(self, name, number, shift, pay):
        employee.Employee.__init__(self, name, number)

        self.__shift = shift

        self.__pay = pay

    def set_shift(self, shift):
        self.__shift = shift


    def set_pay(self, pay):
        self.__pay = pay

    def get_shift(self):
        return self.__shift

    def get_pay(self):
        return self.__pay

    def dayNight(self, shift):
        if shift == 1:
            self.__shift = "Day Shift"
        else:
            self.__shift = "Night Shift"

    def __str__(self):
        return "\nName: " + self.get_name() + "\nNumber: " + self.get_number() + \
               "\nShift: " + self.get_shift() + "\nPay: " + self.__pay

员工测试

import ProductionWorker


def main():
    name = input("Enter the users name: ")
    number = input("Enter the users number: ")
    shift = input("Enter the users shift (1, 2): ")
    pay = input("Enter the users hourly pay rate: ")    


    employee = ProductionWorker.ProductionWorker(name, number, shift, pay)
    empShift = employee.get_shift()
    print(employee)

main()

Answer 1

可以使用dict映射将移位到“1”和“2”，然后使用实例移位值进行查找。你知道吗

class ProductionWorker(employee.Employee):
    def __init__(self, name, number, shift, pay):
        employee.Employee.__init__(self, name, number)
        self.shifts = {"1":"Day shift","2":"Night shift"}  


def main():
    name = input("Enter the users name: ")
    number = input("Enter the users number: ")
    shift = input("Enter the users shift (1, 2): ")
    pay = input("Enter the users hourly pay rate: ")    


    employee = ProductionWorker.ProductionWorker(name, number, shift, pay)
    empShift = employee.get_shift()
    print(employee)
    print(employee.shifts[empShift])

main()

如果类中需要，可以使用方法，逻辑基本相同。你知道吗

要使用__str__打印所有详细信息，只需在get_shift方法中进行查找，而不只是返回self。\u shift:

class ProductionWorker.ProductionWorker(Employee):
    def __init__(self, name, number, shift, pay):
         employee.Employee.__init__(self, name, number)
        self.shifts = {"1": "Day shift", "2": "Night shift"}
        self.__shift = shift
        self.__pay = pay

    def set_shift(self, shift):
        self.__shift = shift

    def set_pay(self, pay):
        self.__pay = pay

    def get_shift(self):
        return self.shifts[self.__shift]

    def get_pay(self):
        return self.__pay

    def dayNight(self, shift):
        return self.shifts[shift]

    def __str__(self):
        return "\nName: {}\nNumber: {}\nShift: {}\nPay: {}".\
            format(self.get_name(), self.get_number(), self.dayNight(self.get_shift()), self.__pay)

输出：

Enter the users name: foo
Enter the users number: 1234
Enter the users shift (1, 2): 2
Enter the users hourly pay rate: 100

Name: foo
Number: 1234
Shift: Night shift
Pay: 100

如果要将DayNight方法与If/else结合使用，则必须将其与字符串（而不是int）进行比较，最简单的方法就是返回字符串：

    def set_shift(self, shift):
        self.__shift = shift

    def set_pay(self, pay):
        self.__pay = pay

    def get_shift(self):
        return self.__shift

    def get_pay(self):
        return self.__pay

    def dayNight(self, shift):
        if shift == "1":
            return"Day Shift"
        else:
            return "Night Shift"

    def __str__(self):
        return "\nName: {}\nNumber: {}\nShift: {}\nPay: {}".\
            format(self.get_name(), self.get_number(), self.dayNight(self.get_shift()), self.__pay)