Python 可以包含0个或多个赋值的函数头

我想创建一个可以有0个或多个赋值的函数头： 例如：

function(1, 2, 3)
returns: [1, 2, 3]

function(4, 5, 6, 7, 8)
returns:[4, 5, 6, 7, 8]

function(1, 2)
returns: [1,2]

它需要能够采取尽可能多的任务作为用户输入，有人能帮我吗

>>> def foo(*args):
...    return list(args)
... 
>>> foo(1, 2, 3)
[1, 2, 3]