query.py 1.68 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
27
28
29
30
31
32
33
@click.option(
    "-l",
    "--limit",
    "limit",
    default=10,
    show_default=True,
    help="maximum number of records to return.",
)
Mark Hewitt's avatar
Mark Hewitt committed
34
@handle_cli_exceptions
35
@command_with_output("results[*].{Id:id,Kind:kind,CreateTime:createTime}")
Mark Hewitt's avatar
Mark Hewitt committed
36
def _click_command(state: State, kind: str, _id: str, limit: int):
Mark Hewitt's avatar
Mark Hewitt committed
37
    """Query search service"""
Mark Hewitt's avatar
Mark Hewitt committed
38
    return query(state, kind, _id, limit)
Mark Hewitt's avatar
Mark Hewitt committed
39
40


Mark Hewitt's avatar
Mark Hewitt committed
41
42
43
def query(
    state: State, kind: str, id: str, limit: int
):  # pylint: disable=invalid-name,redefined-builtin
Mark Hewitt's avatar
Mark Hewitt committed
44
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
    json_response = search_client.query(kind, id, limit)
Mark Hewitt's avatar
Mark Hewitt committed
53
54

    return json_response