Mercurial > hg
view contrib/fuzz/revlog_corpus.py @ 46608:4e4c70401028
rhg: Use clap’s support for global CLI arguments
By default, clap only accepts app-level arguments (as opposed to sub-command
level) to be specified before a sub-command: `rhg -R ./foo log`. Specifying
them after would be rejected: `rhg log -R ./foo`.
Previously we worked around that by registering global arguments both
at the app level and on each sub-command, but that required looking
for their value in two places. It turns out that Clap has built-in support
for what we want to do, so let’s use it.
Also, Clap "settings" turn out to be either global or not too.
Let’s make `AllowInvalidUtf8` apply to sub-commands too.
Differential Revision: https://phab.mercurial-scm.org/D10080
author | Simon Sapin <simon.sapin@octobus.net> |
---|---|
date | Fri, 26 Feb 2021 12:16:43 +0100 |
parents | ba84a1ae4ae5 |
children | 6000f5b25c9b |
line wrap: on
line source
from __future__ import absolute_import import argparse import os import zipfile ap = argparse.ArgumentParser() ap.add_argument("out", metavar="some.zip", type=str, nargs=1) args = ap.parse_args() reporoot = os.path.normpath(os.path.join(os.path.dirname(__file__), '..', '..')) # typically a standalone index changelog = os.path.join(reporoot, '.hg', 'store', '00changelog.i') # an inline revlog with only a few revisions contributing = os.path.join( reporoot, '.hg', 'store', 'data', 'contrib', 'fuzz', 'mpatch.cc.i' ) with zipfile.ZipFile(args.out[0], "w", zipfile.ZIP_STORED) as zf: if os.path.exists(changelog): with open(changelog, 'rb') as f: zf.writestr("00changelog.i", f.read()) if os.path.exists(contributing): with open(contributing, 'rb') as f: zf.writestr("contributing.i", f.read())