List all factory boy factories in a way expected by shell_plus command from django-extensions. Python 3.7.

factories_lookup.py

import pathlib
import re


def factories_lookup(base_dir):
    """Yields all model factories in a way expected by shell_plus command.

    Returned generator is evaluated only when running shell_plus. Evaluation
    itself on my machine takes ± 3.63 ms.

    Usage:
        SHELL_PLUS_POST_IMPORTS = factories_lookup(BASE_DIR)
    """
    modules = pathlib.Path(base_dir).glob('**/factories.py')
    for module in modules:
        m = str(module)
        p = m.replace(base_dir, '')
        p = p[1:]
        p = p.replace('/', '.')
        p = p.replace('.py', '')
        with open(m, 'r') as f:
            c = re.findall('class ([A-Za-z]+Factory)', f.read())
            if c:
                yield p, c