1
0
Fork 0
mirror of https://github.com/veekun/pokedex.git synced 2024-08-20 18:16:34 +00:00

Exit the generator for Python 3.7 and up

This commit is contained in:
Kip Yin 2019-02-04 17:19:31 +08:00 committed by GitHub
parent efd05a4754
commit feb6788b9f
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23

View file

@ -28,6 +28,7 @@ import csv
import io import io
import os import os
import re import re
import sys
from collections import defaultdict from collections import defaultdict
import six import six
@ -377,7 +378,13 @@ def group_by_object(stream):
Yields ((class name, object ID), (list of messages)) pairs. Yields ((class name, object ID), (list of messages)) pairs.
""" """
stream = iter(stream) stream = iter(stream)
current = next(stream) try:
current = next(stream)
except StopIteration:
if sys.version_info >= (3, 7):
return
else:
raise StopIteration
current_key = current.cls, current.id current_key = current.cls, current.id
group = [current] group = [current]
for message in stream: for message in stream: