FWCore/ParameterSet/python/processFromFile.py - master - githubmirror/cmssw

FWCore/ParameterSet/python/processFromFile.py

28 строк749 B

Andrea Bocci

Make processFromFile behave like cmsRun

26 фев 2026, 16:55

26 фев 2026, 16:554e9e13a

import os
import sys
import types
from importlib.machinery import SourceFileLoader

def processFromFile(filename:str, args=None):
    # make processFromFile behave like cmsRun
    sys.path.insert(0, os.path.dirname(os.path.abspath(filename)))

    # pass any optional arguments to the file being loaded
    old_sys_argv = None
    if args is not None:
        old_sys_argv = sys.argv[:]
        sys.argv = [filename]+args

    loader = SourceFileLoader("pycfg", filename)
    mod = types.ModuleType(loader.name)
    loader.exec_module(mod)
    process = mod.process

    # restore the original arguments
    if old_sys_argv is not None:
        sys.argv = old_sys_argv

    # restore the original environment
    sys.path.pop(0)

    return process