11"""Dependency parsing utilities for PyPI packages."""
22
3+ import logging
34import re
4- from typing import Any , Dict , List , Optional , Set , Tuple
5+ from typing import Any
6+
57from packaging .requirements import Requirement
6- from packaging .specifiers import SpecifierSet
78from packaging .version import Version
8- import logging
99
1010logger = logging .getLogger (__name__ )
1111
1212
1313class DependencyParser :
1414 """Parser for Python package dependencies."""
15-
15+
1616 def __init__ (self ):
17- self .parsed_cache : Dict [str , List [Requirement ]] = {}
18-
19- def parse_requirements (self , requires_dist : List [str ]) -> List [Requirement ]:
17+ self .parsed_cache : dict [str , list [Requirement ]] = {}
18+
19+ def parse_requirements (self , requires_dist : list [str ]) -> list [Requirement ]:
2020 """Parse requirements from requires_dist list.
21-
21+
2222 Args:
2323 requires_dist: List of requirement strings from PyPI metadata
24-
24+
2525 Returns:
2626 List of parsed Requirement objects
2727 """
2828 requirements = []
29-
29+
3030 for req_str in requires_dist or []:
3131 if not req_str or not req_str .strip ():
3232 continue
33-
33+
3434 try :
3535 req = Requirement (req_str )
3636 requirements .append (req )
3737 except Exception as e :
3838 logger .warning (f"Failed to parse requirement '{ req_str } { e } )
3939 continue
40-
40+
4141 return requirements
42-
42+
4343 def filter_requirements_by_python_version (
44- self ,
45- requirements : List [Requirement ],
44+ self ,
45+ requirements : list [Requirement ],
4646 python_version : str
47- ) -> List [Requirement ]:
47+ ) -> list [Requirement ]:
4848 """Filter requirements based on Python version.
49-
49+
5050 Args:
5151 requirements: List of Requirement objects
5252 python_version: Target Python version (e.g., "3.10")
53-
53+
5454 Returns:
5555 Filtered list of requirements applicable to the Python version
5656 """
5757 filtered = []
58-
58+
5959 try :
6060 target_version = Version (python_version )
6161 except Exception as e :
6262 logger .warning (f"Invalid Python version '{ python_version } { e } )
6363 return requirements
64-
64+
6565 for req in requirements :
6666 if self ._is_requirement_applicable (req , target_version ):
6767 filtered .append (req )
68-
68+
6969 return filtered
70-
70+
7171 def _is_requirement_applicable (self , req : Requirement , python_version : Version ) -> bool :
7272 """Check if a requirement is applicable for the given Python version.
73-
73+
7474 Args:
7575 req: Requirement object
7676 python_version: Target Python version
77-
77+
7878 Returns:
7979 True if requirement applies to the Python version
8080 """
8181 if not req .marker :
8282 return True
83-
83+
8484 # Create environment for marker evaluation
8585 env = {
8686 'python_version' : str (python_version ),
@@ -90,22 +90,22 @@ def _is_requirement_applicable(self, req: Requirement, python_version: Version)
9090 'implementation_name' : 'cpython' ,
9191 'implementation_version' : str (python_version ),
9292 }
93-
93+
9494 try :
9595 return req .marker .evaluate (env )
9696 except Exception as e :
9797 logger .warning (f"Failed to evaluate marker for { req } { e } )
9898 return True # Include by default if evaluation fails
99-
99+
100100 def categorize_dependencies (
101- self ,
102- requirements : List [Requirement ]
103- ) -> Dict [str , List [Requirement ]]:
101+ self ,
102+ requirements : list [Requirement ]
103+ ) -> dict [str , list [Requirement ]]:
104104 """Categorize dependencies into runtime, development, and optional groups.
105-
105+
106106 Args:
107107 requirements: List of Requirement objects
108-
108+
109109 Returns:
110110 Dictionary with categorized dependencies
111111 """
@@ -115,15 +115,15 @@ def categorize_dependencies(
115115 'optional' : {},
116116 'extras' : {}
117117 }
118-
118+
119119 for req in requirements :
120120 if not req .marker :
121121 # No marker means it's a runtime dependency
122122 categories ['runtime' ].append (req )
123123 continue
124-
124+
125125 marker_str = str (req .marker )
126-
126+
127127 # Check for extra dependencies
128128 if 'extra ==' in marker_str :
129129 extra_match = re .search (r'extra\s*==\s*["\']([^"\']+)["\']' , marker_str )
@@ -133,45 +133,45 @@ def categorize_dependencies(
133133 categories ['extras' ][extra_name ] = []
134134 categories ['extras' ][extra_name ].append (req )
135135 continue
136-
136+
137137 # Check for development dependencies
138138 if any (keyword in marker_str .lower () for keyword in ['dev' , 'test' , 'lint' , 'doc' ]):
139139 categories ['development' ].append (req )
140140 else :
141141 categories ['runtime' ].append (req )
142-
142+
143143 return categories
144-
145- def extract_package_names (self , requirements : List [Requirement ]) -> Set [str ]:
144+
145+ def extract_package_names (self , requirements : list [Requirement ]) -> set [str ]:
146146 """Extract package names from requirements.
147-
147+
148148 Args:
149149 requirements: List of Requirement objects
150-
150+
151151 Returns:
152152 Set of package names
153153 """
154154 return {req .name .lower () for req in requirements }
155-
156- def get_version_constraints (self , req : Requirement ) -> Dict [str , Any ]:
155+
156+ def get_version_constraints (self , req : Requirement ) -> dict [str , Any ]:
157157 """Get version constraints from a requirement.
158-
158+
159159 Args:
160160 req: Requirement object
161-
161+
162162 Returns:
163163 Dictionary with version constraint information
164164 """
165165 if not req .specifier :
166166 return {'constraints' : [], 'allows_any' : True }
167-
167+
168168 constraints = []
169169 for spec in req .specifier :
170170 constraints .append ({
171171 'operator' : spec .operator ,
172172 'version' : str (spec .version )
173173 })
174-
174+
175175 return {
176176 'constraints' : constraints ,
177177 'allows_any' : len (constraints ) == 0 ,
0 commit comments