21 lines
606 B
Python
21 lines
606 B
Python
|
import json
|
||
|
|
import os
|
||
|
|
from conftest import option
|
||
|
|
from genson import SchemaBuilder
|
||
|
|
|
||
|
|
|
||
|
|
class CustomSchemaBuilder(SchemaBuilder):
|
||
|
|
|
||
|
|
def __init__(self, schema_name):
|
||
|
|
super().__init__()
|
||
|
|
self.path = f"{option.base_dir}/schemas/{schema_name}"
|
||
|
|
|
||
|
|
def create_schema(self, response_json):
|
||
|
|
builder = SchemaBuilder()
|
||
|
|
builder.add_object(response_json)
|
||
|
|
schema = builder.to_schema()
|
||
|
|
|
||
|
|
os.makedirs(os.path.dirname(self.path), exist_ok=True)
|
||
|
|
with open(self.path, "w") as file:
|
||
|
|
json.dump(schema, file, sort_keys=True, indent=4, ensure_ascii=False)
|