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

LocalRuntimeInterpreterPath = \

    14

    os.environ.get("BEREMIZPYTHONPATH", sys.executable)

    15

    16

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

    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

                ("\"%s\" \"%s\" -p %s -i "+LocalHost+" %s %s") % (

    38

                    LocalRuntimeInterpreterPath,

    39

                    Bpath("Beremiz_service.py"),

    40

                    self.runtime_port,

    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

            shutil.rmtree(self.local_runtime_tmpdir)

    55

    56

            self.local_runtime = None

    57