rrs/tools/rrs_upstream_history.py: Add recipe upstream script.

This script generates upstream version information and recipe status.

Signed-off-by: Aníbal Limón <anibal.limon@linux.intel.com>

1 files changed, 150 insertions, 0 deletions

diff --git a/rrs/tools/rrs_upstream_history.py b/rrs/tools/rrs_upstream_history.py
new file mode 100755
index 0000000000..1367468fcc
--- /dev/null
+++ b/rrs/tools/rrs_upstream_history.py
@@ -0,0 +1,150 @@
+#!/usr/bin/env python
+
+# Get upstream information for recipes.
+#
+# Copyright (C) 2015 Intel Corporation
+# Author: Anibal Limon <anibal.limon@linux.intel.com>
+#
+# Licensed under the MIT license, see COPYING.MIT for details
+
+import sys
+import os.path
+import optparse
+import logging
+from datetime import datetime
+
+sys.path.insert(0, os.path.realpath(os.path.join(os.path.dirname(__file__))))
+from common import common_setup, update_repo, load_recipes, \
+        get_pv_type, get_logger
+common_setup()
+from layerindex import utils
+
+utils.setup_django()
+from django.db import transaction
+import settings
+
+logger = get_logger("UpstreamHistory", settings)
+fetchdir = settings.LAYER_FETCH_DIR
+if not fetchdir:
+    logger.error("Please set LAYER_FETCH_DIR in settings.py")
+    sys.exit(1)
+
+update_repo(settings.LAYER_FETCH_DIR, 'poky', settings.POKY_REPO_URL,
+                True, logger)
+
+# setup bitbake path
+bitbakepath = os.path.join(fetchdir, 'bitbake')
+sys.path.insert(0, os.path.join(bitbakepath, 'lib'))
+
+from layerindex.models import Recipe, LayerBranch
+from rrs.models import RecipeUpstream, RecipeUpstreamHistory
+
+def get_upstream_info(thread_worker, arg):
+    from bb.utils import vercmp_string
+    from oe.recipeutils import get_recipe_upstream_version, \
+            get_recipe_pv_without_srcpv
+
+    (layerbranch, recipe_data, result) = arg
+
+    pn = recipe_data.getVar('PN', True)
+    recipe = Recipe.objects.get(layerbranch=layerbranch, pn=pn)
+    if not recipe:
+        logger.info("%s: in layer branch %s not found." % \
+                (pn, str(layerbranch)))
+        return
+
+    ru = RecipeUpstream()
+    ru.recipe = recipe
+
+    ru_info = None
+    try:
+        ru_info = get_recipe_upstream_version(recipe_data)
+    except Exception as e:
+        logger.exception("%s: in layer branch %s, %s" % (pn,
+            str(layerbranch), str(e)))
+
+    if ru_info is not None and ru_info['version']:
+        ru.version = ru_info['version']
+        ru.type = ru_info['type']
+        ru.date = ru_info['datetime']
+
+        pv, _, _ = get_recipe_pv_without_srcpv(recipe.pv, 
+                get_pv_type(recipe.pv)) 
+        upv, _, _ = get_recipe_pv_without_srcpv(ru_info['version'],
+                get_pv_type(ru_info['version']))
+
+        cmp_ver = vercmp_string(pv, upv)
+        if cmp_ver == -1:
+            ru.status = 'N' # Not update
+        elif cmp_ver == 0:
+            ru.status = 'Y' # Up-to-date
+        elif cmp_ver == 1:
+            ru.status = 'D' # Downgrade, need to review why
+    else:
+        ru.version = ''
+        ru.type = 'M'
+        ru.date = datetime.now()
+        ru.status = 'U' # Unknown
+
+    ru.no_update_reason = recipe_data.getVar('RECIPE_NO_UPDATE_REASON',
+            True) or ''
+
+    result.append((recipe, ru))
+
+if __name__=="__main__":
+    parser = optparse.OptionParser(usage = """%prog [options]""")
+    
+    parser.add_option("-d", "--debug",
+            help = "Enable debug output",
+            action="store_const", const=logging.DEBUG, dest="loglevel", default=logging.INFO)
+    
+    options, args = parser.parse_args(sys.argv)
+    logger.setLevel(options.loglevel)
+
+    logger.debug("Starting upstream history...")
+
+    transaction.enter_transaction_management()
+    transaction.managed(True)
+    for layerbranch in LayerBranch.objects.all():
+        (tinfoil, d, recipes) = load_recipes(layerbranch, bitbakepath,
+                fetchdir, settings, logger)
+
+        if not recipes:
+            continue
+
+        history = RecipeUpstreamHistory(start_date = datetime.now())
+
+        from oe.utils import ThreadedPool
+        import multiprocessing
+
+        nproc = min(multiprocessing.cpu_count(), len(recipes))
+        pool = ThreadedPool(nproc, len(recipes))
+
+        result = []
+        for recipe_data in recipes:
+            pool.add_task(get_upstream_info, (layerbranch,
+                recipe_data, result))
+
+        pool.start()
+        pool.wait_completion()
+
+        history.end_date = datetime.now()
+        history.save()
+
+        for res in result:
+            (recipe, ru) = res
+
+            ru.history = history
+            ru.save()
+
+            logger.debug('%s: layer branch %s, pv %s, upstream (%s)' % (recipe.pn,
+                str(layerbranch), recipe.pv, str(ru)))
+
+        tinfoil.shutdown()
+
+    transaction.commit()
+    transaction.leave_transaction_management()
+
+    history.end_date = datetime.now()
+    history.save()
+