Export/import a PySpark schema to/from a JSON file

export-pyspark-schema-to-json.py

import json
from pyspark.sql.types import *

# Define the schema
schema = StructType(
    [StructField("name", StringType(), True), StructField("age", IntegerType(), True)]
)

# Write the schema
with open("schema.json", "w") as f:
    json.dump(schema.jsonValue(), f)

# Read the schema
with open("schema.json") as f:
    new_schema = StructType.fromJson(json.load(f))
    print(new_schema.simpleString())