Function emitClientGlobalRegistrationMethod

scripts/codegen/python.ts:3779–3818

(
    lines: string[],
    groupName: string,
    method: RpcMethod,
    resolveType: (name: string) => string
)

}

function emitClientGlobalRegistrationMethod(
    lines: string[],
    groupName: string,
    method: RpcMethod,
    resolveType: (name: string) => string
): void {
    const rpcSegments = method.rpcMethod.split(".");
    const handlerVariableName = `handle_${rpcSegments.map(toSnakeCase).join("_")}`;
    const paramsType = resolveType(pythonParamsTypeName(method));
    const resultSchema = getMethodResultSchema(method);
    const nullableInner = resultSchema ? getNullableInner(resultSchema) : undefined;
    const hasResult = !isVoidSchema(resultSchema) && !nullableInner;
    const handlerField = toSnakeCase(groupName);
    const handlerMethod = clientSessionHandlerMethodName(method.rpcMethod);

    lines.push(`    async def ${handlerVariableName}(params: dict) -> dict | None:`);
    lines.push(`        request = ${paramsType}.from_dict(params)`);
    lines.push(`        handler = handlers.${handlerField}`);
    lines.push(`        if handler is None: raise RuntimeError("No ${handlerField} client-global handler registered")`);
    if (hasResult) {
        lines.push(`        result = await handler.${handlerMethod}(request)`);
        if (isObjectSchema(resultSchema)) {
            lines.push(`        return result.to_dict()`);
        } else {
            lines.push(`        return result.value if hasattr(result, 'value') else result`);
        }
    } else if (nullableInner) {
        lines.push(`        result = await handler.${handlerMethod}(request)`);
        const resolvedInner = resolveSchema(nullableInner, rpcDefinitions) ?? nullableInner;
        if (isObjectSchema(resolvedInner) || nullableInner.$ref) {
            lines.push(`        return result.to_dict() if result is not None else None`);
        } else {
            lines.push(`        return result`);
        }
    } else {
        lines.push(`        await handler.${handlerMethod}(request)`);
        lines.push(`        return None`);
    }
    lines.push(`    client.set_request_handler("${method.rpcMethod}", ${handlerVariableName})`);
}

async function generate(sessionSchemaPath?: string, apiSchemaPath?: string): Promise<void> {
    await generateSessionEvents(sessionSchemaPath);