Chromium Code Reviews| OLD | NEW |
|---|---|
| (Empty) | |
| 1 #!/usr/bin/env python | |
|
Mark Seaborn
2014/05/19 18:55:29
Can you remove "nonsfi" from the file's name, perh
hidehiko
2014/05/20 05:51:47
I'm not still very sure if this should be reused f
Mark Seaborn
2014/05/21 00:08:39
OK, let's not get bogged down in this review with
hidehiko
2014/05/21 13:25:32
Acknowledged. Thanks for reply.
| |
| 2 # Copyright 2014 The Chromium Authors. All rights reserved. | |
| 3 # Use of this source code is governed by a BSD-style license that can be | |
| 4 # found in the LICENSE file. | |
| 5 | |
| 6 """Simple tool to generate NMF file from given arguments for nonsfi.""" | |
|
Mark Seaborn
2014/05/19 18:55:29
Can you add a comment saying why this tool exists,
hidehiko
2014/05/20 05:51:47
Could you tell me the future plans as I asked abov
Mark Seaborn
2014/05/21 00:08:39
The comment doesn't need to describe future plans,
hidehiko
2014/05/21 13:25:32
Done. Thank you for suggestion. How about this?
| |
| 7 | |
| 8 import argparse | |
| 9 import collections | |
| 10 import json | |
| 11 import logging | |
| 12 import os | |
| 13 | |
| 14 _FILES_KEY = 'files' | |
| 15 _PORTABLE_KEY = 'portable' | |
| 16 _PROGRAM_KEY = 'program' | |
| 17 _URL_KEY = 'url' | |
| 18 _X86_32_NONSFI_KEY = 'x86-32-nonsfi' | |
| 19 | |
| 20 | |
| 21 def ParseArgs(): | |
| 22 parser = argparse.ArgumentParser() | |
| 23 parser.add_argument( | |
| 24 '--program', metavar='FILE', help='Main program file') | |
|
Mark Seaborn
2014/05/19 18:55:29
Maybe s/file/nexe/ to be more specific?
hidehiko
2014/05/20 05:51:47
Done.
| |
| 25 # To keep compatibility with create_nmf.py, we use -x and --extra-files | |
| 26 # as flags. | |
| 27 parser.add_argument( | |
| 28 '-x', '--extra-files', action='append', metavar='KEY:FILE', default=[], | |
| 29 help=('Add extra key:file tuple to the "files"' + | |
| 30 ' esection of the .nmf')) | |
|
Mark Seaborn
2014/05/19 18:55:29
Typo: "section"
hidehiko
2014/05/20 05:51:47
Done.
| |
| 31 parser.add_argument( | |
| 32 '--output', metavar='FILE', help='Path to the output nmf file.') | |
| 33 | |
| 34 return parser.parse_args() | |
| 35 | |
| 36 | |
| 37 def BuildNmfMap(root_path, program, extra_files): | |
| 38 """Build simple map representing nmf json.""" | |
| 39 result = { | |
| 40 _PROGRAM_KEY: { | |
| 41 _X86_32_NONSFI_KEY: { | |
| 42 # The program path is relative to the root_path. | |
| 43 _URL_KEY: os.path.relpath(program, root_path) | |
| 44 } | |
| 45 } | |
| 46 } | |
| 47 | |
| 48 if extra_files: | |
| 49 files = {} | |
| 50 for named_file in extra_files: | |
| 51 name, path = named_file.split(':', 1) | |
| 52 files[name] = { | |
| 53 _PORTABLE_KEY: { | |
| 54 # Note use path as is, unlike program path. | |
|
Mark Seaborn
2014/05/19 18:55:29
Add colon -- "Note:"?
hidehiko
2014/05/20 05:51:47
Done.
| |
| 55 _URL_KEY: path | |
| 56 } | |
| 57 } | |
| 58 if files: | |
| 59 result[_FILES_KEY] = files | |
| 60 | |
| 61 return result | |
| 62 | |
| 63 | |
| 64 def OutputNmf(nmf_map, output_path): | |
| 65 """Writes the nmf to an output file at given path in JSON format.""" | |
| 66 with open(output_path, 'w') as output: | |
| 67 json.dump(nmf_map, output, indent=2) | |
| 68 | |
| 69 | |
| 70 def main(): | |
| 71 args = ParseArgs() | |
| 72 if not args.program: | |
| 73 logging.error('--program is not specified.') | |
| 74 sys.exit(1) | |
| 75 if not args.output: | |
| 76 logging.error('--output is not specified.') | |
| 77 sys.exit(1) | |
| 78 | |
| 79 nmf_map = BuildNmfMap(os.path.dirname(args.output), | |
| 80 args.program, args.extra_files) | |
| 81 OutputNmf(nmf_map, args.output) | |
| 82 | |
| 83 | |
| 84 if __name__ == '__main__': | |
| 85 main() | |
| OLD | NEW |