From b3b79af14cd76728740bce442efb468ff83df6dc Mon Sep 17 00:00:00 2001
From: Zach Smith <zach.smith@makespace.com>
Date: Thu, 19 Nov 2015 22:46:26 -0500
Subject: Factor out highlight into more functional style

---
 pycco/compat.py | 7 +++++++
 1 file changed, 7 insertions(+)

(limited to 'pycco/compat.py')

diff --git a/pycco/compat.py b/pycco/compat.py
index 6660531..37cb345 100644
--- a/pycco/compat.py
+++ b/pycco/compat.py
@@ -2,3 +2,10 @@ try:
     pycco_unichr = unichr
 except NameError:
     pycco_unichr = chr
+
+import itertools
+try:
+    pycco_zip_longest = itertools.zip_longest
+except AttributeError:
+    import itertools
+    pycco_zip_longest = itertools.izip_longest
-- 
cgit v1.2.1