yay

plum/__init__.py

@@ -0,0 +1,3 @@
+from .service import Service
+
+__version__ = '1.0.0'

plum/service.py

@@ -0,0 +1,30 @@
+class Service(object):
+    def __init__(self, client, image, command):
+        self.client = client
+        self.image = image
+        self.command = command
+
+    @property
+    def containers(self):
+        return self.client.containers()
+
+    def start(self):
+        if len(self.containers) == 0:
+            self.start_container()
+
+    def stop(self):
+        self.scale(0)
+
+    def scale(self, num):
+        while len(self.containers) < num:
+            self.start_container()
+
+        while len(self.containers) > num:
+            self.stop_container()
+
+    def start_container(self):
+        container = self.client.create_container(self.image, self.command)
+        self.client.start(container['Id'])
+
+    def stop_container(self):
+        self.client.kill(self.containers[0]['Id'])

plum/tests/service_test.py

@@ -0,0 +1,39 @@
+from unittest import TestCase
+from docker import Client
+from plum import Service
+
+
+class ServiceTestCase(TestCase):
+    def setUp(self):
+        self.client = Client('http://127.0.0.1:4243')
+        self.client.pull('ubuntu')
+
+        for c in self.client.containers():
+            self.client.kill(c['Id'])
+
+        self.service = Service(
+            client=self.client,
+            image="ubuntu",
+            command=["/bin/sleep", "300"],
+        )
+
+    def test_up_scale_down(self):
+        self.assertEqual(len(self.service.containers), 0)
+
+        self.service.start()
+        self.assertEqual(len(self.service.containers), 1)
+
+        self.service.start()
+        self.assertEqual(len(self.service.containers), 1)
+
+        self.service.scale(2)
+        self.assertEqual(len(self.service.containers), 2)
+
+        self.service.scale(1)
+        self.assertEqual(len(self.service.containers), 1)
+
+        self.service.stop()
+        self.assertEqual(len(self.service.containers), 0)
+
+        self.service.stop()
+        self.assertEqual(len(self.service.containers), 0)