ocrd_network.cli.client

Complexity

Total Complexity

11

Size/Duplication

Total Lines	102
Duplicated Lines	0 %

Importance

Changes

0

6 Functions

Rating	Name	Duplication	Size	Complexity
A	discovery_cli()	0	6	1
B	send_processing_request()	0	50	6
A	workflow_cli()	0	6	1
A	client_cli()	0	7	1
A	processing_cli()	0	6	1
A	workspace_cli()	0	6	1

import click
from typing import Optional

from ocrd.decorators import (
    parameter_option,
    parameter_override_option
)
from ocrd_network import Client


@click.group('client')
def client_cli():
    """
    A client for interacting with the network modules.
    The client CLI mimics the WebAPI endpoints
    """
    pass


@client_cli.group('discovery')
def discovery_cli():
    """
    The discovery endpoint of the WebAPI
    """
    pass


@client_cli.group('processing')
def processing_cli():
    """
    The processing endpoint of the WebAPI
    """
    pass


@processing_cli.command('processor')
@click.argument('processor_name', required=True, type=click.STRING)
@click.option('--address')
@click.option('-m', '--mets', required=True, default="mets.xml")
@click.option('-I', '--input-file-grp', default='OCR-D-INPUT')
@click.option('-O', '--output-file-grp', default='OCR-D-OUTPUT')
@click.option('-g', '--page-id')
@parameter_option
@click.option('--result-queue-name')
@click.option('--callback-url')
@click.option('--agent-type', default='worker')
def send_processing_request(
        address: Optional[str],
        processor_name: str,
        mets: str,
        input_file_grp: str,
        output_file_grp: Optional[str],
        page_id: Optional[str],
        parameter: Optional[dict],
        result_queue_name: Optional[str],
        callback_url: Optional[str],
        # TODO: This is temporally available to toggle
        #  between the ProcessingWorker/ProcessorServer
        agent_type: Optional[str]
):
    req_params = {
        "path_to_mets": mets,
        "description": "OCR-D Network client request",
        "input_file_grps": input_file_grp.split(','),
        "parameters": parameter if parameter else {},
        "agent_type": agent_type,
    }
    if output_file_grp:
        req_params["output_file_grps"] = output_file_grp.split(',')
    if page_id:
        req_params["page_id"] = page_id
    if result_queue_name:
        req_params["result_queue_name"] = result_queue_name
    if callback_url:
        req_params["callback_url"] = callback_url

    client = Client(
        server_addr_processing=address
    )
    response = client.send_processing_request(
        processor_name=processor_name,
        req_params=req_params
    )
    processing_job_id = response.get('job_id', None)
    print(f"Processing job id: {processing_job_id}")


@client_cli.group('workflow')
def workflow_cli():
    """
    The workflow endpoint of the WebAPI
    """
    pass


@client_cli.group('workspace')
def workspace_cli():
    """
    The workspace endpoint of the WebAPI
    """
    pass