"""
The Tracking Module For Extracting Polygons from Heatmaps
"""


__autogen__ = """
mkinit geowatch.tasks.tracking --lazy --noattrs -w
EAGER_IMPORT_MODULES=geowatch python -c "from geowatch.tasks import tracking"
"""


def lazy_import(module_name, submodules, submod_attrs, eager='auto'):
    import importlib
    import os
    name_to_submod = {
        func: mod for mod, funcs in submod_attrs.items()
        for func in funcs
    }

    def __getattr__(name):
        if name in submodules:
            attr = importlib.import_module(
                '{module_name}.{name}'.format(
                    module_name=module_name, name=name)
            )
        elif name in name_to_submod:
            submodname = name_to_submod[name]
            module = importlib.import_module(
                '{module_name}.{submodname}'.format(
                    module_name=module_name, submodname=submodname)
            )
            attr = getattr(module, name)
        else:
            raise AttributeError(
                'No {module_name} attribute {name}'.format(
                    module_name=module_name, name=name))
        globals()[name] = attr
        return attr

    eager_import_flag = False
    if eager == 'auto':
        eager_import_text = os.environ.get('EAGER_IMPORT', '')
        if eager_import_text:
            eager_import_text_ = eager_import_text.lower()
            if eager_import_text_ in {'true', '1', 'on', 'yes'}:
                eager_import_flag = True

        eager_import_module_text = os.environ.get('EAGER_IMPORT_MODULES', '')
        if eager_import_module_text:
            if eager_import_module_text.lower() in __name__.lower():
                eager_import_flag = True
    else:
        eager_import_flag = eager

    if eager_import_flag:
        for name in submodules:
            __getattr__(name)

        for attrs in submod_attrs.values():
            for attr in attrs:
                __getattr__(attr)
    return __getattr__


__getattr__ = lazy_import(
    __name__,
    submodules={
        'abstract_classes',
        'agg_functions',
        'from_heatmap',
        'from_polygon',
        'normalize',
        'old_polygon_extraction',
        'phase',
        'polygon_extraction',
        'utils',
        'visualize',
    },
    submod_attrs={},
)


def __dir__():
    return __all__

__all__ = ['abstract_classes', 'agg_functions', 'from_heatmap', 'from_polygon',
           'normalize', 'old_polygon_extraction', 'phase',
           'polygon_extraction', 'utils', 'visualize']