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)

    25

            self.crlfpending = 0

    26

    27

    def write_error(self, s):

    28

        if s:

    29

            self.write("Error: "+s)

    30

    31

    def write_warning(self, s):

    32

        if s:

    33

            self.write("Warning: "+s)

    34

    35

    def flush(self):

    36

        sys.stdout.flush()

    37

    38

    def isatty(self):

    39

        return False

    40

    41

    def progress(self, s):

    42

        if s:

    43

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

    44

            self.crlfpending = True

    45

    46

    47

def with_project_loaded(func):

    48

    @wraps(func)

    49

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

    50

        if not self.HasOpenedProject():

    51

            if self.check_and_load_project():

    52

                return 1 

    53

            self.apply_config()

    54

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

    55

    56

    return func_wrapper

    57

    58

def connected(func):

    59

    @wraps(func)

    60

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

    61

        if self._connector is None:

    62

            if self.session.uri:

    63

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

    64

            if not self._Connect():

    65

                return 1

    66

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

    67

    68

    return func_wrapper

    69

    70

class CLIController(LocalRuntimeMixin, ProjectController):

    71

    def __init__(self, session):

    72

        self.session = session

    73

        log = Log()

    74

        LocalRuntimeMixin.__init__(self, log)

    75

        ProjectController.__init__(self, None, log)

    76

    77

    def check_and_load_project(self):

    78

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

    79

            self.logger.write_error(

    80

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

    81

            return True

    82

    83

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

    84

        if error:

    85

            self.logger.write_error(errmsg)

    86

            return True

    87

    88

    def apply_config(self):

    89

        for k,v in self.session.config:

    90

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

    91

    92

    @with_project_loaded

    93

    def build_project(self, target):

    94

    95

        if target:

    96

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

    97

    98

        return 0 if self._Build() else 1

    99

   100

    @with_project_loaded

   101

    @connected

   102

    def transfer_project(self):

   103

   104

        return 0 if self._Transfer() else 1

   105

   106

    @with_project_loaded

   107

    @connected

   108

    def run_project(self):

   109

   110

        return 0 if self._Run() else 1

   111

   112

   113

    def finish(self):

   114

   115

        self._Disconnect()

   116

   117

        if not self.session.keep:

   118

            self.KillLocalRuntime()

   119

   120