wehub-resource-sync
83 lines
2.7 KiB
Python
83 lines
2.7 KiB
Python
# 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,
|
|
|
|
adapted 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}")
|