register_models.py

Go to the documentation of this file.

from GBSModelRegistry     import  GetModelRegistry
from GBSModel             import  GBSModel

from GBSJobAnalyser       import GBSJobAnalyser
from GBSJob               import GBSJob
from GBSManager           import GBSManager
from GBSTask              import GBSTask

##  Register default model

description = """The default model which provides the core functionality.
Other models should only need fine tune to experiment
specific needs.
"""

m = GBSModel("default","The default model",description)

m.AddMapping("Manager", GBSManager, {})
m.AddMapping("Task",GBSTask,{})
m.AddMapping("Job", GBSJob, {})
m.AddMapping("JobAnalyser", GBSJobAnalyser,{})

GetModelRegistry().AddModel(m)

##  If the configuration variable UserModelsPath is defined, try to
##  add user extensions.

import os
from GBSConfig import GetConfigValue

user_path = GetConfigValue("UserModelsPath")
if user_path:
    user_path = os.path.expandvars(user_path)
    if not os.path.isdir(user_path):
        print "Warning: Cannot load user extensions, unable to find directory "\
              + user_path
    else:
        import sys
        sys.path.append(user_path)
        try:
            import register_user_models
            print "Loaded user extensions from " + user_path 
        except ImportError:
            print "Warning: Cannot load user extensions, No file register_user_models.py in "\
                  + user_path

---