1 #!/usr/bin/env python3
2 # SPDX-License-Identifier: (LGPL-2.1 OR BSD-2-Clause)
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 argparse."""
13 if os.path.isdir(path):
14 return path
15 raise argparse.ArgumentTypeError(f'\'{path}\' is not a valid directory')
16
17 def find_archs(start_dir: str, arch: str) -> List[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 architecture directory \'{arch}\'')
27
28 return archs
29
30 def find_mapfiles(start_dir: str, archs: List[str]) -> List[str]:
31 result = []
32 for arch in archs:
33 for item in os.scandir(f'{start_dir}/{arch}'):
34 if item.is_dir():
35 continue
36 if item.name == 'mapfile.csv':
37 result.append(f'{start_dir}/{arch}/mapfile.csv')
38 return result
39
40 def find_cpuids(mapfiles: List[str], cpuids: str) -> List[str]:
41 result = []
42 for mapfile in mapfiles:
43 with open(mapfile, encoding='utf-8') as csvfile:
44 first = False
45 table = csv.reader(csvfile)
46 for row in table:
47 if not first or len(row) == 0 or row[0].startswith('#'):
48 first = True
49 continue
50 # Python regular expressions don't handle xdigit.
51 regex = row[0].replace('[[:xdigit:]]', '[0-9a-fA-F]')
52 for cpuid in cpuids.split(','):
53 if re.match(regex, cpuid):
54 result.append(row[2])
55 return result
56
57 ap = argparse.ArgumentParser()
58 ap.add_argument('arch', help='Architecture name like x86')
59 ap.add_argument('cpuid', default='all', help='List of cpuids to convert to model names')
60 ap.add_argument(
61 'starting_dir',
62 type=dir_path,
63 help='Root of tree containing architecture directories containing json files'
64 )
65 args = ap.parse_args()
66
67 archs = find_archs(args.starting_dir, args.arch)
68 mapfiles = find_mapfiles(args.starting_dir, archs)
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.