Function get_plain_module_render_state

partial_rendering.py:73–116

(plain_module: PlainModule)

def get_plain_module_render_state(plain_module: PlainModule) -> PlainModuleRenderState | None:
    sc = spec_change(plain_module)
    cc = code_change(plain_module)
    all_required_modules = plain_module.all_required_modules
    last_rendered_module_name, last_rendered_frid = plain_module.get_module_render_status()
    if last_rendered_module_name is None and last_rendered_frid is None:
        return None

    if last_rendered_module_name == plain_module.module_name:
        module = plain_module
    else:
        found_module: PlainModule | None = None
        for required_module in all_required_modules:
            if required_module.module_name == last_rendered_module_name:
                found_module = required_module

        if found_module is None:
            raise ModuleDoesNotExistError(
                f"Last rendered module {last_rendered_module_name} not found in {[rmodule.module_name for rmodule in all_required_modules]}"
            )
        module = found_module

    pr = PlainModuleRenderState(
        last_render_module=module,
        last_render_frid=last_rendered_frid,
        change=None,
        change_type=None,
    )

    if sc is None and cc is None:
        return pr

    if sc is not None:
        pr.change = sc
        pr.change_type = "spec_change"

    if cc is not None and (
        pr.change is None
        or (pr.change is not None and module_comes_before_or_equal(all_required_modules, cc, pr.change))
    ):
        pr.change = cc
        pr.change_type = "code_change"

    return pr


def get_all_affected_modules_from_change(