Как проверить, можно ли повторить повторный вызов?

Question

Рассмотрим следующий код:

def my_fun(an_iterable): 
    for val in an_iterable: 
    do_work(val) 
    if some_cond(val): 
     do_some_other_work(an_iterable) 
     break 

если an_iterable является list/tuple, do_some_other_work получите весь список снова. Но если an_iterable был iterator или generator, он получит только остальные элементы в списке. Как я могу различать два случая? Я хочу, чтобы do_some_other_work получил только остальные предметы.

Answer 1

Там нет никакого общего способа узнать, можно ли перебрать объект несколько раз , Файлоподобные объекты, в частности, скорее всего, испортят проверки. К счастью, вам не нужно это проверять. Если вы просто хотите, чтобы убедиться, что do_some_other_work только получает остальную часть пунктов, вы можете явно запросить итератор:

def my_fun(iterable): 
    iterable = iter(iterable) 
    # Do whatever. 

Answer 2

Рассмотрим следующий фрагмент кода:

import types 

def which(obj): 
    if isinstance(obj, types.GeneratorType): 
     print 'Generator' 
    elif hasattr(obj, 'next') and (iter(obj) == obj): 
     print 'Iterator' 
    elif hasattr(obj, '__iter__'): 
     print 'Iterable' 
    else: 
     print 'Object' 

def my_gen(length): 
    for i in xrange(length): 
     yield i 

# Reports 'Generator' 
which(my_gen(10)) 
# Reports 'Iterable' 
which(iter(xrange(10))) 
# Reports 'Object' 
which([i for i in xrange(10)]) 

with open('foo.txt', 'w+') as fout: 
    for _ in xrange(10): 
     fout.write('hello\n') 

# Reports 'Iterable' 
with open('foo.txt') as fin: 
    which(fin) 

Это определяет их как «генератор», «Iterable», а затем «Object»