extract_notifications.py

from itertools import groupby
from operator import itemgetter


def get_subclasses(cls: type):
    for subclass in cls.__subclasses__():
        yield subclass
        yield from get_subclasses(subclass)


def get_notifications(cls: type, prefixes: list):
    classes = map(
        lambda c: (c.__module__, c.__qualname__),
        filter(
            lambda c: not any(map(c.__name__.startswith, prefixes)),
            get_subclasses(cls),
        )
    )
    return sorted(classes)

def format_for_import(classes, single_line=False):
    if single_line:
        for module, cls in classes:
            yield f"from {module} import {cls}"
    else:
        for module, group in groupby(classes, itemgetter(0)):
            classes = map(itemgetter(1), group)
            yield f"from {module} import ({', '.join(classes)})"


if __name__ == '__main__':
    from .notification.base import AbstractNotification
    
    notifications = get_notifications(AbstractNotification, ['_', 'Abstract', 'Base'])

    for statement in format_for_import(notifications, True):
        print(f'{statement}')