34 lines
867 B
Python
34 lines
867 B
Python
import subprocess
|
|
import json
|
|
import os
|
|
|
|
|
|
class Alfred(object):
|
|
"""
|
|
Bindings for the alfred-json utility
|
|
"""
|
|
def __init__(self, unix_sockpath=None):
|
|
self.unix_sock = unix_sockpath
|
|
if unix_sockpath is not None and not os.path.exists(unix_sockpath):
|
|
raise RuntimeError('alfred: invalid unix socket path given')
|
|
|
|
def _fetch(self, data_type):
|
|
cmd = ['/usr/local/bin/alfred-json',
|
|
'-z',
|
|
'-f', 'json',
|
|
'-r', str(data_type)]
|
|
if self.unix_sock:
|
|
cmd.extend(['-s', self.unix_sock])
|
|
|
|
output = subprocess.check_output(cmd)
|
|
return json.loads(output.decode("utf-8")).values()
|
|
|
|
def nodeinfo(self):
|
|
return self._fetch(158)
|
|
|
|
def statistics(self):
|
|
return self._fetch(159)
|
|
|
|
def vis(self):
|
|
return self._fetch(160)
|