Support indents and spaces in imports

ivanlen · Sep 20, 2020 · 52003ca · 52003ca
1 parent 2d814be
commit 52003ca
Show file tree

Hide file tree

Showing 8 changed files with 5 additions and 1 deletion.
diff --git a/.gitignore b/.gitignore
@@ -4,3 +4,4 @@
 /dist/
 /*.egg-info
 tests/
+.ipynb_checkpoints/
diff --git a/LICENSE.txt b/LICENSE.txt
diff --git a/README.md b/README.md
diff --git a/pipreqsnb/__init__.py b/pipreqsnb/__init__.py
diff --git a/pipreqsnb/pipreqsnb.py b/pipreqsnb/pipreqsnb.py
@@ -121,6 +121,9 @@ def main():
                 source = ''.join(valid_lines)
                 imports += get_import_string_from_source(source)
 
+    # hack to remove the indents if imports are inside functions
+    imports = [i.lstrip() for i in imports]
+
     if is_file:
         args.savepath = set_requirements_savepath(args)
         args.path = temp_path

diff --git a/requirements.txt b/requirements.txt
diff --git a/setup.cfg b/setup.cfg
diff --git a/setup.py b/setup.py
@@ -5,7 +5,7 @@
 
 setuptools.setup(
     name='pipreqsnb',
-    version='0.2.1',
+    version='0.2.2',
     description='Pipreqs with jupyter notebook support',
     url='https://github.com/ivanlen/pipreqsnb',
     author='Ivan Lengyel',
-Original file line number
+Diff line change
@@ Expand Up / @@ -4,3 +4,4 @@ @@
     /dist/
     /*.egg-info
     tests/
+    .ipynb_checkpoints/