justpip 1.0.0__tar.gz

justpip-1.0.0/PKG-INFO

@@ -0,0 +1,5 @@
+Metadata-Version: 2.4
+Name: justpip
+Version: 1.0.0
+Summary: Detects missing/misspelled imports, suggests the correct PyPI package, and lets you search all packages
+Requires-Python: >=3.8

justpip-1.0.0/justpip/__init__.py

@@ -0,0 +1,3 @@
+from .installer import activate, search, package_exists
+
+activate()

justpip-1.0.0/justpip/installer.py

@@ -0,0 +1,208 @@
+import sys
+import subprocess
+import difflib
+import re
+import os
+import json
+import time
+import urllib.request
+
+PYPI_SIMPLE_INDEX = "https://pypi.org/simple/"
+CACHE_FILE = os.path.join(os.path.expanduser("~"), ".justpip_pypi_cache.json")
+CACHE_MAX_AGE_DAYS = 7
+
+# Cases where the import name is genuinely different from the pip install name.
+IMPORT_TO_PIP_NAME = {
+    "cv2": "opencv-python",
+    "PIL": "Pillow",
+    "sklearn": "scikit-learn",
+    "bs4": "beautifulsoup4",
+    "yaml": "pyyaml",
+    "Crypto": "pycryptodome",
+    "dotenv": "python-dotenv",
+    "serial": "pyserial",
+    "docx": "python-docx",
+    "pptx": "python-pptx",
+    "fitz": "pymupdf",
+}
+
+
+# ---------- Fetching / caching the full PyPI package list ----------
+
+def _download_all_package_names():
+    with urllib.request.urlopen(PYPI_SIMPLE_INDEX, timeout=30) as resp:
+        html = resp.read().decode("utf-8")
+    names = re.findall(r'<a[^>]*>([^<]+)</a>', html)
+    return sorted(set(names))
+
+
+def _load_cache():
+    if os.path.exists(CACHE_FILE):
+        try:
+            with open(CACHE_FILE, "r", encoding="utf-8") as f:
+                data = json.load(f)
+            age_days = (time.time() - data.get("timestamp", 0)) / 86400
+            if age_days < CACHE_MAX_AGE_DAYS:
+                return data["names"]
+        except Exception:
+            pass
+    return None
+
+
+def _save_cache(names):
+    try:
+        with open(CACHE_FILE, "w", encoding="utf-8") as f:
+            json.dump({"timestamp": time.time(), "names": names}, f)
+    except Exception:
+        pass
+
+
+def get_all_package_names(force_refresh=False):
+    """Returns every package name on PyPI, using a 7-day local cache."""
+    if not force_refresh:
+        cached = _load_cache()
+        if cached:
+            return cached
+
+    print("Fetching the full list of PyPI packages (first time only, ~10-20 seconds)...")
+    try:
+        names = _download_all_package_names()
+        _save_cache(names)
+        return names
+    except Exception as e:
+        print(f"Could not fetch the package list ({e}). Check your internet connection.")
+        return []
+
+
+# ---------- Search feature ----------
+
+def search(prefix=None):
+    """Search PyPI package names by prefix, e.g. search('s') or search('numpy')."""
+    if prefix is None:
+        prefix = input("Enter a letter or prefix to search (e.g. 's', 'numpy'): ").strip()
+
+    names = get_all_package_names()
+    if not names:
+        return []
+
+    prefix_lower = prefix.lower()
+    matches = [n for n in names if n.lower().startswith(prefix_lower)]
+
+    print(f"\nFound {len(matches)} package(s) starting with '{prefix}'.")
+
+    shown = 0
+    while shown < len(matches):
+        n_input = input(
+            f"Enter number of results to see (remaining: {len(matches) - shown}, or 'q' to stop): "
+        ).strip()
+
+        if n_input.lower() == "q":
+            break
+
+        if not n_input.isdigit() or int(n_input) <= 0:
+            print("Please enter a positive number.")
+            continue
+
+        count = int(n_input)
+        end = min(shown + count, len(matches))
+        for name in matches[shown:end]:
+            print(" -", name)
+        shown = end
+        print(f"(Shown {shown} of {len(matches)})")
+
+    return matches
+
+
+def package_exists(name):
+    """Checks if an exact package name exists on PyPI."""
+    names = get_all_package_names()
+    return name.lower() in [n.lower() for n in names]
+
+
+# ---------- Suggestion logic ----------
+
+def suggest_pip_name(missing_module):
+    """Given a missing import name, guess the correct pip package name."""
+    if missing_module in IMPORT_TO_PIP_NAME:
+        return IMPORT_TO_PIP_NAME[missing_module], "known name mismatch"
+
+    all_names = get_all_package_names()
+    if not all_names:
+        return None, None
+
+    lower_names = [n.lower() for n in all_names]
+
+    if missing_module.lower() in lower_names:
+        idx = lower_names.index(missing_module.lower())
+        return all_names[idx], "exact match"
+
+    same_letter = [n for n in all_names if n and n[0].lower() == missing_module[0].lower()]
+    candidates = same_letter if same_letter else all_names
+
+    matches = difflib.get_close_matches(
+        missing_module.lower(),
+        [c.lower() for c in candidates],
+        n=1,
+        cutoff=0.6,
+    )
+    if matches:
+        for c in candidates:
+            if c.lower() == matches[0]:
+                return c, "possible typo"
+
+    return None, None
+
+
+# ---------- Install + crash handling ----------
+
+def install_package(pip_name):
+    print(f"\nInstalling '{pip_name}' ...\n")
+    result = subprocess.run(
+        [sys.executable, "-m", "pip", "install", pip_name],
+        capture_output=False,
+    )
+    if result.returncode == 0:
+        print(f"\n'{pip_name}' installed successfully. Please run your program again.\n")
+    else:
+        print(f"\nSomething went wrong installing '{pip_name}'. Check the error above.\n")
+
+
+def extract_missing_module(exc_value):
+    msg = str(exc_value)
+    match = re.search(r"No module named '([^']+)'", msg)
+    if match:
+        return match.group(1).split(".")[0]
+    return None
+
+
+def custom_excepthook(exc_type, exc_value, exc_tb):
+    if exc_type is ModuleNotFoundError:
+        missing = extract_missing_module(exc_value)
+        if missing:
+            suggestion, reason = suggest_pip_name(missing)
+            if suggestion:
+                print(f"\nIt looks like '{missing}' is not installed.")
+                if reason == "possible typo":
+                    print(f"Did you mean the package '{suggestion}'?")
+                elif reason == "known name mismatch":
+                    print(f"The correct package to install is '{suggestion}'.")
+                else:
+                    print(f"'{suggestion}' exists on PyPI.")
+
+                answer = input(f"Install '{suggestion}' now? (y/n): ").strip().lower()
+                if answer == "y":
+                    install_package(suggestion)
+                    return
+                else:
+                    print("Okay, not installing.")
+                    return
+            else:
+                print(f"\n'{missing}' was not found on PyPI, even as a close match.")
+                print("Tip: run justpip.search('...') to look it up manually.")
+                return
+
+    sys.__excepthook__(exc_type, exc_value, exc_tb)
+
+
+def activate():
+    sys.excepthook = custom_excepthook

justpip-1.0.0/justpip.egg-info/PKG-INFO

@@ -0,0 +1,5 @@
+Metadata-Version: 2.4
+Name: justpip
+Version: 1.0.0
+Summary: Detects missing/misspelled imports, suggests the correct PyPI package, and lets you search all packages
+Requires-Python: >=3.8

justpip-1.0.0/justpip.egg-info/SOURCES.txt

@@ -0,0 +1,7 @@
+pyproject.toml
+justpip/__init__.py
+justpip/installer.py
+justpip.egg-info/PKG-INFO
+justpip.egg-info/SOURCES.txt
+justpip.egg-info/dependency_links.txt
+justpip.egg-info/top_level.txt

justpip-1.0.0/justpip.egg-info/dependency_links.txt

@@ -0,0 +1 @@
+

justpip-1.0.0/justpip.egg-info/top_level.txt

@@ -0,0 +1,2 @@
+dist
+justpip

justpip-1.0.0/pyproject.toml

@@ -0,0 +1,12 @@
+[build-system]
+requires = ["setuptools>=61.0"]
+build-backend = "setuptools.build_meta"
+
+[project]
+name = "justpip"
+version = "1.0.0"
+description = "Detects missing/misspelled imports, suggests the correct PyPI package, and lets you search all packages"
+requires-python = ">=3.8"
+
+[tool.setuptools.packages.find]
+where = ["."]

justpip-1.0.0/setup.cfg

@@ -0,0 +1,4 @@
+[egg_info]
+tag_build = 
+tag_date = 0
+