File: args.py

package info (click to toggle)
siridb-server 2.0.53-2
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid
  • size: 4,612 kB
  • sloc: ansic: 47,501; python: 6,263; sh: 254; makefile: 149
file content (89 lines) | stat: -rw-r--r-- 2,527 bytes parent folder | download | duplicates (4)
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
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
import os
import subprocess
import argparse
from .server import Server
from .color import Color
from .constants import SIRIDBC


def is_valgrind_installed():
    with open(os.devnull, 'w') as fnull:
        try:
            subprocess.call(['valgrind'], stdout=fnull, stderr=fnull)
        except OSError as e:
            if e.errno == os.errno.ENOENT:
                return False
    return True


def print_siridb_version(args):
    fn = SIRIDBC.format(BUILDTYPE=args.build)
    try:
        p = subprocess.Popen(
            [fn, '--version'],
            stdout=subprocess.PIPE,
            stderr=subprocess.PIPE)
    except FileNotFoundError:
        print(Color.error(f'Cannot find: {fn}'))
        exit(1)

    output, err = p.communicate()
    rc = p.returncode
    if rc or err:
        print(Color.error(f'Cannot use: {fn}'))
        exit(1)

    print(f'Test with: {Color.info(output.decode().splitlines()[0])}')


def parse_args():
    parser = argparse.ArgumentParser()

    parser.add_argument(
        '-t', '--terminal',
        choices=['xterm', 'xfce4-terminal'],
        default=None,
        help='Start SiriDB servers in a terminal. If no terminal is given '
        'process put their output in log files.')

    parser.add_argument(
        '-m', '--mem-check',
        action='store_true',
        help='Use `valgrind` for memory errors and leaks.')

    parser.add_argument(
        '-k', '--keep',
        action='store_true',
        help='Only valid when a terminal is used. This will keep the terminal '
        'open.')

    parser.add_argument(
        '-b', '--build',
        choices=['Release', 'Debug'],
        default='Release',
        help='Choose either the Release or Debug build.')

    parser.add_argument(
        '-l', '--log-level',
        default='critical',
        help='set the log level',
        choices=['debug', 'info', 'warning', 'error', 'critical'])

    args = parser.parse_args()

    has_valgrind = is_valgrind_installed()

    print_siridb_version(args)
    print("Test using valgrind for memory errors and leaks: ", end='')
    if args.mem_check and not has_valgrind:
        print(Color.warning('disabled (!! valgrind not found !!)'))
    elif not args.mem_check:
        print(Color.warning('disabled'))
    else:
        print(Color.success('enabled'))

    Server.MEM_CHECK = args.mem_check and has_valgrind
    Server.HOLD_TERM = args.keep
    Server.TERMINAL = args.terminal
    Server.BUILDTYPE = args.build
    Server.LOG_LEVEL = args.log_level.upper()