Flag per function - Argparse

flag_per_function.py

import argparse


class c1(object):
    def cleanup(self):
        print 'c1'


class c2(object):
    def cleanup(self):
        print 'c2'


class c3(object):
    def cleanup(self):
        print 'c3'


parser = argparse.ArgumentParser()

parser.add_argument('-1', const=c1, dest='cleaners', action='append_const')
parser.add_argument('-2', const=c2, dest='cleaners', action='append_const')
parser.add_argument('-3', const=c3, dest='cleaners', action='append_const')

args = parser.parse_args()


if args.cleaners is not None:
    for cleaner_class in args.cleaners:
        cleaner = cleaner_class()
        cleaner.cleanup()