Skip to content

Commit

Permalink
use class_generator, not uv for generate
Browse files Browse the repository at this point in the history
  • Loading branch information
myakove committed Jan 16, 2025
1 parent c91ec4b commit 0988056
Showing 1 changed file with 19 additions and 2 deletions.
21 changes: 19 additions & 2 deletions class_generator/scripts/tools.py
Original file line number Diff line number Diff line change
Expand Up @@ -83,11 +83,28 @@ def generate_resource(kinds: List[str], yes: bool) -> None:
help="Generate missing end comment for all resources under `ocp_resources` directory",
)
@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) -> None:
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():
os.system(f"uv run class_generator/class_generator.py -k {kind} --overwrite")
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!")
# for kind in res["with_end_comment"].keys():
# os.system(f"uv run class_generator/class_generator.py -k {kind} --overwrite")

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

0 comments on commit 0988056

Please sign in to comment.