def main(spec_url, spec_path): codegen = CodeGenerator(spec_url, spec_path) codegen_playground = CodeGenerator(PLAYGROUND_SPEC_URL) spec = codegen.open_api_spec print("=" * 100) print("{}: {}".format(spec.info.title, spec.info.version)) print(spec.info.description) print("=" * 100) for root, dirs, files in os.walk("./cognite/client/data_classes"): for file in files: file_path = os.path.join(root, file) if file_path.endswith(".py"): if "relationships" in file_path: print( "* Generating playground code in {}".format(file_path)) codegen_playground.generate(file_path, file_path) else: print("* Generating normal code in {}".format(file_path)) codegen.generate(file_path, file_path)
import argparse import os from openapi.generator import CodeGenerator # DEFAULT_SPEC_URL = "https://storage.googleapis.com/cognitedata-api-docs/dist/v1.json" PLAYGROUND_SPEC_URL = "https://storage.googleapis.com/cognitedata-api-docs/dist/playground.json" if __name__ == "__main__": codegen_playground = CodeGenerator( PLAYGROUND_SPEC_URL, exclude_schemas=["CustomMenuHierarchy", "CustomMenuHierarchyNode"]) spec = codegen_playground.open_api_spec print("=" * 100) print("{}: {}".format(spec.info.title, spec.info.version)) print(spec.info.description) print("=" * 100) for root, dirs, files in os.walk("./cognite/experimental/data_classes"): for file in files: file_path = os.path.join(root, file) if file_path.endswith(".py"): print("* Generating playground code in {}".format(file_path)) codegen_playground.generate(file_path, file_path)