publish.py

#!/usr/bin/env python
import re
import subprocess
import json
import sys

def execute_cmd(cmd):
  proc = subprocess.Popen(cmd, shell=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
  proc.wait()
  comm = proc.communicate()

  if comm[1] != '':
    print(comm[1].rstrip('\n'))
    exit(-1)


  return comm[0]

def publish_to_aws():
  ecr_response = execute_cmd('aws ecr describe-repositories').rstrip('\n')
  ecr_json = json.loads(ecr_response)

  print  "\nAWS Reposdiscovered for zone: "

  repos = []
  for uri in ecr_json["repositories"]:
    repos.append( uri["repositoryUri"] )

  for idx , r in enumerate(repos,0):
    print str(idx) + ") -> " + r

  print  "\nSelect a repo to publish to: "
  selection = sys.stdin.readline()
  pos = int(selection)
  repoName = repos[pos]

  print  "\nEnter a docker tag: "
  tagName = sys.stdin.readline()

  tag_image(repoName,tagName)
  push_to_repo(repoName,tagName)

def tag_image(repo, tag):
  tag_cmd = "docker tag www-test:latest {repoUri}:{tag}"
  print  "\nTagging image with: " + tag_cmd.format(repoUri=repo, tag=tag)
  print  "\t ->" + tag_cmd.format(repoUri=repo,tag=tag)
  execute_cmd(tag_cmd.format(repoUri=repo, tag=tag))


def push_to_repo(repo, tag):
  push_cmd = "docker push {repoUri}:{tag}"
  print  "\nPublishing image to: " + repo + " with tag: "  + tag
  print  "\t ->" + push_cmd.format(repoUri=repo,tag=tag)
  execute_cmd(push_cmd.format(repoUri=repo,tag=tag))


if __name__ == "__main__":
  publish_to_aws()