Skip to content

Instantly share code, notes, and snippets.

@mloayzagahona
Forked from aseigneurin/register_schema.py
Created December 4, 2018 19:03
Show Gist options
  • Save mloayzagahona/31fe35fed16ef9d972ca73e2261baf1c to your computer and use it in GitHub Desktop.
Save mloayzagahona/31fe35fed16ef9d972ca73e2261baf1c to your computer and use it in GitHub Desktop.
Register an Avro schema against the Confluent Schema Registry
import os
import sys
import requests
schema_registry_url = sys.argv[1]
topic = sys.argv[2]
schema_file = sys.argv[3]
aboslute_path_to_schema = os.path.join(os.getcwd(), schema_file)
print("Schema Registry URL: " + schema_registry_url)
print("Topic: " + topic)
print("Schema file: " + schema_file)
print
with open(aboslute_path_to_schema, 'r') as content_file:
schema = content_file.read()
payload = "{ \"schema\": \"" \
+ schema.replace("\"", "\\\"").replace("\t", "").replace("\n", "") \
+ "\" }"
url = schema_registry_url + "/subjects/" + topic + "-value/versions"
headers = {"Content-Type": "application/vnd.schemaregistry.v1+json"}
r = requests.post(url, headers=headers, data=payload)
if r.status_code == requests.codes.ok:
print("Success")
else:
r.raise_for_status()
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment