client.go.tmpl

{{- define "client" -}}
{{- $typeMap := .TypeMap -}}

{{- if .Services}}
#
# Imports
#
import json
import urllib.request
from urllib.error import HTTPError
from typing import Optional, Dict

#
# Client
#

{{- range .Services}}
class {{.Name}}:
    CONTENT_TYPE_JSON = {'Content-Type': 'application/json'}

    def __init__(self, hostname: str):
        self.hostname = hostname
        self.path = '/rpc/{{.Name}}/'

    def url(self, name: str) -> str:
        return f"{self.hostname}{self.path}{name}"

    def _send_request(self, method_name: str, args, headers: Optional[Dict[str, str]] = None):
        url = self.url(method_name)
        headers = headers or self.CONTENT_TYPE_JSON
        body = json.dumps(args.to_dict()).encode('utf-8')  # Encode to bytes
        req = urllib.request.Request(url, data=body, headers=headers)

        try:
            with urllib.request.urlopen(req) as response:
                response_data = json.loads(response.read().decode('utf-8'))
            return response_data
        except HTTPError as ex:  # Catch only HTTPError
            error_body = ex.read().decode('utf-8')  # Read the response body
            try:
                error_data = json.loads(error_body)  # Attempt to parse it as JSON
            except json.JSONDecodeError:
                error_data = {"error": "Unknown", "message": error_body}
            
            # Directly raise your custom exception
            raise WebrpcRequestFailedError(
                message=error_data.get('msg', 'Unknown error'),
                code=error_data.get('code', -1),
                cause=error_data.get('cause'),
                status=error_data.get('status')
            )


    {{- range .Methods}}
    def {{.Name}}(self, args: {{.Name}}Args, headers: Optional[Dict[str, str]] = None):
        response_data = self._send_request('{{.Name}}', args, headers)
        return {{.Name}}Return(**response_data)
    {{- end}}
{{- end}}
{{- end}}
{{- end -}}