lobster-codebeamer supports refs upstream reference

An optional config file as argument has been added to lobster-codebeamer.
This config file should includes a list of upstream references and respective cb-fieldNames

Resolves #45

CHANGELOG.md

@@ -4,6 +4,8 @@
 
 ### 0.9.18-dev
 
+* The `lobster-codebeamer` tool now supports `refs` as upstream reference
+
 * The `lobster-codebeamer` tool now allows items without summary
 
 * The `lobster-codebeamer` tool now uses codebeamer api v3.

lobster/tools/codebeamer/codebeamer.py

@@ -41,6 +41,8 @@
 import argparse
 import netrc
 from urllib.parse import quote
+from enum import Enum
+import json
 import requests
 
 from lobster.items import Tracing_Tag, Requirement
@@ -49,6 +51,26 @@
 from lobster.io import lobster_read, lobster_write
 
 
+class References(Enum):
+    REFS = "refs"
+
+
+SUPPORTED_REFERENCES = [References.REFS.value]
+
+
+def add_refs_refrences(req, flat_values_list):
+    # refs
+    for value in flat_values_list:
+        if value.get("id"):
+            ref_id = value.get("id")
+            req.add_tracing_target(Tracing_Tag("req", str(ref_id)))
+
+
+map_reference_name_to_function = {
+    References.REFS.value: add_refs_refrences
+}
+
+
 def query_cb_single(cb_config, url):
     assert isinstance(cb_config, dict)
     assert isinstance(url, str)
@@ -193,6 +215,27 @@ def to_lobster(cb_config, cb_item):
         text      = None,
         status    = status)
 
+    if cb_config.get('references'):
+        for reference_name, displayed_chosen_names in cb_config['references'].items():
+            if reference_name not in map_reference_name_to_function.keys():
+                continue
+
+            for displayed_name in displayed_chosen_names:
+                if cb_item.get(displayed_name):
+                    flat_values_list = cb_item.get(displayed_name) if isinstance(cb_item.get(displayed_name), list) \
+                        else [cb_item.get(displayed_name)]
+                else:
+                    flat_values_list = list(value for custom_field in cb_item["customFields"]
+                                            if custom_field["name"] == displayed_name and custom_field.get("values")
+                                            for value in custom_field["values"])
+                if not flat_values_list:
+                    continue
+
+                map_reference_name_to_function[reference_name](req, flat_values_list)
+
+    if "name" not in cb_item:
+        req.error("Item lacks a summary text")
+
     return req
 
 
@@ -210,17 +253,50 @@ def import_tagged(mh, cb_config, items_to_import):
     return rv
 
 
+def ensure_array_of_strings(instance):
+    if isinstance(instance, list) and all(isinstance(item, str) for item in instance):
+        return instance
+    else:
+        return [str(instance)]
+
+
+def parse_cb_config_file(file_name):
+    assert isinstance(file_name, str)
+    assert os.path.isfile(file_name)
+
+    with open(file_name, "r") as file:
+        data = json.loads(file.read())
+
+        provided_config_keys = set(data.keys())
+        supported_references = set(SUPPORTED_REFERENCES)
+
+        if not provided_config_keys.issubset(supported_references):
+            raise Exception("The provided references are not supported! "
+                            "supported referenes: '%s'" % ', '.join(SUPPORTED_REFERENCES))
+
+        json_config = {}
+        for key, value in data.items():
+            json_config[key] = ensure_array_of_strings(value)
+        return json_config
+
+
 def main():
     ap = argparse.ArgumentParser()
 
     modes = ap.add_mutually_exclusive_group(required=True)
     modes.add_argument("--import-tagged",
                        metavar="LOBSTER_FILE",
                        default=None)
+
     modes.add_argument("--import-query",
                        metavar="CB_QUERY_ID",
                        default=None)
 
+    ap.add_argument("--config-file",
+                    help=("name of codebeamer "
+                          "config file, supported references: '%s'" % ', '.join(SUPPORTED_REFERENCES)),
+                    default=None)
+
     ap.add_argument("--ignore-ssl-errors",
                     action="store_true",
                     default=False,
@@ -254,6 +330,12 @@ def main():
         "timeout"    : options.timeout,
     }
 
+    if options.config_file:
+        if os.path.isfile(options.config_file):
+            cb_config["references"] = parse_cb_config_file(options.config_file)
+        else:
+            ap.error("cannot open config file '%s'" % options.config_file)
+
     if cb_config["root"] is None:
         ap.error("please set CB_ROOT or use --cb-root")