95 lines
2.4 KiB
Python
95 lines
2.4 KiB
Python
#!/usr/bin/env python3
|
||
# -*- coding: UTF-8 -*-
|
||
# -----------------------------------------------------------------------------------------------------------
|
||
# Copyright (c) 2025 Huawei Technologies Co., Ltd.
|
||
# This program is free software, you can redistribute it and/or modify it under the terms and conditions of
|
||
# CANN Open Software License Agreement Version 2.0 (the "License").
|
||
# Please refer to the License for details. You may not use this file except in compliance with the License.
|
||
# THIS SOFTWARE IS PROVIDED ON AN "AS IS" BASIS, WITHOUT WARRANTIES OF ANY KIND, EITHER EXPRESS OR IMPLIED,
|
||
# INCLUDING BUT NOT LIMITED TO NON-INFRINGEMENT, MERCHANTABILITY, OR FITNESS FOR A PARTICULAR PURPOSE.
|
||
# See LICENSE in the root of the software repository for the full text of the License.
|
||
# -----------------------------------------------------------------------------------------------------------
|
||
|
||
"""函数基础库。"""
|
||
|
||
import operator
|
||
from collections.abc import Callable, Iterator
|
||
from typing import TypeVar
|
||
|
||
A = TypeVar("A")
|
||
|
||
|
||
def constant(value: A) -> Callable[..., A]:
|
||
"""常量值。"""
|
||
|
||
def constant_inner(*_args, **_kwargs) -> A:
|
||
return value
|
||
|
||
return constant_inner
|
||
|
||
|
||
def dispatch(*funcs):
|
||
"""分派应用。"""
|
||
|
||
def dispatch_inner(*args, **kwargs) -> Iterator:
|
||
return (func(*args, **kwargs) for func in funcs)
|
||
|
||
return dispatch_inner
|
||
|
||
|
||
def pipe(*funcs):
|
||
"""串联多个函数。"""
|
||
|
||
def pipe_func(*args, **k_args):
|
||
result = funcs[0](*args, **k_args)
|
||
for func in funcs[1:]:
|
||
result = func(result)
|
||
return result
|
||
|
||
return pipe_func
|
||
|
||
|
||
def identity(value: A) -> A:
|
||
"""同一。"""
|
||
return value
|
||
|
||
|
||
def invoke(func, *args, **kwargs):
|
||
"""调用。"""
|
||
return func(*args, **kwargs)
|
||
|
||
|
||
def side_effect(*funcs):
|
||
"""调用函数，产生副作用，但不影响管道结果。"""
|
||
|
||
def side_effect_func(arg):
|
||
for func in funcs:
|
||
# 不保留结果
|
||
func(arg)
|
||
return arg
|
||
|
||
return side_effect_func
|
||
|
||
|
||
def star_apply(func):
|
||
"""列表展开再应用。"""
|
||
|
||
def star_apply_func(arg):
|
||
return func(*arg)
|
||
|
||
return star_apply_func
|
||
|
||
|
||
def any_(*funcs) -> Callable:
|
||
"""高阶any。
|
||
注意，any有短路效果。"""
|
||
return pipe(
|
||
dispatch(*funcs),
|
||
any,
|
||
)
|
||
|
||
|
||
def not_(func) -> Callable:
|
||
"""高阶not。"""
|
||
return pipe(func, operator.not_)
|