summaryrefslogtreecommitdiffstats
path: root/make_ids.py
blob: d62d9a658688cec4934ed1f189fe3576a46c5391 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
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)