7b3e51f769* test poetry install * Add nightly builds workflow * remove old comments and fix poetry * remove old debug statement * [autofix.ci] apply automated fixes --------- Co-authored-by: autofix-ci[bot] <114827586+autofix-ci[bot]@users.noreply.github.com>
41 lines
1.2 KiB
Python
Executable file
41 lines
1.2 KiB
Python
Executable file
import os
|
|
import sys
|
|
import re
|
|
|
|
BASE_DIR = os.path.abspath(os.path.join(os.path.dirname(__file__), "../.."))
|
|
|
|
|
|
def update_pyproject_name(pyproject_path: str, new_project_name: str) -> None:
|
|
"""Update the project name in pyproject.toml."""
|
|
filepath = os.path.join(BASE_DIR, pyproject_path)
|
|
with open(filepath, "r") as file:
|
|
content = file.read()
|
|
|
|
# Regex to match the version line under [tool.poetry]
|
|
pattern = re.compile(r'(?<=^name = ")[^"]+(?=")', re.MULTILINE)
|
|
|
|
if not pattern.search(content):
|
|
raise Exception(f'Project name not found in "{filepath}"')
|
|
|
|
content = pattern.sub(new_project_name, content)
|
|
|
|
with open(filepath, "w") as file:
|
|
file.write(content)
|
|
|
|
|
|
def main() -> None:
|
|
if len(sys.argv) != 3:
|
|
raise Exception("Must specify project name and build type, e.g. langflow-nightly base")
|
|
new_project_name = sys.argv[1]
|
|
build_type = sys.argv[2]
|
|
|
|
if build_type == "base":
|
|
update_pyproject_name("src/backend/base/pyproject.toml", new_project_name)
|
|
elif build_type == "main":
|
|
update_pyproject_name("pyproject.toml", new_project_name)
|
|
else:
|
|
raise ValueError(f"Invalid build type: {build_type}")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|