aboutsummaryrefslogtreecommitdiffstats
path: root/tools/docker/test-containers/trafficgen-pods/trex/vppconf.py
blob: ad7d63d88ff4efc49ac33df726e2a5c1df618822 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
import select
import subprocess
import logging
import threading
import sys
import os
import locale
import time
from jinja2 import Environment, FileSystemLoader

CMD_PREFIX = 'cmd : '
VERBOSITY = 'info'
_logger = logging.getLogger(__name__)

def run_task(cmd, logger=_logger, msg=None, check_error=False):
    """Run task, report errors and log overall status.

    Run given task using ``subprocess.Popen``. Log the commands
    used and any errors generated. Prints stdout to screen if
    in verbose mode and returns it regardless. Prints stderr to
    screen always.

    :param cmd: Exact command to be executed
    :param logger: Logger to write details to
    :param msg: Message to be shown to user
    :param check_error: Throw exception on error

    :returns: (stdout, stderr)
    """
    def handle_error(exception):
        """Handle errors by logging and optionally raising an exception.
        """
        logger.error(
            'Unable to execute %(cmd)s. Exception: %(exception)s',
            {'cmd': ' '.join(cmd), 'exception': exception})
        if check_error:
            raise exception

    try:
        proc = subprocess.Popen(map(os.path.expanduser, cmd),
                                stdout=subprocess.PIPE,
                                stderr=subprocess.PIPE)
        output = proc.communicate()[0].decode("utf-8")
    except OSError as ex:
        handle_error(ex)
    else:
       if proc.returncode:
            ex = subprocess.CalledProcessError(proc.returncode, cmd, stderr)
            handle_error(ex)
    return output

ifaces = []
sout = run_task(['/usr/bin/c_sample'])
if sout:
    for line in sout.split('\n'):
        if 'Path=' in line:
            print(line)
            field = line.split(' ')[-1].split('=')[-1]
            ifaces.append(field)
ifacesdir = {
    'if1' : ifaces[0],
    'if2' : ifaces[1]}

if len(ifaces) == 2:
    file_loader = FileSystemLoader('./')
    env = Environment(loader = file_loader)
    fileref = env.get_template('./trex_cfg.yaml.j2')
    renderedcon = fileref.render(data=ifacesdir)
    with open('/etc/trex_cfg.yaml', "w+") as fh:
        fh.write(renderedcon)