diff --git a/tools/index_metadata.py b/tools/index_metadata.py index 9251b85b..15078e7c 100644 --- a/tools/index_metadata.py +++ b/tools/index_metadata.py @@ -22,6 +22,11 @@ def parse_args(): help='Option deletes old index with the same name and creates new ' 'one.' ) + parser.add_argument( + '--all-doc-types', + action='store_true', + help='Upload all doc-types instead of only umn, api-ref and dev' + ) parser.add_argument( '--hosts', metavar='', @@ -62,7 +67,8 @@ def main(): logging.debug("Obtaining data from otc_metadata") data = getData( - environment=args.target_environment + environment=args.target_environment, + all_doc_types=args.all_doc_types ) logging.debug("Indexing data into OpenSearch") @@ -76,10 +82,16 @@ def main(): ) -def getData(environment): - return metadata.service_types_with_doc_types( +def getData(environment, all_doc_types): + metadatadata = metadata.service_types_with_doc_types( environment=environment ) + if not all_doc_types: + for i, doc in enumerate(metadatadata["docs"], 0): + if doc["type"] not in ["umn", "api-ref", "dev"]: + metadatadata["docs"].pop(i) + + return metadatadata def indexData(deleteIndex, hosts, index, username, password, data):