Factor out index_url

Signed-off-by: ddelange <[email protected]>

python/packaging/frontend_sdist/setup.py

@@ -21,6 +21,7 @@
 
 tensorrt_module = "##TENSORRT_MODULE##"
 tensorrt_version = "##TENSORRT_PYTHON_VERSION##"
+index_url = "https://pypi.nvidia.com"
 
 # cherry-pick information from `packaging.markers.default_environment()` needed to find the right wheel
 # https://github.com/pypa/packaging/blob/23.1/src/packaging/markers.py#L175-L190
@@ -38,14 +39,16 @@
 if machine_marker != "x86_64":
     raise RuntimeError("TensorRT currently only builds wheels for x86_64 processors")
 
-libs_wheel = "https://pypi.nvidia.com/{}-libs/{}_libs-{}-py2.py3-none-{}_{}.whl".format(
+libs_wheel = "{}/{}-libs/{}_libs-{}-py2.py3-none-{}_{}.whl".format(
+    index_url,
     tensorrt_module,
     tensorrt_module,
     tensorrt_version,
     platform_marker,
     machine_marker,
 )
-bindings_wheel = "https://pypi.nvidia.com/{}-bindings/{}_bindings-{}-{}-none-{}_{}.whl".format(
+bindings_wheel = "{}/{}-bindings/{}_bindings-{}-{}-none-{}_{}.whl".format(
+    index_url,
     tensorrt_module,
     tensorrt_module,
     tensorrt_version,