import sys
import yaml
import re
publish_messages = [
"StartRecognition","AddAudio","EndOfStream","SetRecognitionConfig"
]
with open("../schemas/realtime.yml", 'r') as stream:
try:
spec = stream.read()
async_spec = yaml.safe_load(spec)
except yaml.YAMLError as exc:
print(exc)
with open("template-openapi.yaml", 'r') as stream:
try:
template = yaml.safe_load(stream)
except yaml.YAMLError as exc:
print(exc)
messages_models_yaml = async_spec['components']['messages']
template["components"] = {"schemas": {}}
for model_name, model_content in messages_models_yaml.items():
payload = model_content['payload']
template['components']['schemas'][model_name] = payload
template['components']['schemas'].update(async_spec['components']['schemas'])
with open('openapi-transformed.yaml', 'w') as outfile:
yaml.dump(template, outfile)