···

       
5
       
5
       
 
       
import os

     

       
6
       
6
       
 
       
from typing import Dict, List, Set, Tuple

     

       
7
       
7
       
 
       
do_abort: bool = False

     

       
8
       
8
       
-
       
packages: Dict[str, Dict[str, List[Dict[str, List[str]]]]] = collections.defaultdict(list)

     

       
8
       
8
       
+
       
packages: Dict[str, Dict[str, Dict[str, List[str]]]] = collections.defaultdict(dict)

     

       
9
       
9
       
 
       
found_paths: Set[Path] = set()

     

       
10
       
10
       
 
       
out_path: Path = Path(os.getenv("out"))

     

       
11
       
11
       
 
       
version: Tuple[int, int] = sys.version_info

     
···

       
32
       
32
       
 
       


     

       
33
       
33
       
 
       
# inserts an entry into 'packages'

     

       
34
       
34
       
 
       
def add_entry(name: str, version: str, store_path: str, parents: List[str]) -> None:

     

       
35
       
35
       
-
       
    if name not in packages:

     

       
36
       
36
       
-
       
        packages[name] = {}

     

       
37
       
37
       
-
       
    if store_path not in packages[name]:

     

       
38
       
38
       
-
       
        packages[name][store_path] = []

     

       
39
       
39
       
-
       
    packages[name][store_path].append(dict(

     

       
35
       
35
       
+
       
    packages[name][store_path] = dict(

     

       
40
       
36
       
 
       
        version=version,

     

       
41
       
37
       
 
       
        parents=parents,

     

       
42
       
42
       
-
       
    ))

     

       
38
       
38
       
+
       
    )

     

       
43
       
39
       
 
       


     

       
44
       
40
       
 
       


     

       
45
       
41
       
 
       
# transitively discover python dependencies and store them in 'packages'

     
···

       
64
       
60
       
 
       
        with open(propagated_build_inputs, "r") as f:

     

       
65
       
61
       
 
       
            build_inputs: List[str] = f.read().strip().split(" ")

     

       
66
       
62
       
 
       
            for build_input in build_inputs:

     

       
67
       
67
       
-
       
                if build_input not in parents:

     

       
68
       
68
       
-
       
                    find_packages(Path(build_input), site_packages_path, parents + [build_input])

     

       
63
       
63
       
+
       
                find_packages(Path(build_input), site_packages_path, parents + [build_input])

     

       
69
       
64
       
 
       


     

       
70
       
65
       
 
       


     

       
71
       
66
       
 
       
find_packages(out_path, site_packages_path, [f"this derivation: {out_path}"])

     
···

       
75
       
70
       
 
       
    if len(store_paths) > 1:

     

       
76
       
71
       
 
       
        do_abort = True

     

       
77
       
72
       
 
       
        print("Found duplicated packages in closure for dependency '{}': ".format(name))

     

       
78
       
78
       
-
       
        for store_path, candidates in store_paths.items():

     

       
79
       
79
       
-
       
            for candidate in candidates:

     

       
80
       
80
       
-
       
                print(f"  {name} {candidate['version']} ({store_path})")

     

       
81
       
81
       
-
       
                print(describe_parents(candidate['parents']))

     

       
73
       
73
       
+
       
        for store_path, candidate in store_paths.items():

     

       
74
       
74
       
+
       
            print(f"  {name} {candidate['version']} ({store_path})")

     

       
75
       
75
       
+
       
            print(describe_parents(candidate['parents']))

     

       
82
       
76
       
 
       


     

       
83
       
77
       
 
       
# fail if duplicates were found

     

       
84
       
78
       
 
       
if do_abort: