File: pflags

package info (click to toggle)
python-linux-procfs 0.7.3-3
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid, trixie
  • size: 188 kB
  • sloc: python: 773; makefile: 17
file content (86 lines) | stat: -rwxr-xr-x 2,596 bytes parent folder | download
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
#! /usr/bin/python3
# -*- python -*-
# -*- coding: utf-8 -*-
#   print process flags
#   Copyright (C) 2015 Red Hat Inc.
#   Arnaldo Carvalho de Melo <acme@redhat.com>
#
#   This application is free software; you can redistribute it and/or
#   modify it under the terms of the GNU General Public License
#   as published by the Free Software Foundation; version 2.
#
#   This application is distributed in the hope that it will be useful,
#   but WITHOUT ANY WARRANTY; without even the implied warranty of
#   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
#   General Public License for more details.


import procfs, re, fnmatch, sys
import argparse
from functools import reduce

ps = None

def thread_mapper(s):
    global ps

    try:
        return [int(s), ]
    except:
        pass
    try:
        return ps.find_by_regex(re.compile(fnmatch.translate(s)))
    except:
        return ps.find_by_name(s)

def main(argv):

    global ps
    ps = procfs.pidstats()

    parser = argparse.ArgumentParser(description='Print process flags')
    parser.add_argument('pid', nargs='*', help='a list of pids or names')
    args = parser.parse_args()

    if len(argv) > 1:
        pids = args.pid
        pids = reduce(lambda i, j: i + j, list(map(thread_mapper, pids)))
    else:
        pids = list(ps.processes.keys())

    pids.sort()
    len_comms = []
    for pid in pids:
        if pid in ps:
            try:
                len(ps[pid]["stat"]["comm"])
            except (TypeError, FileNotFoundError):
                continue
            len_comms.append(len(ps[pid]["stat"]["comm"]))

    max_comm_len = max(len_comms, default=0)
    del len_comms

    for pid in pids:
        if pid not in ps:
            continue
        try:
            flags = ps[pid].stat.process_flags()
        except AttributeError:
            continue
        # Remove flags that were superseeded
        if "PF_THREAD_BOUND" in flags and "PF_NO_SETAFFINITY" in flags:
            flags.remove("PF_THREAD_BOUND")
        if "PF_FLUSHER" in flags and "PF_NPROC_EXCEEDED" in flags:
            flags.remove("PF_FLUSHER")
        if "PF_SWAPOFF" in flags and "PF_MEMALLOC_NOIO" in flags:
            flags.remove("PF_SWAPOFF")
        if "PF_FREEZER_NOSIG" in flags and "PF_SUSPEND_TASK" in flags:
            flags.remove("PF_FREEZER_NOSIG")
        comm = ps[pid].stat["comm"]
        flags.sort()
        sflags = reduce(lambda i, j: "%s|%s" % (i, j), [a[3:] for a in flags])
        print("%6d %*s %s" %(pid, max_comm_len, comm, sflags))

if __name__ == '__main__':
    main(sys.argv)