| pulkit |
| hg-reviewers |
This allows main() and argument parsing to work.
| Lint Skipped |
| Unit Tests Skipped |
| Path | Packages | |||
|---|---|---|---|---|
| M | mercurial/statprof.py (24 lines) |
| Commit | Parents | Author | Summary | Date |
|---|---|---|---|---|
| Gregory Szorc | Oct 12 2018, 1:07 PM |
| if len(argv) == 1: | if len(argv) == 1: | ||||
| printusage() | printusage() | ||||
| return 0 | return 0 | ||||
| displayargs = {} | displayargs = {} | ||||
| optstart = 2 | optstart = 2 | ||||
| displayargs['function'] = None | displayargs['function'] = None | ||||
| if argv[1] == 'hotpath': | if argv[1] == r'hotpath': | ||||
| displayargs['format'] = DisplayFormats.Hotpath | displayargs['format'] = DisplayFormats.Hotpath | ||||
| elif argv[1] == 'lines': | elif argv[1] == r'lines': | ||||
| displayargs['format'] = DisplayFormats.ByLine | displayargs['format'] = DisplayFormats.ByLine | ||||
| elif argv[1] == 'functions': | elif argv[1] == r'functions': | ||||
| displayargs['format'] = DisplayFormats.ByMethod | displayargs['format'] = DisplayFormats.ByMethod | ||||
| elif argv[1] == 'function': | elif argv[1] == r'function': | ||||
| displayargs['format'] = DisplayFormats.AboutMethod | displayargs['format'] = DisplayFormats.AboutMethod | ||||
| displayargs['function'] = argv[2] | displayargs['function'] = argv[2] | ||||
| optstart = 3 | optstart = 3 | ||||
| elif argv[1] == 'flame': | elif argv[1] == r'flame': | ||||
| displayargs['format'] = DisplayFormats.FlameGraph | displayargs['format'] = DisplayFormats.FlameGraph | ||||
| else: | else: | ||||
| printusage() | printusage() | ||||
| return 0 | return 0 | ||||
| # process options | # process options | ||||
| try: | try: | ||||
| opts, args = pycompat.getoptb(sys.argv[optstart:], "hl:f:o:p:", | opts, args = pycompat.getoptb(sys.argv[optstart:], "hl:f:o:p:", | ||||
| ["help", "limit=", "file=", "output-file=", "script-path="]) | ["help", "limit=", "file=", "output-file=", "script-path="]) | ||||
| except getopt.error as msg: | except getopt.error as msg: | ||||
| print(msg) | print(msg) | ||||
| printusage() | printusage() | ||||
| return 2 | return 2 | ||||
| displayargs['limit'] = 0.05 | displayargs['limit'] = 0.05 | ||||
| path = None | path = None | ||||
| for o, value in opts: | for o, value in opts: | ||||
| if o in ("-l", "--limit"): | if o in (r"-l", r"--limit"): | ||||
| displayargs['limit'] = float(value) | displayargs['limit'] = float(value) | ||||
| elif o in ("-f", "--file"): | elif o in (r"-f", r"--file"): | ||||
| path = value | path = value | ||||
| elif o in ("-o", "--output-file"): | elif o in (r"-o", r"--output-file"): | ||||
| displayargs['outputfile'] = value | displayargs['outputfile'] = value | ||||
| elif o in ("-p", "--script-path"): | elif o in (r"-p", r"--script-path"): | ||||
| displayargs['scriptpath'] = value | displayargs['scriptpath'] = value | ||||
| elif o in ("-h", "help"): | elif o in (r"-h", r"help"): | ||||
| printusage() | printusage() | ||||
| return 0 | return 0 | ||||
| else: | else: | ||||
| assert False, "unhandled option %s" % o | assert False, "unhandled option %s" % o | ||||
| if not path: | if not path: | ||||
| print('must specify --file to load') | print(r'must specify --file to load') | ||||
| return 1 | return 1 | ||||
| load_data(path=path) | load_data(path=path) | ||||
| display(**pycompat.strkwargs(displayargs)) | display(**pycompat.strkwargs(displayargs)) | ||||
| return 0 | return 0 | ||||
| if __name__ == "__main__": | if __name__ == r"__main__": | ||||
| sys.exit(main()) | sys.exit(main()) | ||||