1

#!/usr/bin/env python

     2

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

     5

import sys

     6

import tempfile

     7

import random

     8

import shutil

     9

from util.ProcessLogger import ProcessLogger

    10

from util.paths import Bpath

    11

    17

class LocalRuntimeMixin():

    18

    19

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

    20

        self.local_runtime_log = log

    21

        self.local_runtime = None

    22

        self.runtime_port = None

    23

        self.local_runtime_tmpdir = None

    24

        self.use_gui = use_gui

    25

    26

    def StartLocalRuntime(self):

    27

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

    28

            # create temporary directory for runtime working directory

    29

            self.local_runtime_tmpdir = tempfile.mkdtemp()

    30

            # choose an arbitrary random port for runtime

    31

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

    32

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

    33

            # launch local runtime

    34

            self.local_runtime = ProcessLogger(

    35

                self.local_runtime_log,

    36

                "\"%s\" \"%s\" -p %s -i localhost %s %s" % (

    38

                    Bpath("Beremiz_service.py"),

    39

                    self.runtime_port,

    40

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

    41

                    self.local_runtime_tmpdir),

    42

                no_gui=False,

    43

                timeout=500, keyword=self.local_runtime_tmpdir,

    44

                cwd=self.local_runtime_tmpdir)

    45

            self.local_runtime.spin()

    46

        return self.runtime_port

    47

    48

    def KillLocalRuntime(self):

    49

        if self.local_runtime is not None:

    50

            # shutdown local runtime

    51

            self.local_runtime.kill(gently=False)

    52

            # clear temp dir

    53

            shutil.rmtree(self.local_runtime_tmpdir)

    54

    55

            self.local_runtime = None

    56