1

#!/usr/bin/env python

     2

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

     3

     4

import os

     5

import posixpath

     6

import sys

     7

import time

     8

     9

from functools import wraps

    10

from importlib import import_module

    11

    12

import click

    13

    14

class CLISession(object):

    15

    def __init__(self, **kwargs):

    16

        self.__dict__.update(kwargs)

    17

        self.controller = None

    18

    19

pass_session = click.make_pass_decorator(CLISession)

    20

    21

    22

@click.group(chain=True)

    23

@click.option(

    24

    "--project-home",

    25

    envvar="PROJECT_HOME",

    26

    default=".",

    27

    metavar="PATH",

    28

    help="Changes the project folder location.",

    29

)

    30

@click.option(

    31

    "--config",

    32

    nargs=2,

    33

    multiple=True,

    34

    metavar="KEY VALUE",

    35

    help="Overrides a config key/value pair.",

    36

)

    37

@click.option(

    38

    "--keep", "-k", is_flag=True,

    39

    help="Keep local runtime, do not kill it after executing commands.",

    40

)

    41

@click.option("--verbose", "-v", is_flag=True, help="Enables verbose mode.")

    42

@click.option(

    43

    "--buildpath", "-b", help="Where to store files created during build."

    44

)

    45

@click.option(

    46

    "--uri", "-u", help="URI to reach remote PLC."

    47

)

    48

@click.version_option("0.1")

    49

@click.pass_context

    50

def cli(ctx, **kwargs):

    51

    """Beremiz CLI manipulates beremiz projects and runtimes. """

    52

    53

    ctx.obj = CLISession(**kwargs)

    54

    55

def ensure_controller(func):

    56

    @wraps(func)

    57

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

    58

        if session.controller is None:

    59

            session.controller = import_module("CLIController").CLIController(session)

    60

        ret = func(session, *args, **kwargs)

    61

        return ret

    62

    63

    return func_wrapper

    64

    65

@cli.command()

    66

@click.option(

    67

    "--target", "-t", help="Target system triplet."

    68

)

    69

@pass_session

    70

@ensure_controller

    71

def build(session, target):

    72

    """Builds project. """

    73

    def processor():

    74

        return session.controller.build_project(target)

    75

    return processor

    76

    77

@cli.command()

    78

@pass_session

    79

@ensure_controller

    80

def transfer(session):

    81

    """Transfer program to PLC runtim."""

    82

    def processor():

    83

        return session.controller.transfer_project()

    84

    return processor

    85

    86

@cli.command()

    87

@pass_session

    88

@ensure_controller

    89

def run(session):

    90

    """Run program already present in PLC. """

    91

    def processor():

    92

        return session.controller.run_project()

    93

    return processor

    94

    95

    96

@cli.resultcallback()

    97

@pass_session

    98

def process_pipeline(session, processors, **kwargs):

    99

    ret = 0

   100

    for processor in processors:

   101

        ret = processor()

   102

        if ret != 0:

   103

            if len(processors) > 1 :

   104

                click.echo("Command sequence aborted")

   105

            break

   106

   107

    if session.keep:

   108

        click.echo("Press Ctrl+C to quit")

   109

        try:

   110

            while True:

   111

                time.sleep(1)

   112

        except KeyboardInterrupt:

   113

            pass

   114

   115

    session.controller.finish()

   116

   117

    return ret

   118

   119

if __name__ == '__main__':

   120

    cli()

   121