Source code for ejpiaj.registry

EXTRACTORS = {}
ASSERTIONS = {}


[docs]def register_variables_extractor(name, extractor):
    EXTRACTORS[name] = extractor


[docs]def unregister_variables_extractor(name):
    del EXTRACTORS[name]


[docs]def get_variables_extractors():
    return EXTRACTORS


[docs]def get_variables_extractor(name):
    try:
        return EXTRACTORS[name]
    except KeyError:
        raise UnregisteredVariablesExtractor(name)


[docs]class UnregisteredVariablesExtractor(Exception):
    pass


[docs]def register_assertion(name, assertion):
    ASSERTIONS[name] = assertion


[docs]def unregister_assertion(name):
    del ASSERTIONS[name]


[docs]def get_assertions():
    return ASSERTIONS


[docs]def get_assertion(name):
    try:
        return ASSERTIONS[name]
    except KeyError:
        raise UnregisteredAssertion(name)


[docs]class UnregisteredAssertion(Exception):
    pass