mirror of
https://github.com/Red5d/docker-autocompose
synced 2026-01-03 15:18:01 +00:00
Fix setup file and reformat autocompose to fit setup requirements.
This commit is contained in:
Red5d
162
autocompose.py
162
autocompose.py
@@ -3,95 +3,103 @@
|
||||
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()
|
||||
def main():
|
||||
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')
|
||||
generate(args)
|
||||
|
||||
|
||||
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)
|
||||
def generate(args):
|
||||
c = Client(base_url='unix://var/run/docker.sock')
|
||||
|
||||
cinspect = c.inspect_container(cid)
|
||||
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
|
||||
# Build yaml dict structure
|
||||
|
||||
cfile = {}
|
||||
cfile[args.cname] = {}
|
||||
ct = cfile[args.cname]
|
||||
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']
|
||||
}
|
||||
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 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']]
|
||||
# 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:]
|
||||
# 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
|
||||
values['ports'] = ports_value
|
||||
else:
|
||||
values['expose'] = expose_value
|
||||
|
||||
except KeyError:
|
||||
# No ports exposed/bound. Continue without them.
|
||||
ports = None
|
||||
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
|
||||
# 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)
|
||||
# Render yaml file
|
||||
pyaml.p(cfile)
|
||||
|
||||
if __name__ == "__main__":
|
||||
main()
|
||||
|
||||
|
||||
4
setup.py
4
setup.py
@@ -3,10 +3,10 @@ setup(
|
||||
name = "docker-autocompose",
|
||||
version = "1.0",
|
||||
packages = find_packages(),
|
||||
install_requires = ['pyaml, docker-py'],
|
||||
install_requires = ['pyaml>=15.8.2', 'docker-py>=1.6.0'],
|
||||
entry_points={
|
||||
'console_scripts': [
|
||||
'autocompose = autocompose',
|
||||
'autocompose = autocompose:main',
|
||||
]
|
||||
}
|
||||
)
|
||||
|
||||
Reference in New Issue
Block a user