Elixir Pipe в Python

Мне очень нравится оператор Pipe в Elixir. Это что-то из будущего!

Поскольку я чаще работаю с Python, я попытался реализовать эту функциональность pipe в Python с помощью нескольких строк кода.

from functools import reduce


def pipe(*fn_list):
    return reduce(
        lambda x, f: f(x) if not isinstance(x, list) else f(*x),
        fn_list,
    )

Вход в полноэкранный режим Выход из полноэкранного режима

Как это использовать?

Пример —

import json
from dataclasses import dataclass
import requests


@dataclass
class Request:
    url: str
    method: str


def parse_request(req: str):
    return pipe(
        req,
        json.loads,
        lambda x: Request(**x),
    )


def validate_url(req: Request):
    return req if req.url.startswith("http") else None


def validate_method(req: Request):
    return req if req.method in ["GET", "POST", "PUT", "DELETE"] else None


def get_content(req: Request):
    return requests.get(req.url).content


def check_content_type(content: str, content_type: str):
    return content if content.startswith(content_type) else None


print(
    pipe(
        """{"url": "https://google.com", "method": "GET"}""",
        parse_request,
        validate_url,
        validate_method,
        get_content,
        lambda content: content.decode("utf-8"),
        lambda content: check_content_type(content, "<!doctype html>"),
    )
)

Войти в полноэкранный режим Выход из полноэкранного режима

Вы можете использовать lambda и функции в pipe.

Обратите внимание, что последняя функция в pipe, check_content_type принимает два аргумента, первый из которых content передается из предыдущей функции pipe, поэтому мы берем его в качестве аргумента лямбды и затем вызываем check_content_type с этим content и другим аргументом.

Дайте мне знать ваши мысли!

** Изображение обложки взято отсюда.

Оцените статью
Procodings.ru
Добавить комментарий