def on_shield_found(self, ip_address):
    self.ip_address = ip_address
    self.connect()
    atexit.register(self.disconnect)
def on_shield_found(self, ip_address):
    self.ip_address = ip_address
    self.connect()
    atexit.register(self.disconnect)