1 #!/usr/bin/env python3
2 # SPDX-License-Identifier: (LGPL-2.1 OR BSD-2-
3 """List model names from mapfile.csv files."""
4 import argparse
5 import csv
6 import os
7 import re
8 from typing import List
9
10 def main() -> None:
11 def dir_path(path: str) -> str:
12 """Validate path is a directory for ar
13 if os.path.isdir(path):
14 return path
15 raise argparse.ArgumentTypeError(f'\'{
16
17 def find_archs(start_dir: str, arch: str)
18 archs = []
19 for item in os.scandir(start_dir):
20 if not item.is_dir():
21 continue
22 if arch in (item.name, 'all'):
23 archs.append(item.name)
24
25 if len(archs) < 1:
26 raise IOError(f'Missing architectu
27
28 return archs
29
30 def find_mapfiles(start_dir: str, archs: L
31 result = []
32 for arch in archs:
33 for item in os.scandir(f'{start_di
34 if item.is_dir():
35 continue
36 if item.name == 'mapfile.csv':
37 result.append(f'{start_dir
38 return result
39
40 def find_cpuids(mapfiles: List[str], cpuid
41 result = []
42 for mapfile in mapfiles:
43 with open(mapfile, encoding='utf-8
44 first = False
45 table = csv.reader(csvfile)
46 for row in table:
47 if not first or len(row) =
48 first = True
49 continue
50 # Python regular expressio
51 regex = row[0].replace('[[
52 for cpuid in cpuids.split(
53 if re.match(regex, cpu
54 result.append(row[
55 return result
56
57 ap = argparse.ArgumentParser()
58 ap.add_argument('arch', help='Architecture
59 ap.add_argument('cpuid', default='all', he
60 ap.add_argument(
61 'starting_dir',
62 type=dir_path,
63 help='Root of tree containing architec
64 )
65 args = ap.parse_args()
66
67 archs = find_archs(args.starting_dir, args
68 mapfiles = find_mapfiles(args.starting_dir
69 models = find_cpuids(mapfiles, args.cpuid)
70 print(','.join(models))
71
72 if __name__ == '__main__':
73 main()
Linux® is a registered trademark of Linus Torvalds in the United States and other countries.
TOMOYO® is a registered trademark of NTT DATA CORPORATION.