Optionset generator

optionset_generator.py

import collections
import itertools

def optionset_generator(optionset_schema):
    iterables = {}
    
    for key, value in optionset_schema.items():
        if isinstance(value, collections.Iterable):
            iterables[key] = value
        else:
            continue

    for resolved in itertools.product(*iterables.values()):
        optionset = {}
        for key, old_value in optionset_schema.items():
            if key not in iterables:
                optionset[key] = old_value
            else:
                optionset[key] = resolved[list(iterables.keys()).index(key)]
        yield optionset
        
## Usage
## >> optionset_schema = {
##        'window': range(1,4),
##        'chunk_size': 32,
##        'sg': [True, False]
##    }
##
## >> [x for x in optionset_generator(optionset_schema)]
## [{'chunk_size': 32, 'sg': True, 'window': 1},
## {'chunk_size': 32, 'sg': False, 'window': 1},
## {'chunk_size': 32, 'sg': True, 'window': 2},
## {'chunk_size': 32, 'sg': False, 'window': 2},
## {'chunk_size': 32, 'sg': True, 'window': 3},
## {'chunk_size': 32, 'sg': False, 'window': 3}]