|
- # Copyright (c) 2022 PaddlePaddle Authors. All Rights Reserved.
- #
- # Licensed under the Apache License, Version 2.0 (the "License");
- # you may not use this file except in compliance with the License.
- # You may obtain a copy of the License at
- #
- # http://www.apache.org/licenses/LICENSE-2.0
- #
- # Unless required by applicable law or agreed to in writing, software
- # distributed under the License is distributed on an "AS IS" BASIS,
- # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
- # See the License for the specific language governing permissions and
- # limitations under the License.
- from __future__ import annotations
-
- import argparse
- import os
- import subprocess
- import sys
-
- from pkg_resources import parse_version
-
-
- def read_version_of_remote_package(name: str) -> str:
- """get version of remote package,
-
- apdapted from: https://stackoverflow.com/a/58649262/6894382
-
- Args:
- name (str): the name of package
-
- Returns:
- str: the version of package
- """
- latest_version = str(
- subprocess.run(
- [sys.executable, "-m", "pip", "install", "{}==random".format(name)], capture_output=True, text=True
- )
- )
- latest_version = latest_version[latest_version.find("(from versions:") + 15 :]
- latest_version = latest_version[: latest_version.find(")")]
- latest_version = latest_version.replace(" ", "").split(",")[-1]
- return latest_version
-
-
- def read_version_of_local_package(version_file_path: str) -> str:
- """get version of local package
-
- Args:
- version_file_path (str): the path of `VERSION` file
-
- Returns:
- str: the version of local package
- """
- with open(version_file_path, "r", encoding="utf-8") as f:
- version = f.read().strip()
- return version
-
-
- if __name__ == "__main__":
- parser = argparse.ArgumentParser()
- parser.add_argument("--name", required=True)
-
- args = parser.parse_args()
-
- version_file_map = {
- "ppdiffusers": "ppdiffusers/VERSION",
- "paddle-pipelines": "pipelines/VERSION",
- }
- remote_version = read_version_of_remote_package(args.name)
-
- if args.name == "paddlenlp":
- local_version = str(subprocess.check_output(["python", "setup.py", "--version"], text=True))
- elif args.name in version_file_map:
- PROJECT_ROOT = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
- local_version_file = os.path.join(PROJECT_ROOT, version_file_map[args.name])
- local_version = read_version_of_local_package(local_version_file)
- else:
- raise ValueError(f"package<{args.name}> not supported")
-
- should_deploy = str(parse_version(remote_version) < parse_version(local_version)).lower()
- print(f"should_deploy={should_deploy}")
|