edouard@3501: #!/usr/bin/env python
edouard@3501: # -*- coding: utf-8 -*-
edouard@3566: 
edouard@3566: import os
edouard@3501: import sys
edouard@3501: import tempfile
edouard@3501: import random
edouard@3501: import shutil
edouard@3501: from util.ProcessLogger import ProcessLogger
edouard@3501: from util.paths import Bpath
edouard@3501: 
edouard@3566: LocalRuntimeInterpreterPath = \
edouard@3566:     os.environ["BEREMIZPYTHONPATH"] \
edouard@3566:     if os.environ.has_key("BEREMIZPYTHONPATH") \
edouard@3566:     else sys.executable
edouard@3566: 
edouard@3501: class LocalRuntimeMixin():
edouard@3501: 
edouard@3539:     def __init__(self, log, use_gui=True):
edouard@3501:         self.local_runtime_log = log
edouard@3501:         self.local_runtime = None
edouard@3501:         self.runtime_port = None
edouard@3501:         self.local_runtime_tmpdir = None
edouard@3539:         self.use_gui = use_gui
edouard@3501: 
edouard@3539:     def StartLocalRuntime(self):
edouard@3501:         if (self.local_runtime is None) or (self.local_runtime.exitcode is not None):
edouard@3501:             # create temporary directory for runtime working directory
edouard@3501:             self.local_runtime_tmpdir = tempfile.mkdtemp()
edouard@3501:             # choose an arbitrary random port for runtime
edouard@3501:             self.runtime_port = int(random.random() * 1000) + 61131
edouard@3501:             self.local_runtime_log.write(_("Starting local runtime...\n"))
edouard@3501:             # launch local runtime
edouard@3501:             self.local_runtime = ProcessLogger(
edouard@3501:                 self.local_runtime_log,
edouard@3501:                 "\"%s\" \"%s\" -p %s -i localhost %s %s" % (
edouard@3566:                     LocalRuntimeInterpreterPath,
edouard@3501:                     Bpath("Beremiz_service.py"),
edouard@3501:                     self.runtime_port,
edouard@3539:                     {False: "-x 0", True: "-x 1"}[self.use_gui],
edouard@3501:                     self.local_runtime_tmpdir),
edouard@3501:                 no_gui=False,
edouard@3501:                 timeout=500, keyword=self.local_runtime_tmpdir,
edouard@3501:                 cwd=self.local_runtime_tmpdir)
edouard@3501:             self.local_runtime.spin()
edouard@3501:         return self.runtime_port
edouard@3501: 
edouard@3501:     def KillLocalRuntime(self):
edouard@3501:         if self.local_runtime is not None:
edouard@3501:             # shutdown local runtime
edouard@3501:             self.local_runtime.kill(gently=False)
edouard@3501:             # clear temp dir
edouard@3501:             shutil.rmtree(self.local_runtime_tmpdir)
edouard@3501: 
edouard@3501:             self.local_runtime = None
edouard@3501: