compose/plum/service_collection.py

55 lines
1.6 KiB
Python

from .service import Service
def sort_service_dicts(services):
# Sort in dependency order
def cmp(x, y):
x_deps_y = y['name'] in x.get('links', [])
y_deps_x = x['name'] in y.get('links', [])
if x_deps_y and not y_deps_x:
return 1
elif y_deps_x and not x_deps_y:
return -1
return 0
return sorted(services, cmp=cmp)
class ServiceCollection(list):
@classmethod
def from_dicts(cls, client, service_dicts):
"""
Construct a ServiceCollection from a list of dicts representing services.
"""
collection = ServiceCollection()
for service_dict in sort_service_dicts(service_dicts):
# Reference links by object
links = []
if 'links' in service_dict:
for name in service_dict.get('links', []):
links.append(collection.get(name))
del service_dict['links']
collection.append(Service(client=client, links=links, **service_dict))
return collection
@classmethod
def from_config(cls, client, config):
dicts = []
for name, service in config.items():
service['name'] = name
dicts.append(service)
return cls.from_dicts(client, dicts)
def get(self, name):
for service in self:
if service.name == name:
return service
def start(self):
for container in self:
container.start()
def stop(self):
for container in self:
container.stop()