1

#!/usr/bin/env python

     2

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

     3

import sys

     4

import tempfile

     5

import random

     6

import shutil

     7

from util.ProcessLogger import ProcessLogger

     8

from util.paths import Bpath

     9

    10

class LocalRuntimeMixin():

    11

    12

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

    13

        self.local_runtime_log = log

    14

        self.local_runtime = None

    15

        self.runtime_port = None

    16

        self.local_runtime_tmpdir = None

    17

        self.use_gui = use_gui

    18

    19

    def StartLocalRuntime(self):

    20

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

    21

            # create temporary directory for runtime working directory

    22

            self.local_runtime_tmpdir = tempfile.mkdtemp()

    23

            # choose an arbitrary random port for runtime

    24

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

    25

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

    26

            # launch local runtime

    27

            self.local_runtime = ProcessLogger(

    28

                self.local_runtime_log,

    29

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

    30

                    sys.executable,

    31

                    Bpath("Beremiz_service.py"),

    32

                    self.runtime_port,

    33

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

    34

                    self.local_runtime_tmpdir),

    35

                no_gui=False,

    36

                timeout=500, keyword=self.local_runtime_tmpdir,

    37

                cwd=self.local_runtime_tmpdir)

    38

            self.local_runtime.spin()

    39

        return self.runtime_port

    40

    41

    def KillLocalRuntime(self):

    42

        if self.local_runtime is not None:

    43

            # shutdown local runtime

    44

            self.local_runtime.kill(gently=False)

    45

            # clear temp dir

    46

            shutil.rmtree(self.local_runtime_tmpdir)

    47

    48

            self.local_runtime = None

    49