blob: 2674e1ffef29c98abd1d583046302a6bb24ea5f2 (
plain)
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
|
#!/usr/bin/python
# Get the CPU usage and print to file.
import psutil
import time
import argparse
import os
import threading
class LogCpuUsage(threading.Thread):
def __init__(self, fname):
threading.Thread.__init__(self)
self._stopped = True
self._fname = fname
def run(self):
self._stopped = False
with open(self._fname, 'w') as f:
start = time.time()
while self._stopped is False:
now = time.time()
f.write("%f %f\n" % ((now - start), psutil.cpu_percent()))
time.sleep(0.01)
def stop(self):
self._stopped = True
if __name__ == '__main__':
parser = argparse.ArgumentParser()
parser.add_argument('file', help='where to save output')
args = parser.parse_args()
if os.path.isfile(args.file):
replace = raw_input('File %s exists, replace it (y/N)? ' % args.file)
if replace.lower() != 'y':
print 'Bailing out.'
exit(1)
log_cpu = LogCpuUsage(args.file)
log_cpu.run()
|
