I have some topic to discuss. I have a fragment of code with 24 if
s/elif
s. Operation
is my own class that represents functionality similar to Enum
.
Here is a fragment of code:
if operation == Operation.START:
strategy = strategy_objects.StartObject()
elif operation == Operation.STOP:
strategy = strategy_objects.StopObject()
elif operation == Operation.STATUS:
strategy = strategy_objects.StatusObject()
(...)
I have concerns from readability point of view. Is is better to change it into 24 classes and use polymorphism? I am not convinced that it will make my code maintainable... From one hand those if
s are pretty clear and it shouldn't be hard to follow, on the other hand there are too many if
s.
My question is rather general, however I'm writing code in Python so I cannot use constructions like switch
.
What do you think?
UPDATE:
One important thing is that StartObject()
, StopObject()
and StatusObject()
are constructors and I wanted to assign an object to strategy
reference.
You could possibly use a dictionary. Dictionaries store references, which means functions are perfectly viable to use, like so:
operationFuncs = {
Operation.START: strategy_objects.StartObject
Operation.STOP: strategy_objects.StopObject
Operation.STATUS: strategy_objects.StatusObject
(...)
}
It's good to have a default operation just in case, so when you run it use a try except
and handle the exception (ie. the equivalent of your else
clause)
try:
strategy = operationFuncs[operation]()
except KeyError:
strategy = strategy_objects.DefaultObject()
Alternatively use a dictionary's get
method, which allows you to specify a default if the key you provide isn't found.
strategy = operationFuncs.get(operation(), DefaultObject())
Note that you don't include the parentheses when storing them in the dictionary, you just use them when calling your dictionary. Also this requires that Operation.START
be hashable, but that should be the case since you described it as a class similar to an ENUM.