Current File : //opt/imunify360/venv/lib64/python3.11/site-packages/im360/simple_rpc/schema.py
from defence360agent.contracts.config import UserType
from defence360agent.rpc_tools.middleware import (
default_to_items as items_middleware,
)
from imav.simple_rpc.schema import init_validator as init_base_validator
from im360.simple_rpc.middleware import (
add_strategy,
replace_gray_splashscreen_with_gray,
)
def init_validator(schema_validator, validate_middleware, schema_paths):
validator, middleware, middleware_exclude = init_base_validator(
schema_validator, validate_middleware, schema_paths
)
strategy_middleware = (add_strategy, (UserType.ROOT,))
gray_splashscreen_middleware = (
replace_gray_splashscreen_with_gray,
(UserType.ROOT,),
)
# find index with the items middleware
i = next(
i for i, (m, _) in enumerate(middleware[None]) if m is items_middleware
)
# insert middlewares before the items middleware
middleware[None].insert(i, strategy_middleware)
middleware[None].insert(i, gray_splashscreen_middleware)
return validator, middleware, middleware_exclude
Mini Shell