1

#!/usr/bin/env python

     2

# -*- coding: utf-8 -*-

     3

     4

import os

     5

import sys

     6

import tempfile

     7

import random

     8

import shutil

     9

    10

from util.ProcessLogger import ProcessLogger

    11

from util.paths import Bpath

    12

    13

_exec = sys.executable if "darwin" not in sys.platform else sys.executable + 'w'

    14

LocalRuntimeInterpreterPath = os.environ.get("BEREMIZPYTHONPATH", _exec)

    15

    16

LocalHost = os.environ.get("BEREMIZ_LOCAL_HOST", "127.0.0.1")

    17

    18

class LocalRuntimeMixin():

    19

    20

    def __init__(self, log, use_gui=True):

    21

        self.local_runtime_log = log

    22

        self.local_runtime = None

    23

        self.runtime_port = None

    24

        self.local_runtime_tmpdir = None

    25

        self.use_gui = use_gui

    26

    27

    def StartLocalRuntime(self):

    28

        if (self.local_runtime is None) or (self.local_runtime.exitcode is not None):

    29

            # create temporary directory for runtime working directory

    30

            self.local_runtime_tmpdir = tempfile.mkdtemp()

    31

            # choose an arbitrary random port for runtime

    32

            self.runtime_port = int(random.random() * 1000) + 61131

    33

            self.local_runtime_log.write(_("Starting local runtime...\n"))

    34

            # launch local runtime

    35

            self.local_runtime = ProcessLogger(

    36

                self.local_runtime_log,[

    37

                    LocalRuntimeInterpreterPath,

    38

                    Bpath("Beremiz_service.py"),

    39

                    "-p", str(self.runtime_port),

    40

                    "-i", LocalHost,

    41

                    {False: "-x 0", True: "-x 1"}[self.use_gui],

    42

                    self.local_runtime_tmpdir],

    43

                no_gui=False,

    44

                timeout=500, keyword=self.local_runtime_tmpdir,

    45

                cwd=self.local_runtime_tmpdir)

    46

            self.local_runtime.spin()

    47

        return self.runtime_port

    48

    49

    def KillLocalRuntime(self):

    50

        if self.local_runtime is not None:

    51

            # shutdown local runtime

    52

            self.local_runtime.kill(gently=False)

    53

            # clear temp dir

    54

            try:

    55

                shutil.rmtree(self.local_runtime_tmpdir)

    56

            except:

    57

                pass

    58

    59

            self.local_runtime = None

    60