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
|
import os
import re
import requests
import json
SETUP_PY = os.path.join(os.path.dirname(__file__), "..", "setup.py")
with open(SETUP_PY, "r") as f:
setup_py = f.read()
def get_setup_py_version(package_name):
version = re.findall(
rf"{package_name}(>=|~=|==)([0-9]+\.[0-9]+\.[0-9]+)",
setup_py,
)
return version[0][1]
def get_latest_version(package_name):
return json.loads(requests.get(f"https://pypi.org/pypi/{package_name}/json").text)["info"]["version"]
def check_version(package_name):
setup_py_version = get_setup_py_version(package_name)
latest_version = get_latest_version(package_name)
assert (
setup_py_version == latest_version
), f"{package_name} version in setup.py is not the latest version on PyPI ({latest_version})"
|
