id.py 1.6 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
@click.argument("id")
Mark Hewitt's avatar
Mark Hewitt committed
25
26
27
28
29
30
31
32
@click.option(
    "-l",
    "--limit",
    "limit",
    default=10,
    show_default=True,
    help="maximum number of records to return.",
)
Mark Hewitt's avatar
Mark Hewitt committed
33
@handle_cli_exceptions
34
@command_with_output("results[*]")
Mark Hewitt's avatar
Mark Hewitt committed
35
36
37
def _click_command(
    state: State, id: str, limit: int
):  # noqa:W1 pylint: disable=invalid-name,redefined-builtin
Mark Hewitt's avatar
Mark Hewitt committed
38
    """Search for the specified id"""
Mark Hewitt's avatar
Mark Hewitt committed
39
    return query(state, id, limit)
Mark Hewitt's avatar
Mark Hewitt committed
40
41


Mark Hewitt's avatar
Mark Hewitt committed
42
43
44
def query(
    state: State, id: str, limit: int
):  # TO FIX later pylint: disable=invalid-name,redefined-builtin
Mark Hewitt's avatar
Mark Hewitt committed
45
46
47
48
49
50
51
52
    """Search for the specified id

    Args:
        state (State): Global state
    """
    connection = CliOsduClient(state.config)

    search_client = SearchClient(connection)
Mark Hewitt's avatar
Mark Hewitt committed
53
    json_response = search_client.query_by_id(id, limit)
Mark Hewitt's avatar
Mark Hewitt committed
54
55

    return json_response