"""Check CANN build dependencies."""
import argparse
import logging
import os
import sys
from typing import Iterator, List, NamedTuple, Optional
class Receiver(NamedTuple):
"""Collect warning and error messages."""
warn_msgs: List[str]
err_msgs: List[str]
def parse_version_line(line: str) -> str:
"""Parse a Version field from version.info."""
version = line.strip().split("=", maxsplit=1)[1]
return version.split("-")[0]
def read_pkg_version(recv: Receiver, ascend_install_path: str, name: str) -> Optional[str]:
"""Read the installed version of a CANN package."""
filepath = os.path.join(ascend_install_path, "share", "info", name, "version.info")
if not os.path.isfile(filepath):
recv.err_msgs.append(f"{filepath} does not exist in read_pkg_version!")
return None
with open(filepath, encoding="utf-8") as file:
for line in file:
if line.startswith("Version="):
return parse_version_line(line)
recv.err_msgs.append(f"The version field was not found in {filepath} in read_pkg_version!")
return None
def check_build_dep_item(version: str, dep: str) -> bool:
"""Check one version constraint."""
def split_version(value: str) -> List[int]:
return [int(num) for num in value.split(".")]
def check_ge() -> bool:
for req, rel in zip(dep_parts, version_parts):
if req > rel:
return False
if rel > req:
return True
return True
def check_gt() -> bool:
for req, rel in zip(dep_parts, version_parts):
if req > rel:
return False
if rel > req:
return True
return False
def check_le() -> bool:
for req, rel in zip(dep_parts, version_parts):
if rel > req:
return False
if req > rel:
return True
return True
def check_lt() -> bool:
for req, rel in zip(dep_parts, version_parts):
if rel > req:
return False
if req > rel:
return True
return False
def check_eq() -> bool:
return all(req == rel for req, rel in zip(dep_parts, version_parts))
version_parts = split_version(version)
if dep.startswith(">="):
dep = dep[2:]
check_func = check_ge
elif dep.startswith(">"):
dep = dep[1:]
check_func = check_gt
elif dep.startswith("<="):
dep = dep[2:]
check_func = check_le
elif dep.startswith("<"):
dep = dep[1:]
check_func = check_lt
else:
check_func = check_eq
dep_parts = split_version(dep)
if check_func is check_eq:
return check_func()
if len(dep_parts) < len(version_parts):
dep_parts.extend([0] * (len(version_parts) - len(dep_parts)))
elif len(dep_parts) > len(version_parts):
version_parts.extend([0] * (len(dep_parts) - len(version_parts)))
return check_func()
def check_build_dep(version: str, dep_info: str) -> bool:
"""Check a comma-separated dependency expression."""
def check_range(dep: str, deps_iter: Iterator[str]) -> bool:
result = check_build_dep_item(version, dep)
for range_dep in deps_iter:
if range_dep.startswith("<") or range_dep.startswith("<="):
return result and check_build_dep_item(version, range_dep)
result &= check_build_dep_item(version, range_dep)
return result
deps = [dep.strip() for dep in dep_info.split(",")]
deps_iter = iter(deps)
for dep in deps_iter:
if dep.startswith(">=") or dep.startswith(">"):
result = check_range(dep, deps_iter)
else:
result = check_build_dep_item(version, dep)
if result:
return True
return False
def check_build_deps(recv: Receiver, ascend_install_path: str, deps: List[str]) -> None:
"""Check all build dependencies."""
deps_iter = iter(deps)
for dep_pkg in deps_iter:
dep_info = next(deps_iter)
version = read_pkg_version(recv, ascend_install_path, dep_pkg)
if not version:
continue
try:
if not check_build_dep(version, dep_info):
recv.err_msgs.append(
"Check build dependency failed! "
f"Required {dep_pkg} version is {dep_info}, but {dep_pkg} version is {version}."
)
except ValueError:
recv.err_msgs.append(f"Check build dependency error! version is {version}, dep_info is {dep_info}.")
def main() -> bool:
"""Run dependency checks."""
parser = argparse.ArgumentParser()
parser.add_argument("ascend_install_path", help="Ascend install path.")
parser.add_argument("deps", nargs="*", help="Dependency information.")
args = parser.parse_args()
logging.basicConfig(format=f"{os.path.basename(__file__)}: %(levelname)s: %(message)s")
if len(args.deps) % 2 != 0:
logging.error("The deps argument must contain an even number of elements!")
return False
recv = Receiver([], [])
check_build_deps(recv, args.ascend_install_path, args.deps)
for warn_msg in recv.warn_msgs:
logging.warning(warn_msg)
if recv.err_msgs:
for err_msg in recv.err_msgs:
logging.error(err_msg)
return False
return True
if __name__ == "__main__":
if not main():
sys.exit(1)
