82 lines
2.9 KiB
Python
82 lines
2.9 KiB
Python
import re
|
|
from pathlib import Path
|
|
|
|
|
|
def read_version_from_pyproject(file_path):
|
|
with open(file_path, "r") as file:
|
|
for line in file:
|
|
match = re.search(r'version = "(.*)"', line)
|
|
if match:
|
|
return match.group(1)
|
|
return None
|
|
|
|
|
|
# def get_version_from_pypi(package_name):
|
|
# import requests
|
|
|
|
# response = requests.get(f"https://pypi.org/pypi/{package_name}/json")
|
|
# if response.ok:
|
|
# return response.json()["info"]["version"]
|
|
# return None
|
|
|
|
|
|
def get_version_from_pypi(package_name):
|
|
# Use default python lib to make the GET for this because it runs in github actions
|
|
import json
|
|
import urllib.request
|
|
|
|
response = urllib.request.urlopen(f"https://pypi.org/pypi/{package_name}/json")
|
|
if response.getcode() == 200:
|
|
return json.loads(response.read())["info"]["version"]
|
|
return None
|
|
|
|
|
|
def is_development_release(version):
|
|
"""
|
|
Determines if the version is a development version based on PEP 440.
|
|
|
|
We consider a development version (.devN) as our nightly versions
|
|
"""
|
|
return "dev" in version
|
|
|
|
|
|
def update_pyproject_dependency(pyproject_path, version, is_nightly):
|
|
# Strip "v" prefix from version if present
|
|
if version.startswith("v"):
|
|
version = version[1:]
|
|
|
|
pattern = re.compile(r'langflow-base = \{ path = "\./src/backend/base", develop = true \}')
|
|
if is_nightly:
|
|
# NOTE: This process can be simplified; see the note in update_lf_base_dependency.py
|
|
replacement = f'langflow-base-nightly = "{version}"'
|
|
else:
|
|
replacement = f'langflow-base = "^{version}"'
|
|
with open(pyproject_path, "r") as file:
|
|
content = file.read()
|
|
content = pattern.sub(replacement, content)
|
|
with open(pyproject_path, "w") as file:
|
|
file.write(content)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
# Backing up files
|
|
pyproject_path = Path(__file__).resolve().parent / "../pyproject.toml"
|
|
pyproject_path = pyproject_path.resolve()
|
|
with open(pyproject_path, "r") as original, open(pyproject_path.with_name("pyproject.toml.bak"), "w") as backup:
|
|
backup.write(original.read())
|
|
# Now backup poetry.lock
|
|
with open(pyproject_path.with_name("poetry.lock"), "r") as original, open(
|
|
pyproject_path.with_name("poetry.lock.bak"), "w"
|
|
) as backup:
|
|
backup.write(original.read())
|
|
|
|
# Reading version and updating pyproject.toml
|
|
langflow_base_path = Path(__file__).resolve().parent / "../src/backend/base/pyproject.toml"
|
|
version = read_version_from_pyproject(langflow_base_path)
|
|
if version:
|
|
# Nightly versions contain "dev"
|
|
# WARNING: This will cause issues if we release `.dev` versions to `langflow` or `langflow-base`
|
|
is_nightly = is_development_release(version)
|
|
update_pyproject_dependency(pyproject_path, version, is_nightly)
|
|
else:
|
|
print("Error: Version not found.")
|