import socket

class QMemmanClient:

    def request_memory(self, amount):
        self.sock = socket.socket(socket.AF_UNIX)
        self.sock.connect("/var/run/qubes/qmemman.sock")
        self.sock.send(str(amount)+"\n")
        self.received = self.sock.recv(1024).strip()
        if self.received == 'OK':
            return True
        else:
            return False

    def close(self):
        self.sock.close()