Python
/
uv
mirror of https://github.com/astral-sh/uv
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Use shared client for publish tests and increase read timeout (#8538) 

Attempts to address the read timeout case in
https://github.com/astral-sh/uv/issues/8418
This commit is contained in:
Zanie Blue 2024-10-24 14:24:57 -05:00 committed by GitHub
parent f4a3f97fe0
commit f533fe8d8b
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 9 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
scripts/publish/test_publish.py
View File

@ -91,14 +91,14 @@ project_urls = {
} }
def get_new_version(project_name: str) -> str: def get_new_version(project_name: str, client: httpx.Client) -> str:
"""Return the next free path version on pypi""" """Return the next free path version on pypi"""
# To keep the number of packages small we reuse them across targets, so we have to # To keep the number of packages small we reuse them across targets, so we have to
# pick a version that doesn't exist on any target yet # pick a version that doesn't exist on any target yet
versions = set() versions = set()
for url in project_urls[project_name]: for url in project_urls[project_name]:
try: try:
data = httpx.get(url).text data = client.get(url).text
except httpx.HTTPError as err: except httpx.HTTPError as err:
raise RuntimeError(f"Failed to fetch {url}") from err raise RuntimeError(f"Failed to fetch {url}") from err
href_text = "<a[^>]+>([^<>]+)</a>" href_text = "<a[^>]+>([^<>]+)</a>"
@ -116,7 +116,7 @@ def get_new_version(project_name: str) -> str:
return ".".join(str(i) for i in release) return ".".join(str(i) for i in release)
def create_project(project_name: str, uv: Path): def create_project(project_name: str, uv: Path, client: httpx.Client):
if cwd.joinpath(project_name).exists(): if cwd.joinpath(project_name).exists():
rmtree(cwd.joinpath(project_name)) rmtree(cwd.joinpath(project_name))
check_call([uv, "init", "--lib", project_name], cwd=cwd) check_call([uv, "init", "--lib", project_name], cwd=cwd)
@ -124,18 +124,18 @@ def create_project(project_name: str, uv: Path):
# Set to an unclaimed version # Set to an unclaimed version
toml = pyproject_toml.read_text() toml = pyproject_toml.read_text()
new_version = get_new_version(project_name) new_version = get_new_version(project_name, client)
toml = re.sub('version = ".*"', f'version = "{new_version}"', toml) toml = re.sub('version = ".*"', f'version = "{new_version}"', toml)
pyproject_toml.write_text(toml) pyproject_toml.write_text(toml)
def publish_project(target: str, uv: Path): def publish_project(target: str, uv: Path, client: httpx.Client):
project_name = all_targets[target] project_name = all_targets[target]
print(f"\nPublish {project_name} for {target}") print(f"\nPublish {project_name} for {target}")
# Create the project # Create the project
create_project(project_name, uv) create_project(project_name, uv, client)
# Build the project # Build the project
check_call([uv, "build"], cwd=cwd.joinpath(project_name)) check_call([uv, "build"], cwd=cwd.joinpath(project_name))
@ -264,8 +264,9 @@ def main():
else: else:
targets = args.targets targets = args.targets
with httpx.Client(timeout=120) as client:
for project_name in targets: for project_name in targets:
publish_project(project_name, uv) publish_project(project_name, uv, client)
if __name__ == "__main__": if __name__ == "__main__":