Extract schema

extract-schema.ts

type SchemaType =
  | string
  | { [key: string]: SchemaType }
  | [string, SchemaType | null];

function extractSchema(obj: any): SchemaType {
  if (obj === null) {
    return "null";
  }

  if (typeof obj === "object") {
    if (Array.isArray(obj)) {
      return [
        `array[${obj.length}]`,
        obj.length > 0 ? extractSchema(obj[0]) : null,
      ];
    } else {
      const result: { [key: string]: SchemaType } = {};
      for (const [key, value] of Object.entries(obj)) {
        result[key] = extractSchema(value);
      }
      return result;
    }
  }

  return typeof obj;
}

extract_schema.py

def extract_schema(obj):
    if isinstance(obj, dict):
        return {k: extract_schema(v) for k, v in obj.items()}
    if isinstance(obj, list):
        return [f'list[{len(obj)}]', extract_schema(obj[0]) if obj else None]
    return type(obj).__name__