Changeset View
Changeset View
Standalone View
Standalone View
swh/provenance/tests/data/generate_repo.py
| Show First 20 Lines • Show All 100 Lines • ▼ Show 20 Lines | for rev_d in repo_desc: | ||||
| print_ids() | print_ids() | ||||
| @click.command(name="generate-repo") | @click.command(name="generate-repo") | ||||
| @click.argument("input-file") | @click.argument("input-file") | ||||
| @click.argument("output-dir") | @click.argument("output-dir") | ||||
| @click.option("-C", "--clean-output/--no-clean-output", default=False) | @click.option("-C", "--clean-output/--no-clean-output", default=False) | ||||
| def main(input_file: str, output_dir: str, clean_output: bool) -> None: | def main(input_file: str, output_dir: str, clean_output: bool) -> None: | ||||
| repo_desc = yaml.load(open(input_file)) | repo_desc = yaml.load(open(input_file), Loader=yaml.Loader) | ||||
| if clean_output and os.path.exists(output_dir): | if clean_output and os.path.exists(output_dir): | ||||
| shutil.rmtree(output_dir) | shutil.rmtree(output_dir) | ||||
| generate_repo(repo_desc, output_dir) | generate_repo(repo_desc, output_dir) | ||||
| if __name__ == "__main__": | if __name__ == "__main__": | ||||
| main() | main() | ||||