1

#!/usr/bin/env python

     2

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

     3

     4

import os

     5

import sys

     6

from functools import wraps

     7

     8

import click

     9

    10

import fake_wx

    11

    12

from ProjectController import ProjectController

    13

from LocalRuntimeMixin import LocalRuntimeMixin

    14

    15

class Log:

    16

    17

    def __init__(self):

    18

        self.crlfpending = False

    19

    20

    def write(self, s):

    21

        if s:

    22

            if self.crlfpending:

    23

                sys.stdout.write("\n")

    24

            sys.stdout.write(s)

    26

            self.crlfpending = 0

    27

    28

    def write_error(self, s):

    29

        if s:

    30

            self.write("Error: "+s)

    31

    32

    def write_warning(self, s):

    33

        if s:

    34

            self.write("Warning: "+s)

    35

    36

    def flush(self):

    37

        sys.stdout.flush()

    38

    39

    def isatty(self):

    40

        return False

    41

    42

    def progress(self, s):

    43

        if s:

    44

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

    45

            self.crlfpending = True

    46

    47

    48

def with_project_loaded(func):

    49

    @wraps(func)

    50

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

    51

        if not self.HasOpenedProject():

    52

            if self.check_and_load_project():

    53

                return 1 

    54

            self.apply_config()

    55

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

    56

    57

    return func_wrapper

    58

    59

def connected(func):

    60

    @wraps(func)

    61

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

    62

        if self._connector is None:

    63

            if self.session.uri:

    64

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

    65

            if not self._Connect():

    66

                return 1

    67

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

    68

    69

    return func_wrapper

    70

    71

class CLIController(LocalRuntimeMixin, ProjectController):

    72

    def __init__(self, session):

    73

        self.session = session

    74

        log = Log()

    75

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

    76

        ProjectController.__init__(self, None, log)

    77

    78

    def check_and_load_project(self):

    79

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

    80

            self.logger.write_error(

    81

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

    82

            return True

    83

    84

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

    85

        if error:

    86

            self.logger.write_error(errmsg)

    87

            return True

    88

    89

    def apply_config(self):

    90

        for k,v in self.session.config:

    91

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

    92

    93

    @with_project_loaded

    94

    def build_project(self, target):

    95

    96

        if target:

    97

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

    98

    99

        return 0 if self._Build() else 1

   100

   101

    @with_project_loaded

   102

    @connected

   103

    def transfer_project(self):

   104

   105

        return 0 if self._Transfer() else 1

   106

   107

    @with_project_loaded

   108

    @connected

   109

    def run_project(self):

   110

   111

        return 0 if self._Run() else 1

   112

   113

   114

    def finish(self):

   115

   116

        self._Disconnect()

   117

   118

        if not self.session.keep:

   119

            self.KillLocalRuntime()

   120

   121