

def add_to_module_map(domain, module):
    db = get_db()
    if db.doc_exist(DOMAIN_MODULE_KEY):
        module_config = db.open_doc(DOMAIN_MODULE_KEY)
        module_map = module_config.get('module_map')
        if module_map:
            module_map[domain] = module
        else:
            module_config['module_map'][domain] = module
    else:
        module_config = db.save_doc({
            '_id': DOMAIN_MODULE_KEY,
            'module_map': {
                domain: module,
            },
        })
    db.save_doc(module_config)

dataset/ETHPy150Open dimagi/commcare-hq/custom/utils/utils.py add_to_module_map/original