Question

Assume you have a programme with multiple functions defined. Each function is called in a separate for loop. Is it possible to specify which function should be called via the command line?

Example:

python prog.py -x <<<filname>>>

Where -x tells python to go to a particular for loop and then execute the function called in that for loop?

Thanks,
Seafoid.

Answer 1

You want the sys module.

For example:

import sys

#Functions
def a(filename): pass
def b(filename): pass
def c(filename): pass

#Function chooser
func_arg = {"-a": a, "-b": b, "-c": c}

#Do it
if __name__ == "__main__":
    func_arg[sys.argv[1]](sys.argv[2])

Which runs a(filename) if you run python file.py -a filename