From 44320ea668c58f6f70db2aa50fdbaa689403ef87 Mon Sep 17 00:00:00 2001 From: tischrei Date: Mon, 20 Feb 2023 13:50:46 +0000 Subject: [PATCH] add get_service_with_docs_by_service_type() --- otc_metadata/services.py | 19 +++++++++++++++++++ 1 file changed, 19 insertions(+) diff --git a/otc_metadata/services.py b/otc_metadata/services.py index 8fdc1704..76fd35f3 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