1717
1818class PoetryGroup (BaseModel ):
1919 name : str
20- toml_section : str
20+ toml_section : Optional [ str ]
2121
2222 class Config :
2323 frozen = True
2424
2525
26+ TRANSITIVE_GROUP = PoetryGroup (name = "transitive" , toml_section = None )
27+
28+
2629class PoetryDependency (Package ):
2730 name : str
2831 version : str
29- group : PoetryGroup | None
32+ group : PoetryGroup
3033
3134
3235class PoetryToml (BaseModel ):
@@ -45,12 +48,12 @@ def read_content(cls, values):
4548 raise ValueError (f"Error reading file: { str (e )} " )
4649 return values
4750
48- def get_section_dict (self , section : str ) -> dict | None :
51+ def get_section_dict (self , section : str ) -> Optional [ dict ] :
4952 current = self ._content .copy ()
5053 for section in section .split ("." ):
5154 if section not in current :
5255 return None
53- current = current [section ]
56+ current = current [section ] # type: ignore
5457 return current
5558
5659 @property
@@ -83,13 +86,13 @@ class PoetryDependencies(BaseModel):
8386 working_directory : Path
8487
8588 @staticmethod
86- def _extract_from_line (line : str , group : PoetryGroup | None ) -> PoetryDependency :
89+ def _extract_from_line (line : str , group : PoetryGroup ) -> PoetryDependency :
8790 pattern = r"\s+(\d+(?:\.\d+)*)\s+"
8891 match = re .split (pattern , line )
8992 return PoetryDependency (name = match [0 ], version = match [1 ], group = group )
9093
9194 def _extract_from_poetry_show (
92- self , output_text : str , group : PoetryGroup | None
95+ self , output_text : str , group : PoetryGroup
9396 ) -> list [PoetryDependency ]:
9497 return [
9598 self ._extract_from_line (line , group = group )
@@ -125,8 +128,8 @@ def all_dependencies(self) -> dict[str, list[PoetryDependency]]:
125128 for dep in group_list
126129 }
127130 for line in output .stdout .splitlines ():
128- dep = self ._extract_from_line (line = line , group = None )
131+ dep = self ._extract_from_line (line = line , group = TRANSITIVE_GROUP )
129132 if dep .name not in names_direct_dependencies :
130133 transitive_dependencies .append (dep )
131134
132- return direct_dependencies | {"transitive" : transitive_dependencies }
135+ return direct_dependencies | {TRANSITIVE_GROUP . name : transitive_dependencies }
0 commit comments