feat: fix generated class names that are acronyms (#91)

Allow OnCompleteAction use in ARC4 abimethod, baremethod decorators

BREAKING CHANGE: enum class names have changed

scripts/generate_stubs.py

@@ -606,6 +606,9 @@ def map_typed_names(
 
 
 def get_python_enum_class(arg_enum: str) -> str:
+    # don't change acronyms
+    if arg_enum.isupper():
+        return arg_enum
     return snake_case(arg_enum).replace("_", " ").title().replace(" ", "")
 
 
@@ -647,10 +650,10 @@ def get_overriding_immediate(op: Op) -> Immediate | None:
 def build_enum(spec: LanguageSpec, arg_enum: str) -> Iterable[str]:
     values = spec.arg_enums[arg_enum]
     enum_name = get_python_enum_class(arg_enum)
-    yield f"class {enum_name}(enum.StrEnum):"
+    yield f"class {enum_name}(str):"
     yield f'{INDENT}"""Available values for the `{arg_enum}` enum"""'
     for value in values:
-        yield f"{INDENT}{value.name} = ..."
+        yield f"{INDENT}{value.name}: {enum_name} = ..."
     yield ""
 
 
@@ -903,7 +906,6 @@ def output_stub(
     class_ops: list[ClassDef],
 ) -> None:
     stub: list[str] = [
-        "import enum",
         "import typing",
         "",
         f"from puyapy import {CLS_ACCOUNT}, {CLS_BIGINT}, {CLS_BYTES}, {CLS_UINT64}",

scripts/vendor_mypy.py

@@ -8,6 +8,18 @@
 
 MYPY_REPO = "https://github.com/python/mypy.git"
 VCS_ROOT = Path(__file__).parent.parent
+TYPESHED_README = """
+This is PuyaPy's custom typeshed, which is a curated subset of the official MyPy typeshed.
+It only includes the required stubs used by PuyaPy as this speeds up MyPy's parsing speed
+significantly.
+
+However this means certain python modules such as `enum` or `dataclasses` cannot be used in
+PuyaPy stubs unless this typeshed is updated.
+
+The contents of the typeshed are populated by the `scripts/vendor_mypy.py` script, which is used
+to vendor new versions of MyPy or to update the stubs included in this typeshed. So to add new
+stubs, update that script and rerun.
+""".strip()
 
 
 def clone_branch(version: str) -> str:
@@ -63,7 +75,7 @@ def update_puya_typeshed(mypy_typeshed: Path, puya_typeshed: Path) -> None:
         stdlib / "sys.pyi",
         stdlib / "abc.pyi",
         # needed for puyapy
-        stdlib / "enum.pyi",
+        # stdlib / "enum.pyi"
     ]
 
     (puya_typeshed / stdlib).mkdir(exist_ok=True, parents=True)
@@ -77,6 +89,7 @@ def update_puya_typeshed(mypy_typeshed: Path, puya_typeshed: Path) -> None:
             copy_dst.parent.mkdir(exist_ok=True, parents=True)
             shutil.copy(copy_src, copy_dst)
     (puya_typeshed / stdlib / "collections" / "__init__.pyi").touch()
+    (puya_typeshed / "README.md").write_text(TYPESHED_README)
 
 
 if __name__ == "__main__":

src/puya/_typeshed/README.md

@@ -0,0 +1,10 @@
+This is PuyaPy's custom typeshed, which is a curated subset of the official MyPy typeshed.
+It only includes the required stubs used by PuyaPy as this speeds up MyPy's parsing speed 
+significantly.
+
+However this means certain python modules such as `enum` or `dataclasses` cannot be used in 
+PuyaPy stubs unless this typeshed is updated.
+
+The contents of the typeshed are populated by the `scripts/vendor_mypy.py` script, which is used
+to vendor new versions of MyPy or to update the stubs included in this typeshed. So to add new
+stubs, update that script and rerun.