1

#!/usr/bin/env python

     2

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

     3

     4

import os

     5

import sys

     6

from functools import wraps

     7

from threading import Timer

     8

from datetime import datetime

     9

    10

import click

    11

    12

import fake_wx

    13

    14

from ProjectController import ProjectController

    15

from LocalRuntimeMixin import LocalRuntimeMixin

    16

from runtime.loglevels import LogLevelsCount, LogLevels

    17

    18

    19

class Log:

    20

    21

    def __init__(self):

    22

        self.crlfpending = False

    23

    24

    def write(self, s):

    25

        if s:

    26

            if self.crlfpending:

    27

                sys.stdout.write("\n")

    28

            sys.stdout.write(s)

    29

            sys.stdout.flush()

    30

            self.crlfpending = 0

    31

    32

    def write_error(self, s):

    33

        if s:

    34

            self.write("Error: "+s)

    35

    36

    def write_warning(self, s):

    37

        if s:

    38

            self.write("Warning: "+s)

    39

    40

    def flush(self):

    41

        sys.stdout.flush()

    42

    43

    def isatty(self):

    44

        return False

    45

    46

    def progress(self, s):

    47

        if s:

    48

            sys.stdout.write(s+"\r")

    49

            self.crlfpending = True

    50

    51

    52

def with_project_loaded(func):

    53

    @wraps(func)

    54

    def func_wrapper(self, *args, **kwargs):

    55

        if not self.HasOpenedProject():

    56

            if self.check_and_load_project():

    57

                return 1 

    58

            self.apply_config()

    59

        return func(self, *args, **kwargs)

    60

    61

    return func_wrapper

    62

    63

def connected(func):

    64

    @wraps(func)

    65

    def func_wrapper(self, *args, **kwargs):

    66

        if self._connector is None:

    67

            if self.session.uri:

    68

                self.BeremizRoot.setURI_location(self.session.uri)

    69

            if not self._Connect():

    70

                return 1

    71

        return func(self, *args, **kwargs)

    72

    73

    return func_wrapper

    74

    75

class CLIController(LocalRuntimeMixin, ProjectController):

    76

    def __init__(self, session):

    77

        self.session = session

    78

        log = Log()

    79

        LocalRuntimeMixin.__init__(self, log, use_gui=False)

    80

        ProjectController.__init__(self, None, log)

    81

        self.CLIStatusTimer = None

    82

        self.KillCLIStatusTimer = False

    83

    84

    85

    def StartCLIStatusTimer(self):

    86

        if self.CLIStatusTimer is not None:

    87

            return

    88

        self.CLIStatusTimer = Timer(0.5, self.CLIStatusTimerProc)

    89

        self.KillCLIStatusTimer = False

    90

        self.CLIStatusTimer.start()

    91

    92

    def StopCLIStatusTimer(self):

    93

        if self.CLIStatusTimer is None:

    94

            return

    95

        self.KillCLIStatusTimer = True

    96

        self.CLIStatusTimer.cancel()

    97

        self.CLIStatusTimer = None

    98

    99

    def CLIStatusTimerProc(self):

   100

        self.CLIStatusTimer = None

   101

        if not self.KillCLIStatusTimer:

   102

            self.PullPLCStatusProc(None)

   103

            self.StartCLIStatusTimer()

   104

   105

    def _SetConnector(self, connector, update_status=True):

   106

        self._connector = connector

   107

        self.previous_log_count = [None]*LogLevelsCount

   108

        if connector is not None:

   109

            self.StartCLIStatusTimer()

   110

        else:

   111

            self.StopCLIStatusTimer()

   112

            if update_status:

   113

                self.UpdateMethodsFromPLCStatus()

   114

   115

    def UpdatePLCLog(self, log_count):

   116

        connector = self._connector

   117

        new_messages = []

   118

        if connector:

   119

            for level, count, prev in zip(

   120

                xrange(LogLevelsCount), log_count, self.previous_log_count):

   121

                if count is not None and prev != count:

   122

                    if prev is None:

   123

                        dump_end = max(-1, count - 10)

   124

                    else:

   125

                        dump_end = prev - 1

   126

                    for msgidx in range(count-1, dump_end, -1):

   127

                        message = connector.GetLogMessage(level, msgidx)

   128

                        if message is not None:

   129

                            msg, _tick, tv_sec, tv_nsec = message

   130

                            date = datetime.utcfromtimestamp(tv_sec + tv_nsec * 1e-9)

   131

                            txt = "%s at %s: %s\n" % (LogLevels[level], date.isoformat(' '), msg)

   132

                            new_messages.append((date,txt))

   133

                        else:

   134

                            break

   135

                self.previous_log_count[level] = count

   136

            new_messages.sort()

   137

            for date, txt in new_messages:

   138

                self.logger.write(txt)

   139

   140

    def check_and_load_project(self):

   141

        if not os.path.isdir(self.session.project_home):

   142

            self.logger.write_error(

   143

                _("\"%s\" is not a valid Beremiz project\n") % self.session.project_home)

   144

            return True

   145

   146

        errmsg, error = self.LoadProject(self.session.project_home, self.session.buildpath)

   147

        if error:

   148

            self.logger.write_error(errmsg)

   149

            return True

   150

   151

    def apply_config(self):

   152

        for k,v in self.session.config:

   153

            self.SetParamsAttribute("BeremizRoot."+k, v)

   154

   155

    @with_project_loaded

   156

    def build_project(self, target):

   157

   158

        if target:

   159

            self.SetParamsAttribute("BeremizRoot.TargetType", target)

   160

   161

        return 0 if self._Build() else 1

   162

   163

    @with_project_loaded

   164

    @connected

   165

    def transfer_project(self):

   166

   167

        return 0 if self._Transfer() else 1

   168

   169

    @with_project_loaded

   170

    @connected

   171

    def run_project(self):

   172

   173

        return 0 if self._Run() else 1

   174

   175

   176

    def finish(self):

   177

   178

        self._Disconnect()

   179

   180

        if not self.session.keep:

   181

            self.KillLocalRuntime()

   182

   183