Files
2018-11-19 15:15:54 +00:00

50 lines
966 B
Python

#Unguarded calls to next()
def bad1(it):
while True:
yield next(it)
def bad2(seq):
it = iter(seq)
#Not OK as seq may be empty
raise KeyError(next(it))
yield 0
def ok1(seq):
#Not a generator
it = iter(seq)
#Not OK as seq may be empty
raise KeyError(next(it))
def ok2(seq):
if seq:
it = iter(seq)
#OK seq is non-empty so next(it) will not raise StopIteration
raise KeyError(next(it))
yield 0
def explicit_raise_stop_iter(seq):
for i in seq:
yield seq
raise StopIteration()
def ok3(seq):
it = iter(seq)
try:
yield next(iter)
except StopIteration:
return
def ok4(seq, ctx):
try:
with ctx:
yield next(iter)
except StopIteration:
return
#ODASA-6536
def next_in_comp(seq, fields):
seq_iter = iter(seq)
values = [ next(seq_iter) if f.attname in NAMES else DEFAULT for f in fields ]
return values