rrs/tools: add -p/--plan option

Add an option to specify which maintenance plan to operate on (largely
for debugging purposes).

Signed-off-by: Paul Eggleton <paul.eggleton@linux.intel.com>

rrs/tools/rrs_distros.py

@@ -92,9 +92,19 @@ if __name__=="__main__":
             help = "Do not write any data back to the database",
             action="store_true", dest="dry_run", default=False)
 
+    parser.add_option("-p", "--plan",
+            help="Specify maintenance plan to operate on (default is all plans that have updates enabled)",
+            action="store", dest="plan", default=None)
+
     options, args = parser.parse_args(sys.argv)
     logger.setLevel(options.loglevel)
 
+    if options.plan:
+        maintplans = MaintenancePlan.objects.filter(id=int(options.plan))
+        if not maintplans.exists():
+            logger.error('No maintenance plan with ID %s found' % options.plan)
+            sys.exit(1)
+    else:
         maintplans = MaintenancePlan.objects.filter(updates_enabled=True)
         if not maintplans.exists():
             logger.error('No enabled maintenance plans found')

rrs/tools/rrs_maintainer_history.py

@@ -72,6 +72,12 @@ def get_commit_info(info, logger):
 """
 def maintainer_history(options, logger):
     fetchdir = settings.LAYER_FETCH_DIR
+    if options.plan:
+        maintplans = MaintenancePlan.objects.filter(id=int(options.plan))
+        if not maintplans.exists():
+            logger.error('No maintenance plan with ID %s found' % options.plan)
+            sys.exit(1)
+    else:
         maintplans = MaintenancePlan.objects.filter(updates_enabled=True)
         if not maintplans.exists():
             logger.error('No enabled maintenance plans found')
@@ -191,6 +197,10 @@ def maintainer_history(options, logger):
 if __name__=="__main__":
     parser = optparse.OptionParser(usage = """%prog [options]""")
 
+    parser.add_option("-p", "--plan",
+            help="Specify maintenance plan to operate on (default is all plans that have updates enabled)",
+            action="store", dest="plan", default=None)
+
     parser.add_option("--fullreload",
             help="Reload upgrade data from scratch",
             action="store_true", dest="fullreload", default=False)

rrs/tools/rrs_upgrade_history.py

@@ -89,6 +89,12 @@ def run_internal(maintplanlayerbranch, commit, commitdate, options, logger, bitb
 def upgrade_history(options, logger):
     from rrs.models import MaintenancePlan, RecipeUpgrade
 
+    if options.plan:
+        maintplans = MaintenancePlan.objects.filter(id=int(options.plan))
+        if not maintplans.exists():
+            logger.error('No maintenance plan with ID %s found' % options.plan)
+            sys.exit(1)
+    else:
         maintplans = MaintenancePlan.objects.filter(updates_enabled=True)
         if not maintplans.exists():
             logger.error('No enabled maintenance plans found')
@@ -205,6 +211,10 @@ if __name__=="__main__":
             help="Reload upgrade data from scratch",
             action="store_true", dest="fullreload", default=False)
 
+    parser.add_option("-p", "--plan",
+            help="Specify maintenance plan to operate on (default is all plans that have updates enabled)",
+            action="store", dest="plan", default=None)
+
     options, args = parser.parse_args(sys.argv)
     logger.setLevel(options.loglevel)
 

rrs/tools/rrs_upstream_email.py

@@ -123,6 +123,10 @@ def main():
         usage = """
     %prog [options]""")
 
+    parser.add_option("-p", "--plan",
+            help="Specify maintenance plan to operate on (default is all plans that have updates enabled)",
+            action="store", dest="plan", default=None)
+
     parser.add_option("-s", "--subject",
             action="store", dest="subject", help='Override email subject')
     parser.add_option("-f", "--from",
@@ -139,6 +143,12 @@ def main():
     options, args = parser.parse_args(sys.argv)
 
     # get recipes for send email
+    if options.plan:
+        maintplans = MaintenancePlan.objects.filter(id=int(options.plan))
+        if not maintplans.exists():
+            logger.error('No maintenance plan with ID %s found' % options.plan)
+            sys.exit(1)
+    else:
         maintplans = MaintenancePlan.objects.filter(email_enabled=True)
         if not maintplans.exists():
             logger.error('No maintenance plans with email enabled were found')

rrs/tools/rrs_upstream_history.py

@@ -135,6 +135,10 @@ def get_upstream_info(layerbranch, recipe_data, result):
 if __name__=="__main__":
     parser = optparse.OptionParser(usage = """%prog [options]""")
 
+    parser.add_option("-p", "--plan",
+            help="Specify maintenance plan to operate on (default is all plans that have updates enabled)",
+            action="store", dest="plan", default=None)
+
     parser.add_option("-d", "--debug",
             help = "Enable debug output",
             action="store_const", const=logging.DEBUG, dest="loglevel", default=logging.INFO)
@@ -146,13 +150,19 @@ if __name__=="__main__":
     options, args = parser.parse_args(sys.argv)
     logger.setLevel(options.loglevel)
 
-    logger.debug("Starting upstream history...")
+    if options.plan:
-
+        maintplans = MaintenancePlan.objects.filter(id=int(options.plan))
+        if not maintplans.exists():
+            logger.error('No maintenance plan with ID %s found' % options.plan)
+            sys.exit(1)
+    else:
         maintplans = MaintenancePlan.objects.filter(updates_enabled=True)
         if not maintplans.exists():
             logger.error('No enabled maintenance plans found')
             sys.exit(1)
 
+    logger.debug("Starting upstream history...")
+
     lockfn = os.path.join(fetchdir, "layerindex.lock")
     lockfile = utils.lock_file(lockfn)
     if not lockfile: