def testLoggingInitForUsageExamples(self):
g_log = logging.getLogger("my_example")
assert isinstance(g_log, logging.Logger)
with self._redirectLogBase("logging_example", _SAMPLE_CONNECTION_CONTENTS):
# Test logging init for a tool
self.assertEqual(LS.initTool(), LS.initLogging(None, console="stderr", logToFile=True))
# Test logging init for a service
self.assertEqual(LS.initService(), LS.initLogging(None, console="stderr", logToFile=False))
# Test logging init for a test app
self.assertEqual(LS.initTestApp(), LS.initLogging(None, console="stderr", logToFile=False))
def testLoggingInitForUsageExamples(self):
g_log = logging.getLogger("my_example")
assert isinstance(g_log, logging.Logger)
with self._redirectLogBase("logging_example", _SAMPLE_CONNECTION_CONTENTS):
# Test logging init for a tool
self.assertEqual(LS.initTool(), LS.initLogging(None, console="stderr", logToFile=True))
# Test logging init for a service
self.assertEqual(LS.initService(), LS.initLogging(None, console="stderr", logToFile=False))
# Test logging init for a test app
self.assertEqual(LS.initTestApp(), LS.initLogging(None, console="stderr", logToFile=False))
(options, args) = parser.parse_args(argv[1:])
if len(args) > 0:
parser.error("Didn't expect any positional args (%r)." % (args,))
options.concurrency = _getDefaultConcurrency(options.concurrency)
needRestart = True
while needRestart:
with ModelSchedulerService(concurrency=options.concurrency) as scheduler:
needRestart = scheduler.run()
if __name__ == "__main__":
LoggingSupport.initService()
logger = _getLogger()
logger.setLogPrefix('<%s, SERVICE=SCHED> ' % getStandardLogPrefix())
try:
logger.info("{TAG:SWAP.SCHED.START}")
main(sys.argv)
except SystemExit as e:
if e.code != 0:
logger.exception("{TAG:SWAP.SCHED.STOP.ABORT}")
raise
except:
logger.exception("{TAG:SWAP.SCHED.STOP.ABORT}")
raise
(options, args) = parser.parse_args(argv[1:])
if len(args) > 0:
parser.error("Didn't expect any positional args (%r)." % (args, ))
if options.modelID is None:
parser.error("Missing model ID in command-line")
with ModelRunner(modelID=options.modelID) as runner:
runner.run()
if __name__ == "__main__":
# NOTE: we initialize logging as a service so that our logging output will be
# funneled via console to our parent process (model_scheduler service)
LoggingSupport.initService()
logger = _getLogger()
logger.setLogPrefix('<%s, SERVICE=MRUN> ' % getStandardLogPrefix())
try:
logger.info("{TAG:SWAP.MR.START} argv=%r", sys.argv)
main(sys.argv)
except SystemExit as e:
if e.code != 0:
logger.exception("{TAG:SWAP.MR.STOP.ABORT}")
raise
except:
logger.exception("{TAG:SWAP.MR.STOP.ABORT}")
raise