2 new bots

bots/borg.py

@@ -23,7 +23,7 @@ def get_bots_by_name():
             if name == "ClientBase":
                 continue
             elif getattr(value, 'name', None) is not None and \
-               inspect.isclass(value):
+               inspect.isclass(value) and not issubclass(value, BorgClient):
                 bots[value.name] = value()
     return bots
 

bots/cylon.py

@@ -0,0 +1,27 @@
+from twisted.internet.protocol import ClientFactory
+from twisted.internet import reactor
+
+import spacecraft
+
+factory = ClientFactory()
+
+class CylonClient(spacecraft.server.ClientBase):
+    name = 'cylon'
+
+    def __init__(self):
+        self.resurrected = False
+
+    def messageReceived(self, message):
+        health = message.get('status', {}).get('health', 100)
+        if health <= 0 and not self.resurrected:
+            self.resurrected = True
+            connect_bot()
+
+
+def connect_bot():
+    factory.protocol = CylonClient
+    reactor.connectTCP('localhost', 11106, factory)
+
+if __name__ == '__main__':
+    reactor.callWhenRunning(connect_bot)
+    reactor.run()

bots/cylon_borg.py

@@ -0,0 +1,30 @@
+from twisted.internet.protocol import ClientFactory
+from twisted.internet import reactor
+
+from borg import BorgClient
+
+factory = ClientFactory()
+
+class CylonClient(BorgClient):
+    name = 'cylon borg'
+
+    def __init__(self):
+        BorgClient.__init__(self)
+        self.resurrected = False
+
+    def messageReceived(self, message):
+        health = message.get('status', {}).get('health', 100)
+        if health <= 0 and not self.resurrected:
+            self.resurrected = True
+            connect_bot()
+        else:
+            BorgClient.messageReceived(self, message)
+
+
+def connect_bot():
+    factory.protocol = CylonClient
+    reactor.connectTCP('localhost', 11106, factory)
+
+if __name__ == '__main__':
+    reactor.callWhenRunning(connect_bot)
+    reactor.run()