You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
41 lines
1.3 KiB
41 lines
1.3 KiB
import binascii
|
|
import json
|
|
import os
|
|
import socket
|
|
|
|
|
|
class Daemon:
|
|
def __init__(self, port, cookie_dir):
|
|
self.sock = socket.create_connection(('localhost', port))
|
|
self.fd = self.sock.makefile()
|
|
path = os.path.join(os.path.expanduser(cookie_dir), '.cookie')
|
|
cookie = binascii.b2a_base64(open(path, 'rb').read())
|
|
self.cookie = cookie.decode('ascii').strip()
|
|
self.index = 0
|
|
|
|
def request(self, method, params_list):
|
|
obj = [{"method": method, "params": params, "id": self.index}
|
|
for params in params_list]
|
|
request = json.dumps(obj)
|
|
|
|
msg = ('POST / HTTP/1.1\n'
|
|
'Authorization: Basic {}\n'
|
|
'Content-Length: {}\n\n'
|
|
'{}'.format(self.cookie, len(request), request))
|
|
self.sock.sendall(msg.encode('ascii'))
|
|
|
|
status = self.fd.readline().strip()
|
|
while True:
|
|
if self.fd.readline().strip():
|
|
continue # skip headers
|
|
else:
|
|
break # next line will contain the response
|
|
|
|
data = self.fd.readline().strip()
|
|
replies = json.loads(data)
|
|
for reply in replies:
|
|
assert reply['error'] is None, reply
|
|
assert reply['id'] == self.index
|
|
|
|
self.index += 1
|
|
return [d['result'] for d in replies]
|
|
|