소스 검색

Import zip_longest via six.moves for py2 and py3 compatibility

Jari Karppinen 8 년 전
부모
커밋
9d37448ba5
1개의 변경된 파일5개의 추가작업 그리고 2개의 파일을 삭제
  1. 5 2
      filetime_from_git/git_wrapper.py

+ 5 - 2
filetime_from_git/git_wrapper.py

@@ -2,7 +2,10 @@
 """
 Wrap python git interface for compatibility with older/newer version
 """
-import itertools
+try:
+    from itertools import zip_longest
+except ImportError:
+    from six.moves import zip_longest
 import logging
 import os
 from time import mktime
@@ -19,7 +22,7 @@ def grouper(iterable, n, fillvalue=None):
     '''
     # grouper('ABCDEFG', 3, 'x') --> ABC DEF Gxx
     args = [iter(iterable)] * n
-    return itertools.izip_longest(fillvalue=fillvalue, *args)
+    return zip_longest(fillvalue=fillvalue, *args)
 
 
 class _GitWrapperCommon(object):