From 8de5c8ba57458c2e3216014396fa529833c7f475 Mon Sep 17 00:00:00 2001 From: tischrei Date: Tue, 21 Feb 2023 10:44:51 +0000 Subject: [PATCH] add get_service_with_docs_by_service_type() Reviewed-by: Hasko, Vladimir Co-authored-by: tischrei Co-committed-by: tischrei --- otc_metadata/services.py | 19 +++++++++++++++++++ 1 file changed, 19 insertions(+) diff --git a/otc_metadata/services.py b/otc_metadata/services.py index 8fdc1704..195b067e 100644 --- a/otc_metadata/services.py +++ b/otc_metadata/services.py @@ -209,3 +209,22 @@ class Services(object): doc_struct[srv["service_category"]].append(srv_res) return dict(categories=doc_struct) + + def get_service_with_docs_by_service_type(self, service_type): + """Retrieve service and service docs by service_type + + :param str service_type: Filter by service_type + """ + res = dict() + res['service'] = {} + docs = [] + services = self._service_data + for doc in services['documents']: + if doc['service_type'] == service_type: + docs.append(doc) + res['documents'] = docs + for service in services['services']: + if service['service_type'] == service_type: + res['service'] = service + break + return res