From 9ea33801105cc5cfea5f7fd28b1e32f8bb5c9873 Mon Sep 17 00:00:00 2001
From: Red5d <daniel.colson99@gmail.com>
Date: Sat, 2 Jan 2016 17:17:18 -0500
Subject: [PATCH] Added autocompose script.

---
 autocompose.py | 97 ++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 97 insertions(+)
 create mode 100644 autocompose.py

diff --git a/autocompose.py b/autocompose.py
new file mode 100644
index 0000000..297e766
--- /dev/null
+++ b/autocompose.py
@@ -0,0 +1,97 @@
+#! /usr/bin/env python
+
+import pyaml, argparse, sys
+from docker import Client
+
+parser = argparse.ArgumentParser(description='Generate docker-compose yaml definition from running container.')
+parser.add_argument('cname', type=str, help='The name of the container to process.')
+args = parser.parse_args()
+
+c = Client(base_url='unix://var/run/docker.sock')
+
+try:
+    cid = [x['Id'] for x in c.containers() if args.cname in x['Names'][0]][0]
+except IndexError:
+    print("That container is not running.")
+    sys.exit(1)
+
+cinspect = c.inspect_container(cid)
+
+
+# Build yaml dict structure
+
+cfile = {}
+cfile[args.cname] = {}
+ct = cfile[args.cname]
+
+values = {
+    'cap_add': cinspect['HostConfig']['CapAdd'],
+    'cap_drop': cinspect['HostConfig']['CapDrop'],
+    'cgroup_parent': cinspect['HostConfig']['CgroupParent'],
+    'container_name': args.cname,
+    'devices': cinspect['HostConfig']['Devices'],
+    'dns': cinspect['HostConfig']['Dns'],
+    'dns_search': cinspect['HostConfig']['DnsSearch'],
+    'environment': cinspect['Config']['Env'],
+    'extra_hosts': cinspect['HostConfig']['ExtraHosts'],
+    'image': cinspect['Config']['Image'],
+    'labels': cinspect['Config']['Labels'],
+    'links': cinspect['HostConfig']['Links'],
+    'log_driver': cinspect['HostConfig']['LogConfig']['Type'],
+    'log_opt': cinspect['HostConfig']['LogConfig']['Config'],
+    'net': cinspect['HostConfig']['NetworkMode'],
+    'security_opt': cinspect['HostConfig']['SecurityOpt'],
+    'ulimits': cinspect['HostConfig']['Ulimits'],
+    'volumes': cinspect['HostConfig']['Binds'],
+    'volume_driver': cinspect['HostConfig']['VolumeDriver'],
+    'volumes_from': cinspect['HostConfig']['VolumesFrom'],
+    'cpu_shares': cinspect['HostConfig']['CpuShares'],
+    'cpuset': cinspect['HostConfig']['CpusetCpus']+','+cinspect['HostConfig']['CpusetMems'],
+    'entrypoint': cinspect['Config']['Entrypoint'],
+    'user': cinspect['Config']['User'],
+    'working_dir': cinspect['Config']['WorkingDir'],
+    'domainname': cinspect['Config']['Domainname'],
+    'hostname': cinspect['Config']['Hostname'],
+    'ipc': cinspect['HostConfig']['IpcMode'],
+    'mac_address': cinspect['NetworkSettings']['MacAddress'],
+    'mem_limit': cinspect['HostConfig']['Memory'],
+    'memswap_limit': cinspect['HostConfig']['MemorySwap'],
+    'privileged': cinspect['HostConfig']['Privileged'],
+    'restart': cinspect['HostConfig']['RestartPolicy']['Name'],
+    'read_only': cinspect['HostConfig']['ReadonlyRootfs'],
+    'stdin_open': cinspect['Config']['OpenStdin'],
+    'tty': cinspect['Config']['Tty']
+}
+
+# Check for command and add it if present.
+if cinspect['Config']['Cmd'] != None:
+    values['command'] = " ".join(cinspect['Config']['Cmd']),
+
+# Check for exposed/bound ports and add them if needed.
+try:
+    expose_value =  list(cinspect['Config']['ExposedPorts'].keys())
+    ports_value = [cinspect['HostConfig']['PortBindings'][key][0]['HostIp']+':'+cinspect['HostConfig']['PortBindings'][key][0]['HostPort']+':'+key for key in cinspect['HostConfig']['PortBindings']]
+
+    # If bound ports found, don't use the 'expose' value.
+    if (ports_value != None) and (ports_value != "") and (ports_value != []) and (ports_value != 'null') and (ports_value != {}) and (ports_value != "default") and (ports_value != 0) and (ports_value != ",") and (ports_value != "no"):
+        for index, port in enumerate(ports_value):
+            if port[0] == ':':
+                ports_value[index] = port[1:]
+
+        values['ports'] = ports_value
+    else:
+        values['expose'] = expose_value
+
+except KeyError:
+    # No ports exposed/bound. Continue without them.
+    ports = None
+
+# Iterate through values to finish building yaml dict.
+for key in values:
+    value = values[key]
+    if (value != None) and (value != "") and (value != []) and (value != 'null') and (value != {}) and (value != "default") and (value != 0) and (value != ",") and (value != "no"):
+        ct[key] = value
+
+# Render yaml file
+pyaml.p(cfile)
+