query.py 1.84 KB
Newer Older
Mark Hewitt's avatar
Mark Hewitt committed
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
#  Licensed 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.

"""Search service query command"""

import click
from osdu.search import SearchClient

18
from osducli.click_cli import CustomClickCommand, State, command_with_output
Mark Hewitt's avatar
Mark Hewitt committed
19
20
21
22
from osducli.cliclient import CliOsduClient, handle_cli_exceptions


# click entry point
23
@click.command(cls=CustomClickCommand)
Mark Hewitt's avatar
Mark Hewitt committed
24
25
@click.option("-k", "--kind", "kind", help="kind to search for")
@click.option("-id", "--id", "_id", help="id to search for")
Mark Hewitt's avatar
Mark Hewitt committed
26
@click.option("-q", "--query", "_query", help="custom search query")
Mark Hewitt's avatar
Mark Hewitt committed
27
28
29
30
31
32
33
34
@click.option(
    "-l",
    "--limit",
    "limit",
    default=10,
    show_default=True,
    help="maximum number of records to return.",
)
Mark Hewitt's avatar
Mark Hewitt committed
35
@handle_cli_exceptions
36
@command_with_output("results[*].{Id:id,Kind:kind,CreateTime:createTime}")
Mark Hewitt's avatar
Mark Hewitt committed
37
38
39
def _click_command(state: State, kind: str, _id: str, _query: str, limit: int):
    """Search using more advanced query terms"""
    return query(state, kind, _id, _query, limit)
Mark Hewitt's avatar
Mark Hewitt committed
40
41


Mark Hewitt's avatar
Mark Hewitt committed
42
def query(
Mark Hewitt's avatar
Mark Hewitt committed
43
    state: State, kind: str, id: str, custom_query: str, limit: int
Mark Hewitt's avatar
Mark Hewitt committed
44
):  # pylint: disable=invalid-name,redefined-builtin
Mark Hewitt's avatar
Mark Hewitt committed
45
46
47
48
49
50
51
    """Query search service

    Args:
        state (State): Global state
    """
    connection = CliOsduClient(state.config)
    search_client = SearchClient(connection)
Mark Hewitt's avatar
Mark Hewitt committed
52
53
    print(custom_query)
    json_response = search_client.query(kind, id, custom_query, limit)
Mark Hewitt's avatar
Mark Hewitt committed
54
55

    return json_response