search_pack_index() - Code Metrics - Inspection of "Pack management" - StackStorm/st2 - Measure and Improve Code Quality continuously with Scrutinizer

Completed

Pull Request — master (#2842)

by Edward

created 2016-10-19 04:13 UTC

search_pack_index() F

↳ Parent: Project

Complexity

Conditions

Size

Total Lines

Duplication

Lines	0
Ratio	0 %

Importance

Changes	5
Bugs	0	Features	0

Metric	Value
cc	11
c	5
b	0
f	0
dl	0
loc	31
rs	3.1764

How to fix Complexity

# Licensed to the StackStorm, Inc ('StackStorm') under one or more
# contributor license agreements.  See the NOTICE file distributed with
# this work for additional information regarding copyright ownership.
# The ASF licenses this file to You under the Apache License, Version 2.0
# (the "License"); you may not use this file 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.

import itertools
import json

import requests
import six
from oslo_config import cfg

from st2common import log as logging
from st2common.models.api.pack import PackAPI
from st2common.persistence.pack import Pack

__all__ = [
    'get_pack_by_ref',
    'fetch_pack_index',
    'get_pack_from_index',
    'search_pack_index',
    'check_index_health'
]

EXCLUDE_FIELDS = [
    "repo_url",
    "email"
]

SEARCH_PRIORITY = [
    "name",
    "keywords"
]

LOG = logging.getLogger(__name__)


def _build_index_list(index_url):
    if not index_url:
        # Reversing the indexes list from config so that the indexes have
        # descending (left-to-right) priority.
        # When multiple indexes have a pack with a given name, the index
        # that comes first in the list will be used.
        index_urls = cfg.CONF.content.index_url[::-1]
    elif isinstance(index_url, str):
        index_urls = [index_url]
    elif hasattr(index_url, '__iter__'):
        index_urls = index_url
    else:
        raise TypeError('"index_url" should either be a string or an iterable object.')
    return index_urls


def _fetch_and_compile_index(index_urls, logger=None):
    """
    Go through the index list and compile results into a single object.
    """
    status = []
    index = {}

    for index_url in index_urls:

        index_status = {
            'url': index_url,
            'packs': 0,
            'message': None,
            'error': None,
        }
        index_json = None

        try:
            request = requests.get(index_url)
            request.raise_for_status()
            index_json = request.json()
        except ValueError as e:
            index_status['error'] = 'malformed'
            index_status['message'] = repr(e)
        except requests.exceptions.RequestException as e:
            index_status['error'] = 'unresponsive'
            index_status['message'] = repr(e)

        if index_json == {}:
            index_status['error'] = 'empty'
            index_status['message'] = 'The index URL returned an empty object.'
        elif type(index_json) is list:
            index_status['error'] = 'malformed'
            index_status['message'] = 'Expected an index object, got a list instead.'

        if index_status['error']:
            logger.error("Index parsing error: %s" % json.dumps(index_status, indent=4))
        else:
            index_status['message'] = 'Success.'
            index_status['packs'] = len(index_json)
            index.update(index_json)

        status.append(index_status)

    return index, status


def get_pack_by_ref(pack_ref):
    """
    Retrieve PackDB by the provided reference.
    """
    pack_db = Pack.get_by_ref(pack_ref)
    return pack_db


def fetch_pack_index(index_url=None, logger=None):
    """
    Fetch the pack indexes (either from the config or provided as an argument)
    and return the object.
    """
    logger = logger or LOG

    index_urls = _build_index_list(index_url)
    index, status = _fetch_and_compile_index(index_urls, logger)

    # If one of the indexes on the list is unresponsive, we do not throw
    # immediately. The only case where an exception is raised is when no
    # results could be obtained from all listed indexes.
    # This behavior allows for mirrors / backups and handling connection
    # or network issues in one of the indexes.
    if not index:
        raise ValueError("No results from the %s: tried %s.\nStatus: %s" % (
            ("index" if len(index_urls) == 1 else "indexes"),
            ", ".join(index_urls),
            json.dumps(status, indent=4)
        ))
    return index


def check_index_health(index_url=None, status=None, logger=None):
    """
    Check if all listed indexes are healthy: they should be reachable,
    return valid JSON objects, and yield more than one result.
    """
    logger = logger or LOG

    if not status:
        index_urls = _build_index_list(index_url)
        _, status = _fetch_and_compile_index(index_urls, logger)

    health = {
        "indexes": {
            "count": len(status),
            "valid": 0,
            "invalid": 0,
            "errors": {},
            "status": status,
        },
        "packs": {
            "count": 0,
        },
    }

    for index in status:
        if index['error']:
            error_count = health['indexes']['errors'].get(index['error'], 0) + 1
            health['indexes']['invalid'] += 1
            health['indexes']['errors'][index['error']] = error_count
        else:
            health['indexes']['valid'] += 1
        health['packs']['count'] += index['packs']

    return health


def get_pack_from_index(pack):
    """
    Search index by pack name.
    Returns a pack.
    """
    if not pack:
        raise ValueError("Pack name must be specified.")

    index = fetch_pack_index()

    return PackAPI(**index.get(pack))


def search_pack_index(query, exclude=None, priority=None):
    """
    Search the pack index by query.
    Returns a list of matches for a query.
    """
    if not query:
        raise ValueError("Query must be specified.")

    if not exclude:
        exclude = EXCLUDE_FIELDS
    if not priority:
        priority = SEARCH_PRIORITY

    index = fetch_pack_index()

    matches = [[] for _ in range(len(priority) + 1)]
    for pack_dict in six.itervalues(index):
        pack = PackAPI(**pack_dict)

        for key, value in six.iteritems(vars(pack)):
            if not hasattr(value, '__contains__'):
                value = str(value)

            if key not in exclude and query in value:
                if key in priority:
                    matches[priority.index(key)].append(pack)
                else:
                    matches[-1].append(pack)
                break

    return list(itertools.chain.from_iterable(matches))


1			# Licensed to the StackStorm, Inc ('StackStorm') under one or more
2			# contributor license agreements. See the NOTICE file distributed with
3			# this work for additional information regarding copyright ownership.
4			# The ASF licenses this file to You under the Apache License, Version 2.0
5			# (the "License"); you may not use this file except in compliance with
6			# the License. You may obtain a copy of the License at
7			#
8			# http://www.apache.org/licenses/LICENSE-2.0
9			#
10			# Unless required by applicable law or agreed to in writing, software
11			# distributed under the License is distributed on an "AS IS" BASIS,
12			# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13			# See the License for the specific language governing permissions and
14			# limitations under the License.
15
16			import itertools
17			import json
18
19			import requests
20			import six
21			from oslo_config import cfg
22
23			from st2common import log as logging
24			from st2common.models.api.pack import PackAPI
25			from st2common.persistence.pack import Pack
26
27			__all__ = [
28			'get_pack_by_ref',
29			'fetch_pack_index',
30			'get_pack_from_index',
31			'search_pack_index',
32			'check_index_health'
33			]
34
35			EXCLUDE_FIELDS = [
36			"repo_url",
37			"email"
38			]
39
40			SEARCH_PRIORITY = [
41			"name",
42			"keywords"
43			]
44
45			LOG = logging.getLogger(__name__)
46
47
48			def _build_index_list(index_url):
49			if not index_url:
50			# Reversing the indexes list from config so that the indexes have
51			# descending (left-to-right) priority.
52			# When multiple indexes have a pack with a given name, the index
53			# that comes first in the list will be used.
54			index_urls = cfg.CONF.content.index_url[::-1]
55			elif isinstance(index_url, str):
56			index_urls = [index_url]
57			elif hasattr(index_url, '__iter__'):
58			index_urls = index_url
59			else:
60			raise TypeError('"index_url" should either be a string or an iterable object.')
61			return index_urls
62
63
64			def _fetch_and_compile_index(index_urls, logger=None):
65			"""
66			Go through the index list and compile results into a single object.
67			"""
68			status = []
69			index = {}
70
71			for index_url in index_urls:
72
73			index_status = {
74			'url': index_url,
75			'packs': 0,
76			'message': None,
77			'error': None,
78			}
79			index_json = None
80
81			try:
82			request = requests.get(index_url)
83			request.raise_for_status()
84			index_json = request.json()
85			except ValueError as e:
86			index_status['error'] = 'malformed'
87			index_status['message'] = repr(e)
88			except requests.exceptions.RequestException as e:
89			index_status['error'] = 'unresponsive'
90			index_status['message'] = repr(e)
91
92			if index_json == {}:
93			index_status['error'] = 'empty'
94			index_status['message'] = 'The index URL returned an empty object.'
95			elif type(index_json) is list:
96			index_status['error'] = 'malformed'
97			index_status['message'] = 'Expected an index object, got a list instead.'
98
99			if index_status['error']:
100			logger.error("Index parsing error: %s" % json.dumps(index_status, indent=4))
101			else:
102			index_status['message'] = 'Success.'
103			index_status['packs'] = len(index_json)
104			index.update(index_json)
105
106			status.append(index_status)
107
108			return index, status
109
110
111			def get_pack_by_ref(pack_ref):
112			"""
113			Retrieve PackDB by the provided reference.
114			"""
115			pack_db = Pack.get_by_ref(pack_ref)
116			return pack_db
117
118
119			def fetch_pack_index(index_url=None, logger=None):
120			"""
121			Fetch the pack indexes (either from the config or provided as an argument)
122			and return the object.
123			"""
124			logger = logger or LOG
125
126			index_urls = _build_index_list(index_url)
127			index, status = _fetch_and_compile_index(index_urls, logger)
128
129			# If one of the indexes on the list is unresponsive, we do not throw
130			# immediately. The only case where an exception is raised is when no
131			# results could be obtained from all listed indexes.
132			# This behavior allows for mirrors / backups and handling connection
133			# or network issues in one of the indexes.
134			if not index:
135			raise ValueError("No results from the %s: tried %s.\nStatus: %s" % (
136			("index" if len(index_urls) == 1 else "indexes"),
137			", ".join(index_urls),
138			json.dumps(status, indent=4)
139			))
140			return index
141
142
143			def check_index_health(index_url=None, status=None, logger=None):
144			"""
145			Check if all listed indexes are healthy: they should be reachable,
146			return valid JSON objects, and yield more than one result.
147			"""
148			logger = logger or LOG
149
150			if not status:
151			index_urls = _build_index_list(index_url)
152			_, status = _fetch_and_compile_index(index_urls, logger)
153
154			health = {
155			"indexes": {
156			"count": len(status),
157			"valid": 0,
158			"invalid": 0,
159			"errors": {},
160			"status": status,
161			},
162			"packs": {
163			"count": 0,
164			},
165			}
166
167			for index in status:
168			if index['error']:
169			error_count = health['indexes']['errors'].get(index['error'], 0) + 1
170			health['indexes']['invalid'] += 1
171			health['indexes']['errors'][index['error']] = error_count
172			else:
173			health['indexes']['valid'] += 1
174			health['packs']['count'] += index['packs']
175
176			return health
177
178
179			def get_pack_from_index(pack):
180			"""
181			Search index by pack name.
182			Returns a pack.
183			"""
184			if not pack:
185			raise ValueError("Pack name must be specified.")
186
187			index = fetch_pack_index()
188
189			return PackAPI(**index.get(pack))
190
191
192			def search_pack_index(query, exclude=None, priority=None):
193			"""
194			Search the pack index by query.
195			Returns a list of matches for a query.
196			"""
197			if not query:
198			raise ValueError("Query must be specified.")
199
200			if not exclude:
201			exclude = EXCLUDE_FIELDS
202			if not priority:
203			priority = SEARCH_PRIORITY
204
205			index = fetch_pack_index()
206
207			matches = [[] for _ in range(len(priority) + 1)]
208			for pack_dict in six.itervalues(index):
209			pack = PackAPI(**pack_dict)
210
211			for key, value in six.iteritems(vars(pack)):
212			if not hasattr(value, '__contains__'):
213			value = str(value)
214
215			if key not in exclude and query in value:
216			if key in priority:
217			matches[priority.index(key)].append(pack)
218			else:
219			matches[-1].append(pack)
220			break
221
222			return list(itertools.chain.from_iterable(matches))
223

StackStorm / st2

Pull Request — master (#2842)

search_pack_index() F

Complexity

Size

Duplication

Importance

How to fix Complexity

Complexity

Duplication Side-by-Side

Filter issues like