github/pipreqs
1
1
Fork 0
mirror of https://github.com/bndr/pipreqs.git synced 2025-06-07 12:05:33 +00:00
Code Issues 3 Packages Projects Releases Wiki Activity

fixed jupyter notebook support

Browse Source
This commit is contained in:
fernandocrz 2023-10-23 16:58:11 -03:00
parent 190ce1c894
commit 9f12601191
1 changed files with 12 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
pipreqs/pipreqs.py
View File

@ -47,7 +47,12 @@ from docopt import docopt
import requests import requests
from yarg import json2package from yarg import json2package
from yarg.exceptions import HTTPError from yarg.exceptions import HTTPError
from nbconvert import PythonExporter
try:
PythonExporter = None
from nbconvert import PythonExporter
except ImportError:
pass
from pipreqs import __version__ from pipreqs import __version__
@ -115,7 +120,10 @@ def get_all_imports(path, encoding=None, extra_ignore_dirs=None, follow_links=Tr
dirs[:] = [d for d in dirs if d not in ignore_dirs] dirs[:] = [d for d in dirs if d not in ignore_dirs]
candidates.append(os.path.basename(root)) candidates.append(os.path.basename(root))
if PythonExporter:
files = [fn for fn in files if filter_ext(fn, [".py", ".ipynb"])] files = [fn for fn in files if filter_ext(fn, [".py", ".ipynb"])]
else:
files = [fn for fn in files if filter_ext(fn, [".py"])]
candidates = list( candidates = list(
map( map(
@ -130,7 +138,7 @@ def get_all_imports(path, encoding=None, extra_ignore_dirs=None, follow_links=Tr
if filter_ext(file_name, [".py"]): if filter_ext(file_name, [".py"]):
with open(file_name, "r", encoding=encoding) as f: with open(file_name, "r", encoding=encoding) as f:
contents = f.read() contents = f.read()
elif filter_ext(file_name, [".ipynb"]): elif filter_ext(file_name, [".ipynb"]) and PythonExporter:
contents = ipynb_2_py(file_name, encoding=encoding) contents = ipynb_2_py(file_name, encoding=encoding)
try: try:
tree = ast.parse(contents) tree = ast.parse(contents)
@ -147,7 +155,7 @@ def get_all_imports(path, encoding=None, extra_ignore_dirs=None, follow_links=Tr
continue continue
else: else:
logging.error("Failed on file: %s" % file_name) logging.error("Failed on file: %s" % file_name)
if filter_ext(file_name, [".ipynb"]): if filter_ext(file_name, [".ipynb"]) and PythonExporter:
logging.error("Magic command without % might be failed") logging.error("Magic command without % might be failed")
raise exc raise exc