pulpito/pulpito/controllers/util.py at db7bed71e02866d2a90e23f1716e4fb6913f258e · ceph/pulpito · GitHub

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
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
from datetime import datetime, timedelta

timestamp_fmt = '%Y-%m-%d %H:%M:%S'

run_status_map = {
    'pass':    'success',
    'fail':    'danger',
    'dead':    'danger',
    'running': 'warning',
    'unknown': 'warning',
    None:      'warning',
    'waiting': 'info',
}


def remove_msecs(timestamp):
    return timestamp.split('.')[0]


def remove_delta_msecs(delta):
    return delta - timedelta(microseconds=getattr(delta, 'microseconds', 0))


def prettify_run(run):
    set_run_status_class(run)
    set_run_time_info(run)
    set_run_wait_time(run)


def set_run_wait_time(run):
    if run.get('jobs'):
        total_wait_time = None
        for job in run['jobs']:
            if job.get("wait_time"):
                if total_wait_time:
                    total_wait_time = total_wait_time + job['wait_time']
                else:
                    total_wait_time = job['wait_time']

        if total_wait_time:
            run["total_wait_time"] = total_wait_time
            jobs_that_waited = [job for job in run['jobs']
                                if job.get("wait_time")]
            run["avg_wait_time"] = remove_delta_msecs(
                total_wait_time / len(jobs_that_waited)
            )


def set_run_status_class(run):
    for key in run_status_map.keys():
        if key and key in run.get('status', 'unknown'):
            status_class = run_status_map[key]
            run['status_class'] = status_class
            break


def set_run_time_info(run):
    if run.get('scheduled'):
        run['scheduled'] = remove_msecs(run['scheduled'])
    if run.get('posted'):
        run['posted'] = remove_msecs(run['posted'])
    if run.get('updated'):
        run['updated'] = remove_msecs(run['updated'])
    if run.get('started'):
        run['started'] = remove_msecs(run['started'])
        started = datetime.strptime(run['started'], timestamp_fmt)
        if run['status'] in ['running', 'waiting']:
            run['runtime'] = remove_msecs(str(datetime.utcnow() - started))
        else:
            run_updated = run.get('updated', run['posted'])
            if not run_updated:
                run['runtime'] = ''
            else:
                updated = datetime.strptime(run_updated, timestamp_fmt)
                run['runtime'] = remove_msecs(str(updated - started))


def prettify_job(job):
    set_job_status_class(job)
    set_job_time_info(job)
    remove_none_strings(job)
    return job


def set_job_status_class(job):
    job['status_class'] = run_status_map.get(job.get('status', 'unknown'))


def set_job_time_info(job):
    if job.get('posted'):
        job['posted'] = remove_msecs(job['posted'])
    if job.get('started'):
        job['started'] = remove_msecs(job['started'])
    if job.get('updated'):
        job['updated'] = remove_msecs(job['updated'])
    if job.get('started') and job.get('updated'):
        started = datetime.strptime(job['started'], timestamp_fmt)
        if job['status'] in ['running', 'waiting']:
            job['runtime'] = remove_delta_msecs(datetime.utcnow() - started)
        else:
            updated = datetime.strptime(job['updated'], timestamp_fmt)
            job['runtime'] = remove_delta_msecs(updated - started)
    if job.get('duration') and job.get('runtime'):
        duration = timedelta(seconds=job['duration'])
        job['duration'] = str(duration)
        wait_time = remove_delta_msecs(job['runtime'] - duration)
        job['wait_time'] = wait_time
    elif job.get("status") == 'waiting':
        job['wait_time'] = job.get('runtime')


def remove_none_strings(obj):
    for key, value in obj.items():
        if str(value) == 'None':
            obj[key] = ''
    return obj


def get_run_filters(**kwargs):
    filters = dict()
    for (key, value) in kwargs.items():
        if value == '':
            continue
        elif key == 'latest':
            continue
        else:
            filters[key] = value
    return filters


def set_node_status_class(node):
    up = node.get('up')
    locked = node.get('locked')
    if up is True and locked is False:
        status_class = 'success'
    elif up is False:
        status_class = 'danger'
    elif locked is True:
        status_class = 'warning'
    else:
        status_class = ''
    node['status_class'] = status_class