Simple JSONRPC router with a flask-like interface

api.py

class Router(object):
    def __init__(self):
        super(Router, self).__init__()
        self.handlers = {}

    def method(self, name: str):
        def wrapper(func: callable):
            self.add_handler(name, func)
            def wrapped(*args, **kwargs):
                return func(*args, **kwargs)
            return wrapped
        return wrapper

    def __call__(self, rpc):
        method = rpc['method']
        params = rpc['params']
        function = self.handlers[method]
        return function(params)


# Usage:
router = Router()

@router.method('nice')
def method(params):
    print(params)
   
rpc = {
   'method': 'nice',
   'params': [1, 2]
}

router(rpc)
# [1, 2]

models.py

from typing import Union
from pydantic import BaseModel, Field, Extra


class BaseV2(BaseModel):
    jsonrpc: Field(str, const=True) = '2.0'

    class Config:
        extra = Extra.forbid


class Request(BaseV2):
    id: str
    method: str
    params: Union[list, dict] = None


class Notification(BaseV2):
    method: str
    params: Union[list, dict] = None


Batch = Union[Request, Notification]


class Success(BaseV2):
    id: str
    result: Union[dict, list, float, int, str, bool]


class ErrorObject(BaseModel):
    code: int
    message: str
    data: Union[dict, list, float, int, str, bool] = None


class Error(BaseV2):
    id: str
    error: ErrorObject