summaryrefslogtreecommitdiffstats
path: root/make_ids.py
blob: e101293152f86776c89e84c6d5fd325675d6cf61 (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
76
77
78
79
80
81
82
83
84
85
86
# -*- 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
import requests
import urllib

REQUIREMENTS_FILE = "https://nexus.onap.org/content/sites/raw/org.onap.vnfrqts.requirements/master/needs.json"

def get_current_requirements():
    """Read the current ONAP requirements from ONAP doc site"""
    try:
        r = requests.get(REQUIREMENTS_FILE)
        if r.headers.get('content-type') == 'application/json':
            data = r.content
            return json.loads(data)
        else:
            warnings.warning("Unexpected content-type ({}) encountered downloading requirements.json, using last saved copy".format(r.headers.get('content-type')))
    except requests.exceptions.RequestException as e:
        warnings.warn("Error downloading latest JSON, using last saved copy.")
        warnings.warn(UserWarning(e))

def load_all_ids():
    """Loads the """
    data = get_current_requirements()
    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)