mirror of
https://github.com/galaxyproject/galaxy.git
synced 2026-08-31 01:02:04 +08:00
mvdbeek
51 lines
1.5 KiB
Python
51 lines
1.5 KiB
Python
import json
|
|
import os
|
|
import sys
|
|
|
|
sys.path.insert(1, os.path.abspath(os.path.join(os.path.dirname(__file__), os.pardir, "lib")))
|
|
|
|
import click
|
|
import yaml
|
|
from pydantic.networks import AnyUrl
|
|
|
|
from galaxy.webapps.galaxy.fast_app import get_openapi_schema
|
|
|
|
|
|
def _any_url_representer(dumper, data):
|
|
return dumper.represent_scalar("!anyurl", str(data))
|
|
|
|
|
|
class YamlDumper(yaml.SafeDumper):
|
|
pass
|
|
|
|
|
|
YamlDumper.add_representer(AnyUrl, _any_url_representer)
|
|
|
|
|
|
@click.command("Write openapi schema to path")
|
|
@click.argument("schema_path", type=click.Path(dir_okay=False, writable=True), required=False)
|
|
@click.option("--app", type=click.Choice(["gx", "shed"]), required=False, default="gx")
|
|
def write_open_api_schema(schema_path, app: str):
|
|
if app == "shed":
|
|
# Importing this causes the Galaxy schema to generate
|
|
# in a different fashion and causes a diff in downstream
|
|
# typescript generation for instance. So delay this.
|
|
from tool_shed.webapp.fast_app import get_openapi_schema as get_openapi_schema_shed
|
|
|
|
openapi_schema = get_openapi_schema_shed()
|
|
else:
|
|
openapi_schema = get_openapi_schema()
|
|
if schema_path:
|
|
if schema_path.endswith((".yml", ".yaml")):
|
|
with open(schema_path, "w") as f:
|
|
yaml.dump(openapi_schema, f, YamlDumper)
|
|
else:
|
|
with open(schema_path, "w") as f:
|
|
json.dump(openapi_schema, f, sort_keys=True)
|
|
else:
|
|
print(json.dumps(openapi_schema, sort_keys=True))
|
|
|
|
|
|
if __name__ == "__main__":
|
|
write_open_api_schema()
|