network.py 1.8 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758
  1. from __future__ import absolute_import
  2. from __future__ import unicode_literals
  3. import logging
  4. from docker.errors import NotFound
  5. from .config import ConfigurationError
  6. log = logging.getLogger(__name__)
  7. # TODO: support external networks
  8. class Network(object):
  9. def __init__(self, client, project, name, driver=None, driver_opts=None):
  10. self.client = client
  11. self.project = project
  12. self.name = name
  13. self.driver = driver
  14. self.driver_opts = driver_opts
  15. def ensure(self):
  16. try:
  17. data = self.inspect()
  18. if self.driver and data['Driver'] != self.driver:
  19. raise ConfigurationError(
  20. 'Network {} needs to be recreated - driver has changed'
  21. .format(self.full_name))
  22. if data['Options'] != (self.driver_opts or {}):
  23. raise ConfigurationError(
  24. 'Network {} needs to be recreated - options have changed'
  25. .format(self.full_name))
  26. except NotFound:
  27. driver_name = 'the default driver'
  28. if self.driver:
  29. driver_name = 'driver "{}"'.format(self.driver)
  30. log.info(
  31. 'Creating network "{}" with {}'
  32. .format(self.full_name, driver_name)
  33. )
  34. self.client.create_network(
  35. self.full_name, self.driver, self.driver_opts
  36. )
  37. def remove(self):
  38. # TODO: don't remove external networks
  39. log.info("Removing network {}".format(self.full_name))
  40. self.client.remove_network(self.full_name)
  41. def inspect(self):
  42. return self.client.inspect_network(self.full_name)
  43. @property
  44. def full_name(self):
  45. return '{0}_{1}'.format(self.project, self.name)