appbackup.py

'''
  Automates the backing up of customer containers
'''
import salt.client

from etcd import Etcd

class AppBackup(object):
  def __init__(self, manager, logger):
    self.etcd = Etcd(logger)
    self.logger = logger
    self.manager = manager
    self.salt_client = salt.client.LocalClient()

  # Backup this users formation to /mnt/ceph/docker_customer_backups
  def backup_formation(self, user, formation):
    self.logger.info('Saving the formation {formation}'.format(
      formation=formation))
    formation = self.manager.load_formation_from_etcd(user, formation)
    for app in formation.application_list:
      self.logger.info('Running commit on {hostname}'.format(
        hostname=app.hostname))
      # Try to commmit the container and wait 10 mins for this to return
      results = self.salt_client.cmd(app.host_server, 'cmd.run', 
        ['docker commit {container_id}'.format(container_id=app.container_id)],
        expr_form='list', timeout=600)
      if results:
        if "Error: No such container" in results[app.host_server]:
          self.logger.error('Could not find container')
        else:
          commit_id = results[app.host_server]
          self.logger.info('Running save on {hostname}'.format(
            hostname=app.hostname))
          self.salt_client.cmd(app.host_server, 'cmd.run', 
            ['docker save {image_id} > /mnt/ceph/docker_customer_backups/'
              '{hostname}.tar'.format(
                image_id=commit_id[0:12], hostname=app.hostname)], 
            expr_form='list', timeout=600)
          self.logger.info('Cleaning up the commit image')
          self.salt_client.cmd(app.host_server, 'cmd.run',
            ['docker rmi {image_id}'.format(image_id=commit_id[0:12])], 
            expr_form='list')
          self.logger.info('Done saving app')