-
Notifications
You must be signed in to change notification settings - Fork 0
/
process_log.py
66 lines (49 loc) · 1.73 KB
/
process_log.py
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
'''Process logger'''
from os.path import exists
from time import sleep
from datetime import datetime
from sqlite3 import connect
from psutil import process_iter
def get_all_stats():
'''Stats snapshot'''
values = {}
for pro in process_iter():
if "DockerTest/main.py" in pro.cmdline()[-1]:
cpu_percent = pro.cpu_percent()
memory_info = pro.memory_info_ex()
values[pro.pid] = {'name': pro.name(),
'cmd': ' '.join(pro.cmdline()),
'cpu': cpu_percent,
'rss': memory_info.rss / 1024 ** 2,
'vms': memory_info.vms / 1024 ** 2}
return values
def main():
'''Main function'''
# Initial call (the psutil functions need to be called once before averages
# work)
get_all_stats()
new = not exists('process-log.db')
conn = connect('process-log.db')
cur = conn.cursor()
if new:
cur.execute('CREATE TABLE history (date datetime, pid int, name text,'
' cmd text, cpu real, rss real, vms real)')
try:
while True:
sleep(5)
now = datetime.now().isoformat()
values = get_all_stats()
values_to_insert = []
for pid in values:
dic = values[pid]
values_to_insert.append((now, pid, dic['name'], dic['cmd'],
dic['cpu'], dic['rss'], dic['vms']))
cur.executemany('INSERT INTO history VALUES(?,?,?,?,?,?,?)',
values_to_insert)
conn.commit()
except KeyboardInterrupt:
pass
finally:
conn.commit()
conn.close()
main()