Skip to content

Commit

Permalink
class-generator-tools: add --regenerate-generated-file (#2283)
Browse files Browse the repository at this point in the history
* class-generator-tools: add --regenerate-generated-file

* class-generator-tools: add --regenerate-generated-file

* use class_generator, not uv for generate

* remove commented code
  • Loading branch information
myakove authored Jan 16, 2025
1 parent 9c4ddac commit b9877b3
Showing 1 changed file with 23 additions and 2 deletions.
25 changes: 23 additions & 2 deletions class_generator/scripts/tools.py
Original file line number Diff line number Diff line change
@@ -1,6 +1,7 @@
import ast
import os
from typing import Dict, List
import ast

import click

from class_generator.class_generator import class_generator
Expand Down Expand Up @@ -81,8 +82,28 @@ def generate_resource(kinds: List[str], yes: bool) -> None:
is_flag=True,
help="Generate missing end comment for all resources under `ocp_resources` directory",
)
def main(list_generated_file: bool, generated_missing_end_comment: bool, yes: bool) -> None:
@click.option("--regenerate-generated-files", is_flag=True, help="Regenerate all generated files")
def main(
list_generated_file: bool, generated_missing_end_comment: bool, yes: bool, regenerate_generated_files: bool
) -> None:
res = get_generated_files()
if regenerate_generated_files:
click.echo("Regenerating files...")
failed_kinds = []
for kind in res["with_end_comment"].keys():
try:
click.echo(f"Regenerating {kind}...")
if not class_generator(kind=kind, called_from_cli=False, overwrite=True):
failed_kinds.append(kind)

except Exception as exc:
click.echo(f"Failed to regenerate {kind}: {exc}", err=True)
failed_kinds.append(kind)
if failed_kinds:
click.echo(f"Failed to regenerate: {', '.join(failed_kinds)}", err=True)
else:
click.echo("All files regenerated successfully!")

if generated_missing_end_comment:
generate_resource(kinds=list(res["without_end_comment"].keys()), yes=yes)

Expand Down

0 comments on commit b9877b3

Please sign in to comment.