List up the threads of a specific process and their CPU affinity

list-thread-cpu-affinity.py

#!/usr/bin/python3

import os
import sys

for pid in sys.argv[1:]:
    path = '/proc/' + pid + '/task'
    if os.path.exists(path):
        for tid in sorted(os.listdir(path)):
            with open(f"{path}/{tid}/comm", 'r', encoding='utf-8') as r:
                thread_name = r.read().strip()
            print(f"TID={tid} Name={thread_name} Affinity={os.sched_getaffinity(int(tid))}")
    else:
        print(f'PID: {pid} not found', file=sys.stderr)