Builder object, transform config data to docker arguments

paunch/builder/compose1.py

@@ -0,0 +1,123 @@
+#   Licensed under the Apache License, Version 2.0 (the "License"); you may
+#   not use this file except in compliance with the License. You may obtain
+#   a copy of the License at
+#
+#        http://www.apache.org/licenses/LICENSE-2.0
+#
+#   Unless required by applicable law or agreed to in writing, software
+#   distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
+#   WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
+#   License for the specific language governing permissions and limitations
+#   under the License.
+#
+
+import logging
+
+from paunch import runner
+
+LOG = logging.getLogger(__name__)
+
+
+class ComposeV1Builder(object):
+
+    def __init__(self, config_id, config, managed_by, labels=None):
+        self.config_id = config_id
+        self.config = config
+        self.labels = labels
+        self.managed_by = managed_by
+
+    def apply(self):
+
+        stdout = []
+        stderr = []
+        deploy_status_code = 0
+        key_fltr = lambda k: self.config[k].get('start_order', 0)
+        for container in sorted(self.config, key=key_fltr):
+            LOG.debug("Running container: %s" % container)
+            action = self.config[container].get('action', 'run')
+            exit_codes = self.config[container].get('exit_codes', [0])
+
+            if action == 'run':
+                cmd = [
+                    runner.DOCKER_CMD,
+                    'run',
+                    '--name',
+                    runner.unique_container_name(container)
+                ]
+                self.label_arguments(cmd)
+                self.docker_run_args(cmd, container)
+            elif action == 'exec':
+                cmd = [runner.DOCKER_CMD, 'exec']
+                self.docker_exec_args(cmd, container)
+
+            (cmd_stdout, cmd_stderr, returncode) = runner.execute(cmd)
+            if cmd_stdout:
+                stdout.append(cmd_stdout)
+            if cmd_stderr:
+                stderr.append(cmd_stderr)
+
+            if returncode not in exit_codes:
+                LOG.error("Error running %s. [%s]\n" % (cmd, returncode))
+                deploy_status_code = returncode
+            else:
+                LOG.debug('Completed %s' % cmd)
+        return stdout, stderr, deploy_status_code
+
+    def label_arguments(self, cmd, container):
+        if self.labels:
+            for i, v in self.labels.items():
+                cmd.extend(['--label', '%s=%s' % (i, v)])
+        cmd.extend([
+            '--label',
+            'config_id=%s' % self.config_id,
+            '--label',
+            'container_name=%s' % container,
+            '--label',
+            'managed_by=%s' % self.managed_by
+        ])
+
+    def docker_run_args(self, cmd, container):
+        cconfig = self.config[container]
+        if cconfig.get('detach', True):
+            cmd.append('--detach=true')
+        for v in cconfig.get('environment', []):
+            if v:
+                cmd.append('--env=%s' % v)
+        if 'net' in cconfig:
+            cmd.append('--net=%s' % cconfig['net'])
+        if 'pid' in cconfig:
+            cmd.append('--pid=%s' % cconfig['pid'])
+        if 'privileged' in cconfig:
+            cmd.append('--privileged=%s' % str(cconfig['privileged']).lower())
+        if 'restart' in cconfig:
+            cmd.append('--restart=%s' % cconfig['restart'])
+        if 'user' in cconfig:
+            cmd.append('--user=%s' % cconfig['user'])
+        for v in cconfig.get('volumes', []):
+            if v:
+                cmd.append('--volume=%s' % v)
+        for v in cconfig.get('volumes_from', []):
+            if v:
+                cmd.append('--volumes_from=%s' % v)
+
+        cmd.append(cconfig.get('image', ''))
+        cmd.extend(self.command_argument(cmd, cconfig.get('command')))
+
+    def docker_exec_args(self, cmd, container):
+        cconfig = self.config[container]
+        if 'privileged' in cconfig:
+            cmd.append('--privileged=%s' % str(cconfig['privileged']).lower())
+        if 'user' in cconfig:
+            cmd.append('--user=%s' % cconfig['user'])
+        command = self.command_argument(cmd, cconfig.get('command'))
+        # for exec, the first argument is the container name,
+        # make sure the correct one is used
+        command[0] = runner.discover_container_name(command[0], self.config_id)
+        cmd.extend(command)
+
+    def command_argument(self, cmd, command):
+        if not command:
+            return []
+        if not isinstance(command, list):
+            return [command]
+        return command