Spaces:
Build error
Build error
File size: 1,135 Bytes
6250360 |
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 |
import multiprocessing
def cpu_count():
return multiprocessing.cpu_count()
def get_pool(processes):
pool = multiprocessing.Pool(processes = processes)
return pool
def wait_for_pool(pool):
pool.close()
pool.join()
def set_proc_name(name):
import setproctitle
setproctitle.setproctitle(name)
def kill(pid):
import util
if type(pid) == list:
for p in pid:
kill(p)
elif type(pid) == int:
cmd = 'kill -9 %d'%(pid)
print cmd
print util.cmd.cmd(cmd)
elif type(pid) == str:
pids = get_pid(pid)
kill(pids)
else:
raise ValueError, 'Not supported parameter type:', type(pid)
def ps_aux_grep(pattern):
import util
cmd = 'ps aux|grep %s'%(pattern)
return util.cmd.cmd(cmd)
def get_pid(pattern):
import util
cmd = 'ps aux|grep %s'%(pattern)
results = util.cmd.cmd(cmd)
results = util.str.split(results, '\n')
pids = []
for result in results:
info = result.split()
if len(info) > 0:
pid = int(info[1])
pids.append(pid)
return pids
|