factory

ensembl.io.genomio.schemas.json.factory

Generates one JSON file per metadata type inside manifest, including the manifest itself.

main()

Main script entry-point.

Source code in src/python/ensembl/io/genomio/schemas/json/factory.py

def main() -> None:
    """Main script entry-point."""
    parser = ArgumentParser(
        description="Generates one JSON file per metadata type in the provided manifest, including itself."
    )
    parser.add_argument_src_path(
        "--manifest_dir", required=True, help="Folder containing the 'manifest.json' file to check"
    )
    parser.add_argument(
        "--metadata_types", required=True, nargs="+", metavar="TYPE", help="Metadata types to extract"
    )
    parser.add_argument_dst_path(
        "--output_dir", default=Path.cwd(), help="Folder to store the produced files"
    )
    parser.add_argument("--version", action="version", version=ensembl.io.genomio.__version__)
    parser.add_log_arguments()
    args = parser.parse_args()
    init_logging_with_args(args)

    schema_factory(**vars(args))

schema_factory(manifest_dir, metadata_types, output_dir)

Generates one JSON file per metadata type inside manifest, including "manifest.json" itself.

Each JSON file will have the file name of the metadata type, e.g. "seq_region.json".

Parameters:

Name	Type	Description	Default
manifest_dir	PathLike	Path to the folder with the manifest JSON file to check.	required
metadata_types	List[str]	Metadata types to extract from manifest as JSON files.	required
output_dir	PathLike	Path to the folder where to generate the JSON files.	required

Source code in src/python/ensembl/io/genomio/schemas/json/factory.py

def schema_factory(manifest_dir: PathLike, metadata_types: List[str], output_dir: PathLike) -> None:
    """Generates one JSON file per metadata type inside `manifest`, including "manifest.json" itself.

    Each JSON file will have the file name of the metadata type, e.g. "seq_region.json".

    Args:
        manifest_dir: Path to the folder with the manifest JSON file to check.
        metadata_types: Metadata types to extract from `manifest` as JSON files.
        output_dir: Path to the folder where to generate the JSON files.

    """
    manifest_path = Path(manifest_dir, "manifest.json")
    with manifest_path.open() as manifest_file:
        content = json.load(manifest_file)
        shutil.copyfile(manifest_path, Path(output_dir, "manifest.json"))
        json_files = {}
        # Use dir name from the manifest
        for name in content:
            if "file" in content[name]:
                file_name = content[name]["file"]
                json_files[name] = manifest_path.parent / file_name
            else:
                for key in content[name]:
                    if "file" in content[name][key]:
                        file_name = content[name][key]["file"]
                        json_files[name] = {key: manifest_path.parent / file_name}
        # Check the other JSON schemas
        for metadata_key in metadata_types:
            if metadata_key in json_files:
                if isinstance(json_files[metadata_key], dict):
                    for key, filepath in json_files[metadata_key].items():
                        shutil.copyfile(filepath, Path(output_dir, f"{metadata_key}_{key}.json"))
                else:
                    shutil.copyfile(json_files[metadata_key], Path(output_dir, f"{metadata_key}.json"))