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

    82

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

    83

        self._connector = connector

    84

        self.previous_log_count = [None]*LogLevelsCount

    85

        if connector is None and update_status:

    86

                self.UpdateMethodsFromPLCStatus()

    87

    88

    def UpdatePLCLog(self, log_count):

    89

        connector = self._connector

    90

        new_messages = []

    91

        if connector:

    92

            for level, count, prev in zip(

    93

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

    94

                if count is not None and prev != count:

    95

                    if prev is None:

    96

                        dump_end = max(-1, count - 10)

    97

                    else:

    98

                        dump_end = prev - 1

    99

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

   100

                        message = connector.GetLogMessage(level, msgidx)

   101

                        if message is not None:

   102

                            msg, _tick, tv_sec, tv_nsec = message

   103

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

   104

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

   105

                            new_messages.append((date,txt))

   106

                        else:

   107

                            break

   108

                self.previous_log_count[level] = count

   109

            new_messages.sort()

   110

            for date, txt in new_messages:

   111

                self.logger.write(txt)

   112

   113

    def check_and_load_project(self):

   114

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

   115

            self.logger.write_error(

   116

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

   117

            return True

   118

   119

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

   120

            self.session.project_home = os.path.join(os.getcwd(), self.session.project_home)

   121

   122

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

   123

        if error:

   124

            self.logger.write_error(errmsg)

   125

            return True

   126

   127

    def apply_config(self):

   128

        for k,v in self.session.config:

   129

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

   130

   131

    @with_project_loaded

   132

    def build_project(self, target):

   133

   134

        if target:

   135

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

   136

   137

        return 0 if self._Build() else 1

   138

   139

    @with_project_loaded

   140

    @connected

   141

    def transfer_project(self):

   142

   143

        return 0 if self._Transfer() else 1

   144

   145

    @with_project_loaded

   146

    @connected

   147

    def run_project(self):

   148

   149

        return 0 if self._Run() else 1

   150

   151

    @with_project_loaded

   152

    @connected

   153

    def stop_project(self):

   154

   155

        return 0 if self._Stop() else 1

   156

   157

   158

    def finish(self):

   159

   160

        self._Disconnect()

   161

   162

        if not self.session.keep:

   163

            self.KillLocalRuntime()

   164

   165