77from rich .console import Console
88
99
10- def print_brick_imports ( brick_imports : dict ) -> None :
10+ def _print_imports ( bricks : dict , brick_type : str ) -> None :
1111 console = Console (theme = theme .poly_theme )
1212
13- bases = brick_imports [ "bases" ]
14- components = brick_imports [ "components" ]
13+ items = sorted ( bricks . items ())
14+ tag = "base" if brick_type == "bases" else "comp"
1515
16- bricks = {** bases , ** components }
16+ for item in items :
17+ key , values = item
1718
18- for key , values in bricks .items ():
1919 imports_in_brick = values .difference ({key })
2020
2121 if not imports_in_brick :
2222 continue
2323
24- joined = ", " .join (imports_in_brick )
25- message = f":information: [data ]{ key } [/] is importing [data]{ joined } [/]"
24+ joined = ", " .join (sorted ( imports_in_brick ) )
25+ message = f":information: [{ tag } ]{ key } [/] is importing [data]{ joined } [/]"
2626 console .print (message )
2727
2828
29+ def print_brick_imports (brick_imports : dict ) -> None :
30+ bases = brick_imports ["bases" ]
31+ components = brick_imports ["components" ]
32+
33+ _print_imports (bases , "bases" )
34+ _print_imports (components , "components" )
35+
36+
2937def print_missing_deps (diff : Set [str ], project_name : str ) -> None :
3038 if not diff :
3139 return
@@ -37,12 +45,6 @@ def print_missing_deps(diff: Set[str], project_name: str) -> None:
3745 console .print (f":thinking_face: Cannot locate { missing } in { project_name } " )
3846
3947
40- def fetch_brick_imports (root : Path , ns : str , all_imports : dict ) -> dict :
41- extracted = grouping .extract_brick_imports (all_imports , ns )
42-
43- return collect .with_unknown_components (root , ns , extracted )
44-
45-
4648def collect_all_imports (root : Path , ns : str , project_data : dict ) -> dict :
4749 bases = {b for b in project_data .get ("bases" , [])}
4850 components = {c for c in project_data .get ("components" , [])}
@@ -54,8 +56,8 @@ def collect_all_imports(root: Path, ns: str, project_data: dict) -> dict:
5456 all_imports_in_components = imports .fetch_all_imports (components_paths )
5557
5658 brick_imports = {
57- "bases" : fetch_brick_imports ( root , ns , all_imports_in_bases ),
58- "components" : fetch_brick_imports ( root , ns , all_imports_in_components ),
59+ "bases" : grouping . extract_brick_imports ( all_imports_in_bases , ns ),
60+ "components" : grouping . extract_brick_imports ( all_imports_in_components , ns ),
5961 }
6062
6163 third_party_imports = {
0 commit comments