From 092a3e7e06caed3ee11df68ff6145b0919a8ad5f Mon Sep 17 00:00:00 2001
From: Marc Abramowitz <marc@marc-abramowitz.com>
Date: Thu, 12 Jun 2014 12:57:36 -0700
Subject: [PATCH] Support Python 2.6

---
 pipdeptree.py             | 16 ++++++++--------
 setup.py                  |  7 ++++++-
 tests/pipdeptree_tests.py |  4 ++--
 tox.ini                   |  2 +-
 4 files changed, 17 insertions(+), 12 deletions(-)

diff --git a/pipdeptree.py b/pipdeptree.py
index 45d2286..9a27657 100644
--- a/pipdeptree.py
+++ b/pipdeptree.py
@@ -33,7 +33,7 @@ def top_pkg_name(pkg):
     :rtype: string
 
     """
-    return '{}=={}'.format(pkg.project_name, pkg.version)
+    return '{0}=={1}'.format(pkg.project_name, pkg.version)
 
 
 def non_top_pkg_name(req, pkg):
@@ -57,8 +57,8 @@ def non_top_pkg_name(req, pkg):
         vers.append(('installed', pkg.version))
     if not vers:
         return req.key
-    ver_str = ', '.join(['{}: {}'.format(k, v) for k, v in vers])
-    return '{} [{}]'.format(pkg.project_name, ver_str)
+    ver_str = ', '.join(['{0}: {1}'.format(k, v) for k, v in vers])
+    return '{0} [{1}]'.format(pkg.project_name, ver_str)
 
 
 def top_pkg_src(pkg):
@@ -127,8 +127,8 @@ def render_tree(pkgs, pkg_index, req_map, list_all,
     :rtype: str
 
     """
-    pkg_index = {p.key: p for p in pkgs}
-    req_map = {p: p.requires() for p in pkgs}
+    pkg_index = dict((p.key, p) for p in pkgs)
+    req_map = dict((p, p.requires()) for p in pkgs)
     non_top = set(r.key for r in flatten(req_map.values()))
     top = [p for p in pkgs if p.key not in non_top]
 
@@ -185,8 +185,8 @@ def main():
     pkgs = pip.get_installed_distributions(local_only=args.local_only,
                                            skip=skip)
 
-    pkg_index = {p.key: p for p in pkgs}
-    req_map = {p: p.requires() for p in pkgs}
+    pkg_index = dict((p.key, p) for p in pkgs)
+    req_map = dict((p, p.requires()) for p in pkgs)
 
     # show warnings about possibly confusing deps if found and
     # warnings are enabled
@@ -198,7 +198,7 @@ def main():
                 for i, (p, d) in enumerate(xs):
                     pkg = top_pkg_name(p)
                     req = non_top_pkg_name(d, pkg_index[d.key])
-                    tmpl = '  {} -> {}' if i > 0 else '* {} -> {}'
+                    tmpl = '  {0} -> {1}' if i > 0 else '* {0} -> {1}'
                     print(tmpl.format(pkg, req), file=sys.stderr)
             print('-'*72, file=sys.stderr)
 
diff --git a/setup.py b/setup.py
index 17f3a1a..949dbb3 100644
--- a/setup.py
+++ b/setup.py
@@ -1,9 +1,14 @@
+import sys
+
 from setuptools import setup
 
 
 with open('./README.rst') as f:
     long_desc = f.read()
 
+install_requires = ["pip >= 1.4.1"]
+if sys.version_info < (2, 7):
+    install_requires.append('argparse')
 
 setup(
     name='pipdeptree',
@@ -14,7 +19,7 @@ setup(
     license='MIT License',
     description='Command line utility to show dependency tree of packages',
     long_description=long_desc,
-    install_requires=["pip >= 1.4.1"],
+    install_requires=install_requires,
     py_modules=['pipdeptree'],
     entry_points={
         'console_scripts': [
diff --git a/tests/pipdeptree_tests.py b/tests/pipdeptree_tests.py
index fb592ba..6e9109d 100644
--- a/tests/pipdeptree_tests.py
+++ b/tests/pipdeptree_tests.py
@@ -9,8 +9,8 @@ with open('tests/pkgs.pickle', 'rb') as f:
     pkgs = pickle.load(f)
 
 
-pkg_index = {p.key: p for p in pkgs}
-req_map = {p: p.requires() for p in pkgs}
+pkg_index = dict([(p.key, p) for p in pkgs])  # {p.key: p for p in pkgs}
+req_map = dict([(p, p.requires()) for p in pkgs])  # {p: p.requires() for p in pkgs}
 
 
 def find_req(req, parent):
diff --git a/tox.ini b/tox.ini
index c3c1844..e4955f9 100644
--- a/tox.ini
+++ b/tox.ini
@@ -4,7 +4,7 @@
 # and then run "tox" from this directory.
 
 [tox]
-envlist = py27, py32
+envlist = py26, py27, py32
 
 [testenv]
 commands = nosetests tests/