1818# https://docs.python.org/3/library/platform.html#platform.architecture
1919bits = 8 * (8 if sys .maxsize > 2 ** 32 else 4 )
2020
21+ _POLICY_JSON_MAP = {
22+ Libc .GLIBC : _HERE / "manylinux-policy.json" ,
23+ Libc .MUSL : _HERE / "musllinux-policy.json" ,
24+ }
25+
26+
27+ class WheelPolicies :
28+ @staticmethod
29+ def load ():
30+ libc_variant = get_libc ()
31+ policies_path = _POLICY_JSON_MAP [libc_variant ]
32+ policy_dict = json .loads (policies_path .read_text ())
33+ return WheelPolicies (policy_dict )
34+
35+ def __init__ (self , policies : dict ) -> None :
36+ self ._policies = []
37+ self ._musl_policy = _get_musl_policy ()
38+ self ._arch_name = get_arch_name ()
39+ self ._libc_variant = get_libc ()
40+
41+ _validate_pep600_compliance (policies )
42+ for policy in policies :
43+ if self ._musl_policy is not None and policy ["name" ] not in
44+ "linux" ,
45+ self ._musl_policy ,
46+ }:
47+ continue
48+ if (
49+ self ._arch_name in policy ["symbol_versions" ].keys ()
50+ or policy ["name" ] == "linux"
51+ ):
52+ if policy ["name" ] != "linux" :
53+ policy ["symbol_versions" ] = policy ["symbol_versions" ][
54+ self ._arch_name
55+ ]
56+ policy ["name" ] = policy ["name" ] + "_" + self ._arch_name
57+ policy ["aliases" ] = [
58+ alias + "_" + self ._arch_name for alias in policy ["aliases" ]
59+ ]
60+ policy ["lib_whitelist" ] = _fixup_musl_libc_soname (
61+ policy ["lib_whitelist" ]
62+ )
63+ self ._policies .append (policy )
64+
65+ if self ._libc_variant == Libc .MUSL :
66+ assert len (self ._policies ) == 2 , self ._policies
67+
68+ @property
69+ def policies (self ):
70+ return self ._policies
71+
72+ @property
73+ def priority_highest (self ):
74+ return max (p ["priority" ] for p in self ._policies )
75+
76+ @property
77+ def priority_lowest (self ):
78+ return min (p ["priority" ] for p in self ._policies )
79+
80+ def get_policy_by_name (self , name : str ) -> dict | None :
81+ matches = [
82+ p for p in self ._policies if p ["name" ] == name or name in p ["aliases" ]
83+ ]
84+ if len (matches ) == 0 :
85+ return None
86+ if len (matches ) > 1 :
87+ raise RuntimeError ("Internal error. Policies should be unique" )
88+ return matches [0 ]
89+
90+ def get_policy_name (self , priority : int ) -> str | None :
91+ matches = [p ["name" ] for p in self ._policies if p ["priority" ] == priority ]
92+ if len (matches ) == 0 :
93+ return None
94+ if len (matches ) > 1 :
95+ raise RuntimeError ("Internal error. priorities should be unique" )
96+ return matches [0 ]
97+
98+ def get_priority_by_name (self , name : str ) -> int | None :
99+ policy = self .get_policy_by_name (name )
100+ return None if policy is None else policy ["priority" ]
101+
21102
22103def get_arch_name () -> str :
23104 machine = _platform_module .machine ()
@@ -65,22 +146,13 @@ def _validate_pep600_compliance(policies) -> None:
65146 symbol_versions [arch ] = symbol_versions_arch
66147
67148
68- _POLICY_JSON_MAP = {
69- Libc .GLIBC : _HERE / "manylinux-policy.json" ,
70- Libc .MUSL : _HERE / "musllinux-policy.json" ,
71- }
72-
73-
74149def _get_musl_policy ():
75150 if _LIBC != Libc .MUSL :
76151 return None
77152 musl_version = get_musl_version (find_musl_libc ())
78153 return f"musllinux_{ musl_version .major } { musl_version .minor }
79154
80155
81- _MUSL_POLICY = _get_musl_policy ()
82-
83-
84156def _fixup_musl_libc_soname (whitelist ):
85157 if _LIBC != Libc .MUSL :
86158 return whitelist
@@ -105,60 +177,6 @@ def _fixup_musl_libc_soname(whitelist):
105177 return new_whitelist
106178
107179
108- with _POLICY_JSON_MAP [_LIBC ].open () as f :
109- _POLICIES = []
110- _policies_temp = json .load (f )
111- _validate_pep600_compliance (_policies_temp )
112- for _p in _policies_temp :
113- if _MUSL_POLICY is not None and _p ["name" ] not in "linux" , _MUSL_POLICY }:
114- continue
115- if _ARCH_NAME in _p ["symbol_versions" ].keys () or _p ["name" ] == "linux" :
116- if _p ["name" ] != "linux" :
117- _p ["symbol_versions" ] = _p ["symbol_versions" ][_ARCH_NAME ]
118- _p ["name" ] = _p ["name" ] + "_" + _ARCH_NAME
119- _p ["aliases" ] = [alias + "_" + _ARCH_NAME for alias in _p ["aliases" ]]
120- _p ["lib_whitelist" ] = _fixup_musl_libc_soname (_p ["lib_whitelist" ])
121- _POLICIES .append (_p )
122- if _LIBC == Libc .MUSL :
123- assert len (_POLICIES ) == 2 , _POLICIES
124-
125- POLICY_PRIORITY_HIGHEST = max (p ["priority" ] for p in _POLICIES )
126- POLICY_PRIORITY_LOWEST = min (p ["priority" ] for p in _POLICIES )
127-
128-
129- def load_policies ():
130- return _POLICIES
131-
132-
133- def _load_policy_schema ():
134- with open (join (dirname (abspath (__file__ )), "policy-schema.json" )) as f_ :
135- schema = json .load (f_ )
136- return schema
137-
138-
139- def get_policy_by_name (name : str ) -> dict | None :
140- matches = [p for p in _POLICIES if p ["name" ] == name or name in p ["aliases" ]]
141- if len (matches ) == 0 :
142- return None
143- if len (matches ) > 1 :
144- raise RuntimeError ("Internal error. Policies should be unique" )
145- return matches [0 ]
146-
147-
148- def get_policy_name (priority : int ) -> str | None :
149- matches = [p ["name" ] for p in _POLICIES if p ["priority" ] == priority ]
150- if len (matches ) == 0 :
151- return None
152- if len (matches ) > 1 :
153- raise RuntimeError ("Internal error. priorities should be unique" )
154- return matches [0 ]
155-
156-
157- def get_priority_by_name (name : str ) -> int | None :
158- policy = get_policy_by_name (name )
159- return None if policy is None else policy ["priority" ]
160-
161-
162180def get_replace_platforms (name : str ) -> list [str ]:
163181 """Extract platform tag replacement rules from policy
164182
@@ -185,10 +203,14 @@ def get_replace_platforms(name: str) -> list[str]:
185203from .external_references import lddtree_external_references # noqa
186204from .versioned_symbols import versioned_symbols_policy # noqa
187205
206+ def _load_policy_schema ():
207+ with open (join (dirname (abspath (__file__ )), "policy-schema.json" )) as f_ :
208+ schema = json .load (f_ )
209+ return schema
210+
211+
188212__all__ = [
189213 "lddtree_external_references" ,
190214 "versioned_symbols_policy" ,
191- "load_policies" ,
192- "POLICY_PRIORITY_HIGHEST" ,
193- "POLICY_PRIORITY_LOWEST" ,
215+ "WheelPolicies" ,
194216]
0 commit comments