Mercurial > vim-crosoft

annotate scripts/list_sln_files.py @ 5:bac97082e229

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
Add more caching to listing solution files.
author Ludovic Chabant <ludovic@chabant.com>
date Thu, 24 Oct 2019 11:16:11 -0700
parents 5d2c0db51914
children d5ddd9ffaf11
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
0
5d2c0db51914 Initial commit
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
1 import argparse
5d2c0db51914 Initial commit
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
2 import logging
5d2c0db51914 Initial commit
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
3 import os.path
5d2c0db51914 Initial commit
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
4 from logutil import setup_logging
5
bac97082e229 Add more caching to listing solution files.
Ludovic Chabant <ludovic@chabant.com>
parents: 0
diff changeset
5 from vsutil import SolutionCache, ITEM_TYPE_SOURCE_FILES
0
5d2c0db51914 Initial commit
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
6
5d2c0db51914 Initial commit
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
7
5d2c0db51914 Initial commit
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
8 logger = logging.getLogger(__name__)
5d2c0db51914 Initial commit
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
9
5d2c0db51914 Initial commit
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
10
5d2c0db51914 Initial commit
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
11 def main(args=None):
5d2c0db51914 Initial commit
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
12 parser = argparse.ArgumentParser()
5d2c0db51914 Initial commit
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
13 parser.add_argument('solution',
5d2c0db51914 Initial commit
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
14 help="The path to the Visual Studio solution file.")
5
bac97082e229 Add more caching to listing solution files.
Ludovic Chabant <ludovic@chabant.com>
parents: 0
diff changeset
15 parser.add_argument('-c', '--cache',
bac97082e229 Add more caching to listing solution files.
Ludovic Chabant <ludovic@chabant.com>
parents: 0
diff changeset
16 help="The solution cache file to load.")
bac97082e229 Add more caching to listing solution files.
Ludovic Chabant <ludovic@chabant.com>
parents: 0
diff changeset
17 parser.add_argument('--list-cache',
bac97082e229 Add more caching to listing solution files.
Ludovic Chabant <ludovic@chabant.com>
parents: 0
diff changeset
18 help=("If the solution cache is valid, use this "
bac97082e229 Add more caching to listing solution files.
Ludovic Chabant <ludovic@chabant.com>
parents: 0
diff changeset
19 "pre-saved file list. Otherwise, compute the "
bac97082e229 Add more caching to listing solution files.
Ludovic Chabant <ludovic@chabant.com>
parents: 0
diff changeset
20 "file list and save it to the given path."))
0
5d2c0db51914 Initial commit
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
21 parser.add_argument('-p', '--project',
5d2c0db51914 Initial commit
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
22 help="Only list files in the named project.")
5
bac97082e229 Add more caching to listing solution files.
Ludovic Chabant <ludovic@chabant.com>
parents: 0
diff changeset
23 parser.add_argument('-t', '--type',
bac97082e229 Add more caching to listing solution files.
Ludovic Chabant <ludovic@chabant.com>
parents: 0
diff changeset
24 action='append',
bac97082e229 Add more caching to listing solution files.
Ludovic Chabant <ludovic@chabant.com>
parents: 0
diff changeset
25 help="The type(s) of items to list.")
0
5d2c0db51914 Initial commit
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
26 parser.add_argument('-v', '--verbose',
5d2c0db51914 Initial commit
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
27 action='store_true',
5d2c0db51914 Initial commit
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
28 help="Show verbose information.")
5d2c0db51914 Initial commit
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
29 args = parser.parse_args(args)
5d2c0db51914 Initial commit
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
30 setup_logging(args.verbose)
5d2c0db51914 Initial commit
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
31
5
bac97082e229 Add more caching to listing solution files.
Ludovic Chabant <ludovic@chabant.com>
parents: 0
diff changeset
32 cache, loaded = SolutionCache.load_or_rebuild(args.solution, args.cache)
bac97082e229 Add more caching to listing solution files.
Ludovic Chabant <ludovic@chabant.com>
parents: 0
diff changeset
33 if loaded and args.list_cache:
bac97082e229 Add more caching to listing solution files.
Ludovic Chabant <ludovic@chabant.com>
parents: 0
diff changeset
34 cache_dt = os.path.getmtime(args.cache)
bac97082e229 Add more caching to listing solution files.
Ludovic Chabant <ludovic@chabant.com>
parents: 0
diff changeset
35 list_cache_dt = os.path.getmtime(args.list_cache)
bac97082e229 Add more caching to listing solution files.
Ludovic Chabant <ludovic@chabant.com>
parents: 0
diff changeset
36 if list_cache_dt > cache_dt:
bac97082e229 Add more caching to listing solution files.
Ludovic Chabant <ludovic@chabant.com>
parents: 0
diff changeset
37 logger.debug("Solution cache was valid, re-using the file list cache.")
bac97082e229 Add more caching to listing solution files.
Ludovic Chabant <ludovic@chabant.com>
parents: 0
diff changeset
38 try:
bac97082e229 Add more caching to listing solution files.
Ludovic Chabant <ludovic@chabant.com>
parents: 0
diff changeset
39 with open(args.list_cache, 'r') as fp:
bac97082e229 Add more caching to listing solution files.
Ludovic Chabant <ludovic@chabant.com>
parents: 0
diff changeset
40 print(fp.read())
bac97082e229 Add more caching to listing solution files.
Ludovic Chabant <ludovic@chabant.com>
parents: 0
diff changeset
41 return
bac97082e229 Add more caching to listing solution files.
Ludovic Chabant <ludovic@chabant.com>
parents: 0
diff changeset
42 except OSError:
bac97082e229 Add more caching to listing solution files.
Ludovic Chabant <ludovic@chabant.com>
parents: 0
diff changeset
43 logger.debug("File list cache unreachable, recomputing it.")
bac97082e229 Add more caching to listing solution files.
Ludovic Chabant <ludovic@chabant.com>
parents: 0
diff changeset
44 else:
bac97082e229 Add more caching to listing solution files.
Ludovic Chabant <ludovic@chabant.com>
parents: 0
diff changeset
45 logger.debug("Solution cache was valid but file list cache was older, "
bac97082e229 Add more caching to listing solution files.
Ludovic Chabant <ludovic@chabant.com>
parents: 0
diff changeset
46 "recomputing it.")
bac97082e229 Add more caching to listing solution files.
Ludovic Chabant <ludovic@chabant.com>
parents: 0
diff changeset
47
0
5d2c0db51914 Initial commit
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
48 slnobj = cache.slnobj
5d2c0db51914 Initial commit
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
49
5d2c0db51914 Initial commit
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
50 projs = slnobj.projects
5d2c0db51914 Initial commit
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
51 if args.project:
5d2c0db51914 Initial commit
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
52 projs = [slnobj.find_project_by_name(args.project)]
5d2c0db51914 Initial commit
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
53 projs = list(filter(lambda p: not p.is_folder, projs))
5d2c0db51914 Initial commit
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
54
5
bac97082e229 Add more caching to listing solution files.
Ludovic Chabant <ludovic@chabant.com>
parents: 0
diff changeset
55 itemtypes = args.type or ITEM_TYPE_SOURCE_FILES
bac97082e229 Add more caching to listing solution files.
Ludovic Chabant <ludovic@chabant.com>
parents: 0
diff changeset
56 items = []
bac97082e229 Add more caching to listing solution files.
Ludovic Chabant <ludovic@chabant.com>
parents: 0
diff changeset
57
0
5d2c0db51914 Initial commit
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
58 for p in projs:
5d2c0db51914 Initial commit
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
59 ig = p.defaultitemgroup()
5
bac97082e229 Add more caching to listing solution files.
Ludovic Chabant <ludovic@chabant.com>
parents: 0
diff changeset
60 for i in ig.get_items_of_types(itemtypes):
0
5d2c0db51914 Initial commit
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
61 file_path = os.path.abspath(os.path.join(p.absdirpath, i.include))
5d2c0db51914 Initial commit
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
62 print(file_path)
5
bac97082e229 Add more caching to listing solution files.
Ludovic Chabant <ludovic@chabant.com>
parents: 0
diff changeset
63 items.append(file_path + '\n')
bac97082e229 Add more caching to listing solution files.
Ludovic Chabant <ludovic@chabant.com>
parents: 0
diff changeset
64
bac97082e229 Add more caching to listing solution files.
Ludovic Chabant <ludovic@chabant.com>
parents: 0
diff changeset
65 if args.list_cache:
bac97082e229 Add more caching to listing solution files.
Ludovic Chabant <ludovic@chabant.com>
parents: 0
diff changeset
66 logger.debug("Writing file list cache: %s" % args.list_cache)
bac97082e229 Add more caching to listing solution files.
Ludovic Chabant <ludovic@chabant.com>
parents: 0
diff changeset
67 with open(args.list_cache, 'w') as fp:
bac97082e229 Add more caching to listing solution files.
Ludovic Chabant <ludovic@chabant.com>
parents: 0
diff changeset
68 fp.writelines(items)
0
5d2c0db51914 Initial commit
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
69
5d2c0db51914 Initial commit
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
70
5d2c0db51914 Initial commit
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
71 if __name__ == '__main__':
5d2c0db51914 Initial commit
Ludovic Chabant <ludovic@chabant.com>
parents:
diff changeset
72 main()