cgput.py

#!/usr/bin/env python2
# -*- coding: utf-8 -*-
from __future__ import print_function

import argparse
parser = argparse.ArgumentParser(
	description='Put given tgids into a control group.')
parser.add_argument('cgroup', help='Cgroup name.')
parser.add_argument('tgids', nargs='+',
	help='Thread group (process) ids or thread ids.')
optz = parser.parse_args()

import itertools as it, operator as op, functools as ft
from glob import iglob
import os, sys

dst_set = set(it.imap( os.path.realpath,
	iglob('/sys/fs/cgroup/*/{}/cgroup.procs'.format(optz.cgroup)) ))
if not dst_set:
	parser.error('Cgroup not found in any hierarchy: {}'.format(optz.cgroup))

for dst in dst_set:
	for tgid in optz.tgids:
		open(dst, 'wb').write('{}\n'.format(tgid))