From 4f2e657243cc13bfd0e99f2cb2f4f4b3591ae31b Mon Sep 17 00:00:00 2001 From: MRXLT Date: Mon, 31 Aug 2020 10:52:18 +0800 Subject: [PATCH] bug fix --- build_server_py3/python/gen_version.py | 43 ++++++++++++++++++++++++++ 1 file changed, 43 insertions(+) create mode 100644 build_server_py3/python/gen_version.py diff --git a/build_server_py3/python/gen_version.py b/build_server_py3/python/gen_version.py new file mode 100644 index 00000000..b11f8cd7 --- /dev/null +++ b/build_server_py3/python/gen_version.py @@ -0,0 +1,43 @@ +# Copyright (c) 2020 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. + +import sys +import re +import os +import subprocess + + +def update_info(file_name, feature, info): + new_str = "" + with open(file_name, "r") as f: + for line in f.readlines(): + if re.match(feature, line): + if isinstance(info, str): + line = feature + " = \"" + info.strip() + "\"" + else: + line = feature + " = \"" + info.decode("utf-8").strip( + ) + "\"" + new_str = new_str + line + + with open(file_name, "w") as f: + f.write(new_str) + + +if len(sys.argv) > 2: + update_info("paddle_serving_server_gpu/version.py", "cuda_version", + sys.argv[2]) + +path = "paddle_serving_" + sys.argv[1] +commit_id = subprocess.check_output(['git', 'rev-parse', 'HEAD']) +update_info(path + "/version.py", "commit_id", commit_id) -- GitLab