posthog/bin/build-schema-python.sh

#!/usr/bin/env bash

set -e

# Generate schema.py from schema.json
datamodel-codegen \
    --class-name='SchemaRoot' --collapse-root-models --target-python-version 3.11 --disable-timestamp \
    --use-one-literal-as-default --use-default --use-default-kwarg --use-subclass-enum \
    --input frontend/src/queries/schema.json --input-file-type jsonschema \
    --output posthog/schema.py --output-model-type pydantic_v2.BaseModel \
    --custom-file-header "# mypy: disable-error-code=\"assignment\"" \
    --set-default-enum-member --capitalise-enum-members \
    --wrap-string-literal

# Format schema.py
ruff format posthog/schema.py

# Check schema.py and autofix
ruff check --fix posthog/schema.py

# Replace class Foo(str, Enum) with class Foo(StrEnum) for proper handling in format strings in python 3.11
# Remove this when https://github.com/koxudaxi/datamodel-code-generator/issues/1313 is resolved
if [[ "$OSTYPE" == "darwin"* ]]; then
    # sed needs `-i` to be followed by `''` on macOS
    sed -i '' -e 's/str, Enum/StrEnum/g' posthog/schema.py
    sed -i '' 's/from enum import Enum/from enum import Enum, StrEnum/g' posthog/schema.py
else
    sed -i -e 's/str, Enum/StrEnum/g' posthog/schema.py
    sed -i 's/from enum import Enum/from enum import Enum, StrEnum/g' posthog/schema.py
fi