Merge lp:~stgraber/ubuntu-archive-tools/amis-add-argparse into lp:ubuntu-archive-tools

[Stéphane Graber]

In preparation for the cloud images being posted to the tracker directly from
their nusakan-equivalent machine, I'm adding some argument handling to the
script, allowing the milestone to be overriden from the command line.

[Colin Watson (cjwatson)]

review: Approve

=== modified file 'post-amis-to-iso-tracker'
--- post-amis-to-iso-tracker	2012-09-05 15:32:27 +0000
+++ post-amis-to-iso-tracker	2012-09-26 16:29:22 +0000
@@ -16,6 +16,8 @@
 
 from __future__ import print_function
 
+import argparse
+import os
 import sys
 
 # import isotracker
@@ -83,9 +85,22 @@
 
 
 def main():
-    with open(sys.argv[1], 'r') as handle:
-        milestone = isotracker.default_milestone()
-
+    parser = argparse.ArgumentParser(
+              description="Publish a provided list of AMIs to the QA tracker.")
+    parser.add_argument("--milestone", type=isotracker.milestone_code,
+                        help="A milestone name to use instead of the default",
+                        default=isotracker.default_milestone())
+    parser.add_argument("input_file", type=str,
+                        help="An input file (published-ec2-daily.txt)")
+    args = parser.parse_args()
+
+    print("Publishing to milestone: %s" % args.milestone)
+
+    if not os.path.exists(args.input_file):
+        print("Can't find input file: %s" % args.input_file)
+        sys.exit(1)
+
+    with open(args.input_file, 'r') as handle:
         for line in handle:
             zone, ami, arch, store = line.split()[0:4]
             if not ami.startswith('ami-'):
@@ -99,7 +114,7 @@
                 continue
 
             try:
-                isotracker.post_build(product, ami, milestone)
+                isotracker.post_build(product, ami, args.milestone)
             except KeyError as e:
                 print(e)
                 continue