ocrd_network.client

Complexity

Total Complexity

6

Size/Duplication

Total Lines	38
Duplicated Lines	0 %

Importance

Changes

0

2 Methods

Rating	Name	Duplication	Size	Complexity
A	Client.__init__()	0	9	1
A	Client.send_processing_request()	0	9	1

import json
from os import environ
import requests


# TODO: This is just a conceptual implementation and first try to
#  trigger further discussions on how this should look like.
class Client:
    def __init__(
            self,
            server_addr_processing: str = environ.get('OCRD_NETWORK_SERVER_ADDR_PROCESSING', ''),
            server_addr_workflow: str = environ.get('OCRD_NETWORK_SERVER_ADDR_WORKFLOW', ''),
            server_addr_workspace: str = environ.get('OCRD_NETWORK_SERVER_ADDR_WORKSPACE', ''),
    ):
        self.server_addr_processing = server_addr_processing
        self.server_addr_workflow = server_addr_workflow
        self.server_addr_workspace = server_addr_workspace

    def send_processing_request(self, processor_name: str, req_params: dict):
        verify_server_protocol(self.server_addr_processing)
        req_url = f'{self.server_addr_processing}/processor/{processor_name}'
        req_headers = {"Content-Type": "application/json; charset=utf-8"}
        req_json = json.loads(json.dumps(req_params))

        print(f'Sending processing request to: {req_url}')
        response = requests.post(url=req_url, headers=req_headers, json=req_json)
        return response.json()


def verify_server_protocol(address: str):
    protocol_matched = False
    for protocol in ['http://', 'https://']:
        if address.startswith(protocol):
            protocol_matched = True
            break
    if not protocol_matched:
        raise ValueError(f'Wrong/Missing protocol in the server address: {address}')