From b0423cdfe68172c7c4689c70606d9cfa413bd631 Mon Sep 17 00:00:00 2001 From: "aalok.sathe" Date: Tue, 12 Feb 2019 10:36:27 -0500 Subject: [PATCH] make it work with python3 --- pipreqs/pipreqs.py | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) diff --git a/pipreqs/pipreqs.py b/pipreqs/pipreqs.py index 4e7b39f..38032fb 100755 --- a/pipreqs/pipreqs.py +++ b/pipreqs/pipreqs.py @@ -1,4 +1,4 @@ -#!/usr/bin/env python +#!/usr/bin/env python3 # -*- coding: utf-8 -*- """pipreqs - Generate pip requirements.txt file based on imports @@ -32,7 +32,7 @@ Options: --clean Clean up requirements.txt by removing modules that are not imported in project. """ -from __future__ import print_function, absolute_import +# from __future__ import print_function, absolute_import from contextlib import contextmanager import os import sys @@ -46,7 +46,10 @@ import requests from yarg import json2package from yarg.exceptions import HTTPError -from pipreqs import __version__ +try: + from pipreqs import __version__ +except ImportError: + from __init__ import __version__ REGEXP = [ re.compile(r'^import (.+)$'),