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, ToDoBeforeQuit

    15

from LocalRuntimeMixin import LocalRuntimeMixin

    16

from runtime.loglevels import LogLevelsCount, LogLevels

    17

    18

    19

class Log:

    20

    21

    def __init__(self):

    22

        self.last_progress = None

    23

    24

    def write(self, s):

    25

        if s:

    26

            if self.last_progress:

    27

                sys.stdout.write(f"{self.last_progress}\n")

    28

                self.last_progress = None

    29

            sys.stdout.write(s)

    30

            sys.stdout.flush()

    31

            self.crlfpending = 0

    32

    33

    def write_error(self, s):

    34

        if s:

    35

            self.write("Error: "+s)

    36

    37

    def write_warning(self, s):

    38

        if s:

    39

            self.write("Warning: "+s)

    40

    41

    def flush(self):

    42

        sys.stdout.flush()

    43

    44

    def isatty(self):

    45

        return False

    46

    47

    def progress(self, s):

    48

        self.last_progress = s

    49

    50

def with_project_loaded(func):

    51

    @wraps(func)

    52

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

    53

        if not self.HasOpenedProject():

    54

            if self.check_and_load_project():

    55

                return 1 

    56

            self.apply_config()

    57

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

    58

    59

    return func_wrapper

    60

    61

def connected(func):

    62

    @wraps(func)

    63

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

    64

        if self._connector is None:

    65

            if self.session.uri:

    66

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

    67

            if not self._Connect():

    68

                return 1

    69

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

    70

    71

    return func_wrapper

    72

    73

class CLIController(LocalRuntimeMixin, ProjectController):

    74

    def __init__(self, session):

    75

        self.session = session

    76

        log = Log()

    77

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

    78

        ProjectController.__init__(self, None, log)

    79

    80

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

    81

        self._connector = connector

    82

        self.previous_log_count = [None]*LogLevelsCount

    83

        if connector is None and update_status:

    84

                self.UpdateMethodsFromPLCStatus()

    85

    86

    def UpdatePLCLog(self, log_count):

    87

        connector = self._connector

    88

        new_messages = []

    89

        if connector:

    90

            for level, count, prev in zip(

    91

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

    92

                if count is not None and prev != count:

    93

                    if prev is None:

    94

                        dump_end = max(-1, count - 10)

    95

                    else:

    96

                        dump_end = prev - 1

    97

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

    98

                        message = connector.GetLogMessage(level, msgidx)

    99

                        if message is not None:

   100

                            msg, _tick, tv_sec, tv_nsec = message

   101

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

   102

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

   103

                            new_messages.append((date,txt))

   104

                        else:

   105

                            break

   106

                self.previous_log_count[level] = count

   107

            new_messages.sort()

   108

            for date, txt in new_messages:

   109

                self.logger.write(txt)

   110

   111

    def check_and_load_project(self):

   112

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

   113

            self.logger.write_error(

   114

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

   115

            return True

   116

   117

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

   118

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

   119

   120

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

   121

        if error:

   122

            self.logger.write_error(errmsg)

   123

            return True

   124

   125

    def apply_config(self):

   126

        for k,v in self.session.config:

   127

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

   128

   129

    @with_project_loaded

   130

    def build_project(self, target):

   131

   132

        if target:

   133

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

   134

   135

        return 0 if self._Build() else 1

   136

   137

    @with_project_loaded

   138

    @connected

   139

    def transfer_project(self):

   140

   141

        return 0 if self._Transfer() else 1

   142

   143

    @with_project_loaded

   144

    @connected

   145

    def run_project(self):

   146

   147

        return 0 if self._Run() else 1

   148

   149

    @with_project_loaded

   150

    @connected

   151

    def stop_project(self):

   152

   153

        return 0 if self._Stop() else 1

   154

   155

   156

    def finish(self):

   157

        global ToDoBeforeQuit

   158

   159

        self._Disconnect()

   160

   161

        for Thing in ToDoBeforeQuit:

   162

            Thing()

   163

        ToDoBeforeQuit = []

   164

   165

        if not self.session.keep:

   166

            self.KillLocalRuntime()

   167

   168