Skip to content

Commit

Permalink
fix: message about auto-import missing (#43)
Browse files Browse the repository at this point in the history
  • Loading branch information
caioariede committed Sep 19, 2023
1 parent 26aade3 commit 9fa8d1f
Showing 1 changed file with 15 additions and 10 deletions.
25 changes: 15 additions & 10 deletions src/dpv
Original file line number Diff line number Diff line change
Expand Up @@ -473,16 +473,21 @@ unsafe_dpv_cmd_try() {
cat <<EOF >"$dpv_python_startup"
def dpv_get_modules(packages):
import importlib.metadata
import pkg_resources
packages = [r.name for r in pkg_resources.parse_requirements(packages)]
print(f"loading packages: {', '.join(packages)}")
print("")
for m, names in importlib.metadata.packages_distributions().items():
for n in names:
if n in packages:
yield m
try:
from importlib.metadata import packages_distributions
except ImportError:
import logging
logging.warning("dpv only auto-import modules with Python 3.10+")
else:
import pkg_resources
packages = [r.name for r in pkg_resources.parse_requirements(packages)]
print(f"loading packages: {', '.join(packages)}")
print("")
for m, names in packages_distributions().items():
for n in names:
if n in packages:
yield m
print("")
print("welcome to dpv try shell! :)")
Expand Down

0 comments on commit 9fa8d1f

Please sign in to comment.