mirror of
https://github.com/wassname/pip-package-list.git
synced 2026-06-27 16:10:20 +08:00
Swen Kooij
36 lines
894 B
Python
36 lines
894 B
Python
import os
|
|
|
|
from typing import Generator
|
|
|
|
import setuptools
|
|
|
|
from .requirements import (
|
|
RequirementsEntry,
|
|
RequirementsEntrySource,
|
|
parse_requirements,
|
|
)
|
|
|
|
|
|
def parse_setup_py(file_path: str) -> Generator[RequirementsEntry, None, None]:
|
|
setup_kwargs = {}
|
|
|
|
def _setup_proxy(*args, **kwargs):
|
|
setup_kwargs.update(dict(kwargs))
|
|
|
|
setuptools.setup = _setup_proxy
|
|
|
|
path = os.path.join(file_path, "setup.py")
|
|
with open(path, "r") as fp:
|
|
exec(fp.read())
|
|
|
|
source = RequirementsEntrySource(path=path, line=None, line_number=None)
|
|
|
|
requirements = setup_kwargs.get("install_requires") or []
|
|
|
|
extras_require = setup_kwargs.get("extras_require") or {}
|
|
for _, extra_requirements in extras_require.items():
|
|
requirements.extend(extra_requirements)
|
|
|
|
for requirement in parse_requirements(source, requirements):
|
|
yield requirement
|