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())