python - pyProcedure method which adds a new mission

Question

class Rocket:

    def __init__(self, name, mission):
        self.__name = name      
        self.__mission = mission

    def getMission(self): 
        return self.__mission
    def addMission(self, mission): 
        # procedure method which adds a new mission. There can be one (str) or multiple (list) existing missions

I can't understand how to implement this addMission func within a class correctly.

Output should be something like this: Missions: ['Dragon 2 pad abort test', 'Dragon 2 in-flight abort test']. What I've "come up with" simply produces the right answer if that default input is placed, albeit it has the wrong inner mechanics. Thus, with different input it won't produce a thing:

self.__mission = [self.__mission, mission]

How do I... make it work?

question from:https://stackoverflow.com/questions/65901140/pyprocedure-method-which-adds-a-new-mission

Answer 1

Try this:

class Rocket:
    def __init__(self, name, mission):
        if not type(mission) in [list, tuple]:
            raise TypeError("constructor argument mission must be of type list or tuple")
        self.__name = name      
        self.__mission = mission

    def getMission(self): 
        return self.__mission
    def addMission(self, mission): 
        # procedure method which adds a new mission. There can be one (str) or multiple (list) existing missions
        if type(mission) == str:
            self.__mission.append(str)
        elif type(mission) in [list, tuple]:
                self.__mission.extend(miss)
        else:
            raise TypeError("addMission argument mission must have types str, list or tuple")