Salome HOME
sat prepare fix products list leading ',' as ',KERNEL,...'
[tools/sat.git] / commands / prepare.py
index eef6968f8b26c7b541946c49bb62b0a865b7005d..d5093a214573e0dacc12649e4f357f3216dfd7f3 100644 (file)
 #  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
 
 import re
+import os
+import pprint as PP
 
 import src
+import src.debug as DBG
+
+PROPERTY_EXPRESSION = "^.+:.+$"
 
 # Define all possible option for prepare command :  sat prepare <options>
 parser = src.options.Options()
 parser.add_option('p', 'products', 'list2', 'products',
-    _('products to prepare. This option can be'
+    _('Optional: products to prepare. This option can be'
     ' passed several time to prepare several products.'))
+parser.add_option('', 'properties', 'string', 'properties',
+    _('Optional: Filter the products by their properties.\n\tSyntax: '
+      '--properties <property>:<value>'))
 parser.add_option('f', 'force', 'boolean', 'force', 
-    _("force to prepare the products in development mode."))
+    _("Optional: force to prepare the products in development mode."))
 parser.add_option('', 'force_patch', 'boolean', 'force_patch', 
-    _("force to apply patch to the products in development mode."))
+    _("Optional: force to apply patch to the products in development mode."))
 
 def get_products_list(options, cfg, logger):
     '''method that gives the product list with their informations from 
@@ -59,26 +67,47 @@ def get_products_list(options, cfg, logger):
     # the products name and their definition
     products_infos = src.product.get_products_infos(products, cfg)
     
+    # if the property option was passed, filter the list
+    if options.properties:
+        [prop, value] = options.properties.split(":")
+        products_infos = [(p_name, p_info) for 
+                          (p_name, p_info) in products_infos 
+                          if "properties" in p_info and 
+                          prop in p_info.properties and 
+                          p_info.properties[prop] == value]
+        
     return products_infos
 
-def remove_products(arguments, l_products_info, logger):
-    '''method that removes the products in l_products_info from arguments list.
+def find_products_already_getted(l_products):
+    '''function that returns the list of products that have an existing source 
+       directory.
     
-    :param arguments str: The arguments from which to remove products
-    :param l_products_info list: List of 
-                                 (str, Config) => (product_name, product_info)
-    :param logger Logger: The logger instance to use for the display and logging
-    :return: The updated arguments.
-    :rtype: str
+    :param l_products List: The list of products to check
+    :return: The list of product configurations that have an existing source 
+             directory.
+    :rtype: List
     '''
-    args = arguments
-    for i, (product_name, __) in enumerate(l_products_info):
-        args = args.replace(',' + product_name, '')
-        end_text = ', '
-        if i+1 == len(l_products_info):
-            end_text = '\n'            
-        logger.write(product_name + end_text, 1)
-    return args
+    l_res = []
+    for p_name_p_cfg in l_products:
+        __, prod_cfg = p_name_p_cfg
+        if os.path.exists(prod_cfg.source_dir):
+            l_res.append(p_name_p_cfg)
+    return l_res
+
+def find_products_with_patchs(l_products):
+    '''function that returns the list of products that have one or more patches.
+    
+    :param l_products List: The list of products to check
+    :return: The list of product configurations that have one or more patches.
+    :rtype: List
+    '''
+    l_res = []
+    for p_name_p_cfg in l_products:
+        __, prod_cfg = p_name_p_cfg
+        l_patchs = src.get_cfg_param(prod_cfg, "patches", [])
+        if len(l_patchs)>0:
+            l_res.append(p_name_p_cfg)
+    return l_res
 
 def description():
     '''method that is called when salomeTools is called with --help option.
@@ -87,7 +116,8 @@ def description():
     :rtype: str
     '''
     return _("The prepare command gets the sources of "
-             "the application products and apply the patches if there is any.")
+             "the application products and apply the patches if there is any."
+             "\n\nexample:\nsat prepare SALOME-master --products KERNEL,GUI")
   
 def run(args, runner, logger):
     '''method that is called when salomeTools is called with prepare parameter.
@@ -99,55 +129,68 @@ def run(args, runner, logger):
     # check that the command has been called with an application
     src.check_config_has_application( runner.cfg )
 
+    # Verify the --properties option
+    if options.properties:
+        oExpr = re.compile(PROPERTY_EXPRESSION)
+        if not oExpr.search(options.properties):
+            msg = _('WARNING: the "--properties" options must have the '
+                    'following syntax:\n--properties <property>:<value>')
+            logger.write(src.printcolors.printcWarning(msg), 1)
+            logger.write("\n", 1)
+            options.properties = None
+
     products_infos = get_products_list(options, runner.cfg, logger)
 
     # Construct the arguments to pass to the clean, source and patch commands
-    args_appli = runner.cfg.VARS.application + ' '
-    args_product_opt = '--products '
+    args_appli = config.VARS.application + " "  # useful whitespace
     if options.products:
-        for p_name in options.products:
-            args_product_opt += ',' + p_name
-    else:
-        for p_name, __ in products_infos:
-            args_product_opt += ',' + p_name
+        listProd = list(options.products)
+    else: # no product interpeted as all products
+        listProd = [name for name, tmp in products_infos]
+
+    DBG.write("prepare products", sorted(listProd))
+    args_product_opt = '--products ' + ",".join(listProd)
+    do_source = (len(listProd) > 0)
+
 
     ldev_products = [p for p in products_infos if src.product.product_is_dev(p[1])]
-    args_product_opt_clean = args_product_opt
+    newList = listProd # default
     if not options.force and len(ldev_products) > 0:
-        msg = _("Do not get the source of the following products "
-                "in development mode\nUse the --force option to"
-                " overwrite it.\n")
-        logger.write(src.printcolors.printcWarning(msg), 1)
-        args_product_opt_clean = remove_products(args_product_opt_clean,
-                                                 ldev_products,
-                                                 logger)
-        logger.write("\n", 1)
+        l_products_not_getted = find_products_already_getted(ldev_products)
+        listNot = [i for i, tmp in l_products_not_getted]
+        newList, removedList = removeInList(listProd, listNot)
+        if len(removedList) > 0:
+            msg = _("""\
+Do not get the source of the following products in development mode.
+Use the --force option to overwrite it.
+""")
+            msg += "\n%s\n" % ",".join(removedList)
+            logger.write(src.printcolors.printcWarning(msg), 1)
 
+    args_product_opt_clean = '--products ' + ",".join(newList)
+    do_clean = (len(newList) > 0)
     
-    args_product_opt_patch = args_product_opt
+    newList = listProd # default
     if not options.force_patch and len(ldev_products) > 0:
-        msg = _("do not patch the following products "
-                "in development mode\nUse the --force_patch option to"
-                " overwrite it.\n")
-        logger.write(src.printcolors.printcWarning(msg), 1)
-        args_product_opt_patch = remove_products(args_product_opt_patch,
-                                                 ldev_products,
-                                                 logger)
-        logger.write("\n", 1)
-
+        l_products_with_patchs = find_products_with_patchs(ldev_products)
+        listNot = [i for i, tmp in l_products_with_patchs]
+        newList, removedList = removeInList(listProd, listNot)
+        if len(removedList) > 0:
+            msg = _("""\
+Do not patch the following products in development mode.
+Use the --force_patch option to overwrite it.
+""")
+            msg += "\n%s\n" % ",".join(removedList)
+            logger.write(src.printcolors.printcWarning(msg), 1)
+                                                     
+    args_product_opt_patch = '--products ' + ",".join(newList)
+    do_patch = (len(newList) > 0)
+      
     # Construct the final commands arguments
     args_clean = args_appli + args_product_opt_clean + " --sources"
     args_source = args_appli + args_product_opt  
     args_patch = args_appli + args_product_opt_patch
-
-    # If there is no more any product in the command arguments,
-    # do not call the concerned command 
-    oExpr = re.compile("^--products *$")
-    do_clean = not(oExpr.search(args_product_opt_clean))
-    do_source = not(oExpr.search(args_product_opt))
-    do_patch = not(oExpr.search(args_product_opt_patch))
-    
-    
+      
     # Initialize the results to a failing status
     res_clean = 1
     res_source = 1
@@ -158,8 +201,7 @@ def run(args, runner, logger):
         msg = _("Clean the source directories ...")
         logger.write(msg, 3)
         logger.flush()
-        res_clean = runner.clean(args_clean, batch=True, verbose = 0,
-                                    logger_add_link = logger)
+        res_clean = runner.clean(args_clean, batch=True, verbose = 0, logger_add_link = logger)
         if res_clean == 0:
             logger.write('%s\n' % src.printcolors.printc(src.OK_STATUS), 3)
         else:
@@ -167,8 +209,7 @@ def run(args, runner, logger):
     if do_source:
         msg = _("Get the sources of the products ...")
         logger.write(msg, 5)
-        res_source = runner.source(args_source,
-                                    logger_add_link = logger)
+        res_source = runner.source(args_source, logger_add_link = logger)
         if res_source == 0:
             logger.write('%s\n' % src.printcolors.printc(src.OK_STATUS), 5)
         else:
@@ -176,11 +217,24 @@ def run(args, runner, logger):
     if do_patch:
         msg = _("Patch the product sources (if any) ...")
         logger.write(msg, 5)
-        res_patch = runner.patch(args_patch,
-                                    logger_add_link = logger)
+        res_patch = runner.patch(args_patch, logger_add_link = logger)
         if res_patch == 0:
             logger.write('%s\n' % src.printcolors.printc(src.OK_STATUS), 5)
         else:
             logger.write('%s\n' % src.printcolors.printc(src.KO_STATUS), 5)
     
-    return res_clean + res_source + res_patch
\ No newline at end of file
+    return res_clean + res_source + res_patch
+
+
+def removeInList(aList, removeList):
+    """Removes elements of removeList list from aList
+    
+    :param aList: (list) The list from which to remove elements
+    :param removeList: (list) The list which contains elements to remove
+    :return: (list, list) (list with elements removed, list of elements removed) 
+    """
+    res1 = [i for i in aList if i not in removeList]
+    res2 = [i for i in aList if i in removeList]
+    return (res1, res2)
+
+