mirror of
https://github.com/revng/revng
synced 2026-06-21 14:07:57 +00:00
Giacomo Vercesi
07c2457a16
This commit introduces some miscellaneous fixes to revng.cli: * Remove useless subclasses in pipeline_tools.py; * Improve type annotations in _commands/ptml and fix a help string; * Add some extra type annotations for `Command`; * Remove useless namespaces in `commands_registry` and add help text for the ones in use;
81 lines
2.3 KiB
Python
81 lines
2.3 KiB
Python
#
|
|
# This file is distributed under the MIT License. See LICENSE.md for details.
|
|
#
|
|
|
|
import argparse
|
|
import sys
|
|
from typing import Dict, Optional
|
|
|
|
from revng.cli.commands_registry import Command, Options
|
|
|
|
from .common import (
|
|
handle_file,
|
|
log,
|
|
normalize_filter_extract,
|
|
strip_ptml,
|
|
suppress_brokenpipe,
|
|
yaml_dump,
|
|
)
|
|
|
|
|
|
def strip_ptml_many(content: Dict[str, str]) -> str:
|
|
result = {key: strip_ptml(value) for key, value in content.items()}
|
|
return yaml_dump(result)
|
|
|
|
|
|
def cmd_strip(args):
|
|
if args.inplace and args.input == sys.stdin:
|
|
log("Cannot strip inplace while reading from stdin")
|
|
return 1
|
|
|
|
filters = normalize_filter_extract(args.filter, args.extract)
|
|
content = args.input.read()
|
|
result = handle_file(content, strip_ptml, strip_ptml_many, filters)
|
|
if args.inplace:
|
|
args.input.seek(0)
|
|
args.input.truncate(0)
|
|
args.input.write(result)
|
|
else:
|
|
args.output.write(result)
|
|
return 0
|
|
|
|
|
|
class PTMLStripCommand(Command):
|
|
def __init__(self):
|
|
super().__init__(("ptml", "strip"), "Strip PTML from a file")
|
|
|
|
def register_arguments(self, parser: argparse.ArgumentParser):
|
|
parser_filter = parser.add_mutually_exclusive_group()
|
|
parser_filter.add_argument(
|
|
"-f",
|
|
"--filter",
|
|
type=str,
|
|
action="append",
|
|
default=[],
|
|
required=False,
|
|
help="Only show the keys specified (if present)",
|
|
)
|
|
parser_filter.add_argument(
|
|
"-e", "--extract", type=str, required=False, help="Extract the specified key"
|
|
)
|
|
parser.add_argument(
|
|
"input",
|
|
type=argparse.FileType("r+"),
|
|
default=sys.stdin,
|
|
nargs="?",
|
|
help="Input file (stdin if omitted)",
|
|
)
|
|
parser_strip_out = parser.add_mutually_exclusive_group()
|
|
parser_strip_out.add_argument("-i", "--inplace", action="store_true", help="Strip inplace")
|
|
parser_strip_out.add_argument(
|
|
"output",
|
|
type=argparse.FileType("w"),
|
|
default=sys.stdout,
|
|
nargs="?",
|
|
help="Output file (stdout if omitted)",
|
|
)
|
|
parser.set_defaults(func=cmd_strip)
|
|
|
|
def run(self, options: Options) -> Optional[int]:
|
|
return suppress_brokenpipe(cmd_strip, options.parsed_args)
|