Cool.py
Make Python code cooler. 100% coverage. Use and enjoy this code!
Install
pip install cool
Or fetch from github
pip install git+https://github.com/abersheeran/[email protected]
Usage
Pipe
Note: as fast as you didn't use F!
Use pipeline to pass data as a positional parameter to the next function.
from cool import F
assert range(10) | F(filter, lambda x: x % 2) | F(sum) == 25
Or you need to pass multiple parameters through the pipeline. Note that FF
can only accept one parameter, and it must be an iterable object.
from cool import FF
assert (1, 2) | FF(lambda x, y: x + y) == 3
You can use ...
as a placeholder. This is useful when you need to pass non-continuous parameters to create a partial function.
from functools import reduce
from cool import F
assert range(10) | F(reduce, lambda x, y: x + y) == 45
assert range(10) | F(reduce, lambda x, y: x + y, ..., 10) == 55
square = F(pow, ..., 2)
assert range(10) | F(map, square) | F(sum) == 285
The range(10) | F(reduce, lambda x, y: x + y, ..., 10)
is equivalent to reduce(lambda x, y: x + y, range(10), 10)
.
Redirect
Just like the redirection symbol in Shell
, you can redirect the output to a specified file or TextIO
object through >
or >>
.
from pathlib import PurePath
from cool import R
# Redirect output to specified filepath
R(lambda : print("hello")) > PurePath("your-filepath")
# Append mode
R(lambda : print("hello")) >> PurePath("your-filepath")
Redirect to opened file or other streams.
from io import StringIO
from cool import R
with open("filepath", "a+", encoding="utf8") as file:
R(lambda : print("hello")) >> file
out = StringIO("")
R(lambda : print("hello")) > out
out.seek(0, 0)
assert out.read() == "hello\n"
Maybe you also want to block the output, just like > /dev/null
.
from cool import R
R(lambda : print("hello")) > None
# Or
R(lambda : print("hello")) >> None
Note that after the calculation is over, R
will faithfully return the return value of your function. Try the following example.
from pathlib import PurePath
from cool import F, R
def func(num):
return range(num) | F(map, lambda x: print(x) or x) | F(sum)
result = R(lambda : func(10)) > PurePath("filepath")
assert result == 45
Set Global
Maybe you don't want to use from cool import F
in every file of the entire project, you can use the following code to set it as a global function, just like min
/max
/sum
.
import cool
cool.set_global(cool.F, cool.FF)
Maybe you also want to expose functools.reduce
to the world, just like map
/filter
.
import functools
import cool
cool.set_global(cool.F, cool.FF, functools.reduce)