mirror of
https://github.com/revng/revng
synced 2026-06-21 14:07:57 +00:00
Giacomo Vercesi
6fc6d38228
Python 3.14 will deprecate `argparse.FileType` [1]. This is due to a variety of factors, one of which is that the class leaks file descriptors when used. Remove all usages in the codebase and instead use a wrapper function for `open`. [1] https://docs.python.org/3.14/library/argparse.html#argparse.FileType
44 lines
1.4 KiB
Python
44 lines
1.4 KiB
Python
#
|
|
# This file is distributed under the MIT License. See LICENSE.md for details.
|
|
#
|
|
|
|
import argparse
|
|
from typing import Optional
|
|
|
|
import yaml
|
|
|
|
from revng.internal.cli.commands_registry import Command, CommandsRegistry, Options
|
|
from revng.internal.cli.support import extract_tar, file_wrapper, to_string, to_yaml
|
|
|
|
|
|
def auto_process(filename: str, raw: bytes) -> str:
|
|
# TODO: add base64 for non-text stuff
|
|
if filename.endswith(".yml"):
|
|
return to_yaml(filename, raw)
|
|
else:
|
|
return to_string(filename, raw)
|
|
|
|
|
|
class TarToYAMLCommand(Command):
|
|
def __init__(self):
|
|
super().__init__(("tar", "to-yaml"), "Turn a tar archive into YAML")
|
|
|
|
def register_arguments(self, parser: argparse.ArgumentParser):
|
|
parser.description = "Turn a tar archive into YAML"
|
|
parser.add_argument("input", nargs="?", help="Input file (stdin if omitted)")
|
|
parser.add_argument(
|
|
"-o", "--output", nargs="?", metavar="FILE", help="Output file (stdout if omitted)"
|
|
)
|
|
|
|
def run(self, options: Options) -> Optional[int]:
|
|
args = options.parsed_args
|
|
with file_wrapper(args.input, "rb") as input_file:
|
|
output = yaml.dump(extract_tar(input_file.read(), auto_process))
|
|
with file_wrapper(args.output, "w") as output_file:
|
|
output_file.write(output)
|
|
return 0
|
|
|
|
|
|
def setup(commands_registry: CommandsRegistry):
|
|
commands_registry.register_command(TarToYAMLCommand())
|