1 files changed, 75 insertions, 0 deletions
diff --git a/make_ids.py b/make_ids.py
new file mode 100644
index 0000000..d62d9a6
--- /dev/null
+++ b/make_ids.py
@@ -0,0 +1,75 @@
+# -*- coding: utf8 -*-
+# org.onap.vnfrqts/requirements
+# ============LICENSE_START====================================================
+# Copyright © 2018 AT&T Intellectual Property. All rights reserved.
+#
+# Unless otherwise specified, all software contained herein is licensed
+# under the Apache License, Version 2.0 (the "License");
+# you may not use this software except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#             http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#
+# Unless otherwise specified, all documentation contained herein is licensed
+# under the Creative Commons License, Attribution 4.0 Intl. (the "License");
+# you may not use this documentation except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#             https://creativecommons.org/licenses/by/4.0/
+#
+# Unless required by applicable law or agreed to in writing, documentation
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#
+# ============LICENSE_END============================================
+import argparse
+import itertools
+import json
+import random
+
+REQUIREMENTS_FILE = "docs/data/needs.json"
+
+
+def load_all_ids():
+    """Loads the """
+    with open(REQUIREMENTS_FILE, "r") as f:
+        data = json.load(f)
+        result = set()
+        for version in data["versions"]:
+            result.update(data["versions"][version]["needs"].keys())
+        return result
+
+
+def generate_ids():
+    """Generates a stream of unique requirement IDs"""
+    all_ids = load_all_ids()
+    while True:
+        new_id = "R-{:0>5d}".format(random.randint(0, 999999))
+        if new_id in all_ids:
+            continue  # skip this one and generate another one
+        all_ids.add(new_id)
+        yield new_id
+
+
+if __name__ == "__main__":
+    parser = argparse.ArgumentParser(
+        description="""
+        Generate random, unique requirement IDs for use when adding new requirements
+        to the RST documentation.
+        """
+    )
+    parser.add_argument("num_ids", action="store", nargs="?", type=int, default=1,
+                        help="Number of IDs to generate")
+    args = parser.parse_args()
+    for req_id in itertools.islice(generate_ids(), args.num_ids):
+        print(req_id)
+
+