wip

split-services-and-network.py

@@ -0,0 +1,36 @@
+#!/usr/bin/env python
+
+import yaml
+
+input_file, output_file = sys.argv[1:3]
+
+stack = yaml.load(input_file)
+
+# The ambassadors need to know the service port to use.
+# Those ports must be declared here.
+ports = dict(
+    redis=6379,
+    rng=80,
+    hasher=80,
+    )
+
+# Links are stored as {from: [to, to, ...]}.
+links = {}
+
+# First, collect all links.
+for service_name, service in stack.itemize():
+    if "links" in service:
+        for link in service["links"]:
+            if link not in ports:
+                print("Skipping link {} in service {} "
+                      "(no port mapping defined). "
+                      "Your code will probably break."
+                      .format(link, service_name))
+                continue
+            if service_name not in links:
+                links[service_name] = []
+            links[service_name].append(link)
+        del service["links"]
+
+yaml.safe_dump(stack, open(output_file, "w"))
+