68 lines
2.2 KiB
Python
68 lines
2.2 KiB
Python
"""
|
|
Copyright 2013 Rackspace, Inc.
|
|
|
|
Licensed under the Apache License, Version 2.0 (the "License");
|
|
you may not use this file except in compliance with the License.
|
|
You may obtain a copy of the License at
|
|
|
|
http://www.apache.org/licenses/LICENSE-2.0
|
|
|
|
Unless required by applicable law or agreed to in writing, software
|
|
distributed under the License is distributed on an "AS IS" BASIS,
|
|
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
See the License for the specific language governing permissions and
|
|
limitations under the License.
|
|
"""
|
|
|
|
import simplejson as json
|
|
from twisted.internet.protocol import ReconnectingClientFactory
|
|
from twisted.internet import reactor
|
|
from twisted.python import log
|
|
|
|
from teeth_agent.protocol import TeethAgentProtocol
|
|
|
|
AGENT_VERSION = '0.1-dev'
|
|
|
|
|
|
class AgentClientHandler(TeethAgentProtocol):
|
|
def __init__(self):
|
|
TeethAgentProtocol.__init__(self, json.JSONEncoder())
|
|
self.handlers['v1'] = {
|
|
'prepare_image': self.prepare_image,
|
|
}
|
|
|
|
def connectionMade(self):
|
|
def _response(result):
|
|
log.msg(format='Handshake successful, connection ID is %(connection_id)s', connection_id=result['id'])
|
|
|
|
self.send_command('handshake', 'a:b:c:d', AGENT_VERSION).addCallback(_response)
|
|
|
|
def prepare_image(self, image_id):
|
|
log.msg(format='Preparing image %(image_id)s', image_id=image_id)
|
|
return {'image_id': image_id, 'status': 'PREPARED'}
|
|
|
|
|
|
class AgentClientFactory(ReconnectingClientFactory):
|
|
protocol = AgentClientHandler
|
|
initialDelay = 1.0
|
|
maxDelay = 120
|
|
|
|
def buildProtocol(self, addr):
|
|
self.resetDelay()
|
|
return self.protocol()
|
|
|
|
def clientConnectionFailed(self, connector, reason):
|
|
log.err('Failed to connect, re-trying', delay=self.delay)
|
|
ReconnectingClientFactory.clientConnectionFailed(self, connector, reason)
|
|
|
|
def clientConnectionLost(self, connector, reason):
|
|
log.err('Lost connection, re-connecting', delay=self.delay)
|
|
ReconnectingClientFactory.clientConnectionLost(self, connector, reason)
|
|
|
|
|
|
class TeethAgent(object):
|
|
client_factory = AgentClientFactory()
|
|
|
|
def start(self, host, port):
|
|
reactor.connectTCP(host, port, self.client_factory)
|