1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53
|
"""
Updates all the libraries in ``sunpy/extern``
"""
import json
import urllib.request
from pathlib import Path
SUNPY_EXTERN_DIR = Path(__file__).parent.parent / "sunpy" / "extern"
# "filename": ["user", "repository", "path_to_file"]
PACKAGES = {
"appdirs.py": ["ActiveState", "appdirs", "appdirs.py"],
"appdirs_license.txt": ["ActiveState", "appdirs", "LICENSE.txt"],
"distro.py": ["python-distro", "distro", "src/distro/distro.py"],
"distro_license.rst": ["python-distro", "distro", "LICENSE"],
# Inflect is not here to avoid a dependency on pydantic.
# Since it only does one specific thing, we should be able to pin the version permanently.
"parse.py": ["r1chardj0n3s", "parse", "parse.py"],
"parse_license.txt": ["r1chardj0n3s", "parse", "LICENSE"],
}
def download_github_file(user: str, repo: str, src: Path, dest: Path):
"""
Download a file from Github.
"""
try:
response = urllib.request.urlopen(f"https://api.github.com/repos/{user}/{repo}")
except urllib.error.HTTPError as e:
print(f"Error: {e}")
return
url = f"https://api.github.com/repos/{user}/{repo}/tags"
response = urllib.request.urlopen(url)
if response.status != 200:
raise ValueError(f"tags for {user}/{repo} does not exist.")
response = json.load(response)
version = response[0]["name"]
url = f"https://raw.githubusercontent.com/{user}/{repo}/{version}/{src}"
response = urllib.request.urlopen(url)
if response.status != 200:
raise ValueError(f"{url} does not exist.")
with open(dest, "wb") as f:
print(f"Updating {dest} to {user}/{repo}:refs/tags/{version}")
f.write(response.read())
if __name__ == "__main__":
for package, (user, repo, src) in PACKAGES.items():
download_github_file(user, repo, src, SUNPY_EXTERN_DIR / package)
|