chore: import upstream snapshot with attribution
This commit is contained in:
@@ -0,0 +1,82 @@
|
||||
# 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}")
|
||||
Reference in New Issue
Block a user