-
Notifications
You must be signed in to change notification settings - Fork 0
/
py_spdk.py
66 lines (58 loc) · 1.99 KB
/
py_spdk.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
import psutil
import re
import os
import subprocess
class pyspdk(object):
def __init__(self, pname):
super(pyspdk, self).__init__()
self.pid = None
self.pname = pname
def init_hugepages(self, spdk_dir):
huge_dir = os.path.join(spdk_dir, 'scripts/')
file_dir = self._search_file(huge_dir, 'setup.sh')
print file_dir
os.chdir(file_dir)
p = subprocess.Popen(
'sudo ./setup.sh',
shell=True, stdout=subprocess.PIPE,
stderr=subprocess.PIPE)
out, err = p.communicate()
return out
def _search_file(self, spdk_dir, file_name):
for dirpath, dirnames, filenames in os.walk(spdk_dir):
for filename in filenames:
if filename == file_name:
return dirpath
def _get_process_id(self, pname):
for proc in psutil.process_iter():
try:
pinfo = proc.as_dict(attrs=['pid', 'cmdline'])
if re.search(self.pname, str(pinfo.get('cmdline'))):
self.pid = pinfo.get('pid')
return self.pid
except psutil.NoSuchProcess:
print "NoSuchProcess:%s" % self.pname
print "NoSuchProcess:%s" % self.pname
return self.pid
def is_alive(self):
self.pid = self._get_process_id(self.pname)
if self.pid:
p = psutil.Process(self.pid)
if p.is_running():
return True
return False
def exec_rpc(self, method, server='127.0.0.1', port=5260, sub_args=None):
exec_cmd = ["./rpc.py", "-s", "-p"]
exec_cmd.insert(2, server)
exec_cmd.insert(4, str(port))
exec_cmd.insert(5, method)
if sub_args is None:
sub_args = []
exec_cmd.extend(sub_args)
p = subprocess.Popen(
exec_cmd,
stdout=subprocess.PIPE,
stderr=subprocess.PIPE
)
out, err = p.communicate()
return out