mirror of
https://github.com/revng/revng
synced 2026-06-21 14:07:57 +00:00
Giacomo Vercesi
90e8cce41e
Rework the python mixin infrastructure for the model. Mixins are now specified explicitly instead of using the `ast` module. The mixins are loaded relative to the generated python file, this allows to have two models: `revng.model` and `revng.project.model`. The latter is augmented via mixins to have project-specific facilities such as `get_artifact` and artifact accessors.
84 lines
2.2 KiB
Python
Executable File
84 lines
2.2 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
|
|
#
|
|
# This file is distributed under the MIT License. See LICENSE.md for details.
|
|
#
|
|
|
|
# This script reads the `mixins.py.tpl` template, which contains the mixins for
|
|
# the model in `revng.project.model`. The script will render this via Jinja.
|
|
|
|
import argparse
|
|
import re
|
|
import sys
|
|
from collections import defaultdict
|
|
from dataclasses import dataclass
|
|
from pathlib import Path
|
|
|
|
import yaml
|
|
from jinja2 import Environment, FileSystemLoader
|
|
|
|
from revng.support.artifacts import Artifact
|
|
|
|
|
|
@dataclass
|
|
class ArtifactData:
|
|
name: str
|
|
type_: str
|
|
|
|
|
|
def get_container_type(data, container: str):
|
|
container_type = data["containers"][container]
|
|
for entry in data["container_types"]:
|
|
if entry["class"] == container_type:
|
|
return entry
|
|
raise ValueError
|
|
|
|
|
|
def parse_artifacts(data) -> dict[str, list[ArtifactData]]:
|
|
result: dict[str, list[ArtifactData]] = defaultdict(list)
|
|
for entry in data["artifacts"]:
|
|
if not entry["category"]["show_by_default"]:
|
|
continue
|
|
|
|
container_type = get_container_type(data, entry["container"])
|
|
artifact = Artifact.make(b"", container_type["mime_type"])
|
|
result[container_type["kind"]].append(
|
|
ArtifactData(entry["name"], artifact.__class__.__name__)
|
|
)
|
|
|
|
return result
|
|
|
|
|
|
def normalize(string: str):
|
|
return re.sub(r"[^a-zA-Z]", "_", string)
|
|
|
|
|
|
def main():
|
|
parser = argparse.ArgumentParser()
|
|
parser.add_argument("-o", "--output", default=sys.stdout)
|
|
parser.add_argument("input")
|
|
args = parser.parse_args()
|
|
|
|
data = yaml.safe_load(sys.stdin)
|
|
artifacts = parse_artifacts(data)
|
|
|
|
input_path = Path(args.input)
|
|
jinja_env = Environment(loader=FileSystemLoader(input_path.parent))
|
|
jinja_env.filters["normalize"] = normalize
|
|
template = jinja_env.get_template(input_path.name)
|
|
rendered_template = template.render(
|
|
binary_artifacts=artifacts["binary"],
|
|
function_artifacts=artifacts["function"],
|
|
typedefinition_artifacts=artifacts["type-definition"],
|
|
)
|
|
|
|
if args.output is sys.stdout:
|
|
sys.stdout.write(rendered_template)
|
|
else:
|
|
with open(args.output, "w") as f:
|
|
f.write(rendered_template)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|