def register_device(self): if not self.device_token: ip, port = self.__resolve_stm_config() transport = TSocket.TSocket(ip, port) # Make socket try: transport = TTransport.TBufferedTransport( transport ) # Buffering is critical. Raw sockets are very slow protocol = TBinaryProtocol.TBinaryProtocol( transport) # Wrap in a protocol client = ShortMemoryService.Client( protocol) # Create a client to use the protocol encoder transport.open() # Connect! inputDevice = DeviceTokenInput() inputDevice.ip = '127.0.0.1' inputDevice.devicetype = 'Development' self.device_token = client.generateDeviceToken(inputDevice) transport.close() except Thrift.TException as tx: print('%s' % (tx.message)) except Exception as ex: print('whot??? %s' % ex) finally: transport.close() return self.device_token
def confirmDevice(self, input): ip, port = self.__resolve_ShortMemory_config() transport = TSocket.TSocket(ip, port) try: transport = TTransport.TBufferedTransport(transport) protocol = TBinaryProtocol.TBinaryProtocol(transport) client = ShortMemoryService.Client(protocol) transport.open() output = client.confirmDevice(input, True) print(output) transport.close() return output except Thrift.TException as tx: print("%s" % (tx.message)) finally: transport.close()
def __get_token_from_service(self, inputToken): ip, port = self.__resolve_config() transport = TSocket.TSocket(ip, port) # Make socket try: transport = TTransport.TBufferedTransport(transport) # Buffering is critical. Raw sockets are very slow protocol = TBinaryProtocol.TBinaryProtocol(transport) # Wrap in a protocol client = ShortMemoryService.Client(protocol) # Create a client to use the protocol encoder transport.open() # Connect! output = client.generateToken(token=inputToken) transport.close() return output except Thrift.TException as tx: print('%s' % (tx.message)) except Exception as ex: print('whot??? %s' % ex) finally: transport.close()
def confirmDevice(self, devicetoken, active): print('short term memory handler') ip, port = self.__resolve_config() transport = TSocket.TSocket(ip, port) # Make socket try: transport = TTransport.TBufferedTransport(transport) # Buffering is critical. Raw sockets are very slow protocol = TBinaryProtocol.TBinaryProtocol(transport) # Wrap in a protocol client = ShortMemoryService.Client(protocol) # Create a client to use the protocol encoder transport.open() client.confirmDevice(deviceToken=devicetoken, enabled=active) transport.close() except Thrift.TException as tx: print('%s' % (tx.message)) except Exception as ex: print('whot??? %s' % ex) finally: transport.close()
def getDeviceList(self): print('short term memory handler') ip, port = self.__resolve_config() transport = TSocket.TSocket(ip, port) # Make socket try: transport = TTransport.TBufferedTransport(transport) # Buffering is critical. Raw sockets are very slow protocol = TBinaryProtocol.TBinaryProtocol(transport) # Wrap in a protocol client = ShortMemoryService.Client(protocol) # Create a client to use the protocol encoder transport.open() output = client.getDeviceList() transport.close() return output except Thrift.TException as tx: print('%s' % (tx.message)) except Exception as ex: print('whot??? %s' % ex) finally: transport.close()
def __write_log(self, input): print('short term memory handler, write log') ip, port = self.__resolve_config() transport = TSocket.TSocket(ip, port) # Make socket try: transport = TTransport.TBufferedTransport( transport) # Buffering is critical. Raw sockets are very slow protocol = TBinaryProtocol.TBinaryProtocol( transport) # Wrap in a protocol client = ShortMemoryService.Client( protocol) # Create a client to use the protocol encoder transport.open() # Connect! client.writeLog(input) transport.close() except Thrift.TException as tx: print('%s' % (tx.message)) except Exception as ex: print('whot??? %s' % ex) finally: transport.close()
def create_server(): handler = ShortTermMemoryThriftServer() return TServer.TSimpleServer(ShortMemoryService.Processor(handler), TSocket.TServerSocket(port=port), TTransport.TBufferedTransportFactory(), TBinaryProtocol.TBinaryProtocolFactory())