Лучший способ сделать функцию, которая может принимать список или любое количество аргументов?

Question

Как правильно реализовать функцию for_each, чтобы она могла принимать любое количество аргументов, список или кортеж в качестве параметра?

def do_something(arg):
    print("done", arg)


def for_each(func, *args):
    if len(args) == 1:  # How to do this, since this gives an
        args = args[0]  # error if there's only one parameter besides func?
    for arg in args:
        func(arg)


for_each(do_something, 1, 2)
for_each(do_something, ['foo', 'bar'])
for_each(do_something, (3, 4, 5))

Вывод:

done 1
done 2
done foo
done bar
done 3
done 4
done 5

Как правильно этого добиться?Так как это сломается, если вызывать так:

for_each(do_something, 1)

Traceback (most recent call last):
  File "main.py", line 12, in <module>
    for_each(do_something, 1)
  File "main.py", line 8, in for_each
    for arg in args:
TypeError: 'int' object is not iterable

Answer 1

Вы хотите проверить, является ли ваш первый элемент списком или кортежем, подобным этому:

(вам нужно проверить экземпляр, в случае, если пользователь просто использует одно целое, например, ваш код потерпит неудачу)

def for_each(func, *args):
  if len(args) == 1 and (isinstance(args[0], tuple) or isinstance(args[0], list)): # You can even check on Iterable by importing collections.abc.Iterable
    args = args[0]
  for arg in args:
    func(arg)

Однако вы можете пойти дальше и позволить пользователю вводить несколько Iterable, а не только кортежи или списки, например:

from collections.abc import Iterable
from itertools import chain


def do_something(arg):
    print("done", arg)

def for_each(func, *args):
  if all(map(lambda x: isinstance(x, Iterable), args)) and not any(map(lambda x: isinstance(x, str), args)):
    args = chain(*args)
  for arg in args:
    func(arg)


for_each(do_something, [0, 1], [0, 2])

Answer 2

Хорошо, теперь, когда я понимаю ваш вопрос:

не делайте этого:

if len(args) == 1:  # is there really no cleaner
    args = args[0]  # way to do this part?

Это выглядит симпатичнее, если вам не нужно возвращаться, просто хотитевызвать функцию.

def for_each(func, *args):
    for arg in args: 
        if any([isinstance(args, myiter) for myiter in [tuple, list, set]]): 
            [for_each(func, arg) for arg in args]
        else:
            func(args)



for_each(do_something, 1, 2) 
for_each(do_something, ['foo', 'bar']) 
for_each(do_something, (3, 4, 5))                                                                                                                                                              
done 1
done 2
done foo
done bar
done 3
done 4
done 5
for_each(do_something, 1)                                                                                                                                                                      
done 1