Merge pull request #14 from ThemeParks/fix/regen-auto-format

fix(scripts): auto-format generated models in regenerate.py

Author: cubehouse

scripts/regenerate.py

@@ -161,6 +161,14 @@ def main() -> None:
     else:
         print("  no changes (already patched?)")
 
+    # Auto-format the generated file so `ruff format --check` in CI doesn't
+    # fail on quote-style or whitespace differences from datamodel-codegen.
+    print("Formatting generated models with ruff...")
+    subprocess.run(
+        [sys.executable, "-m", "ruff", "format", str(OUTPUT)],
+        check=True,
+    )
+
     # Post-generation sanity check. The running process already imported the
     # stale pre-patch module (if at all), so spawn a subprocess for a fresh
     # import that reflects what downstream users will see.

themeparks/_generated/models.py

@@ -10,31 +10,31 @@
 
 
 class EntityType(Enum):
-    DESTINATION = 'DESTINATION'
-    PARK = 'PARK'
-    ATTRACTION = 'ATTRACTION'
-    RESTAURANT = 'RESTAURANT'
-    HOTEL = 'HOTEL'
-    SHOW = 'SHOW'
+    DESTINATION = "DESTINATION"
+    PARK = "PARK"
+    ATTRACTION = "ATTRACTION"
+    RESTAURANT = "RESTAURANT"
+    HOTEL = "HOTEL"
+    SHOW = "SHOW"
 
 
 class LiveStatusType(Enum):
-    OPERATING = 'OPERATING'
-    DOWN = 'DOWN'
-    CLOSED = 'CLOSED'
-    REFURBISHMENT = 'REFURBISHMENT'
+    OPERATING = "OPERATING"
+    DOWN = "DOWN"
+    CLOSED = "CLOSED"
+    REFURBISHMENT = "REFURBISHMENT"
 
 
 class ReturnTimeState(Enum):
-    AVAILABLE = 'AVAILABLE'
-    TEMP_FULL = 'TEMP_FULL'
-    FINISHED = 'FINISHED'
+    AVAILABLE = "AVAILABLE"
+    TEMP_FULL = "TEMP_FULL"
+    FINISHED = "FINISHED"
 
 
 class BoardingGroupState(Enum):
-    AVAILABLE = 'AVAILABLE'
-    PAUSED = 'PAUSED'
-    CLOSED = 'CLOSED'
+    AVAILABLE = "AVAILABLE"
+    PAUSED = "PAUSED"
+    CLOSED = "CLOSED"
 
 
 class PriceData(BaseModel):
@@ -157,9 +157,9 @@ class EntityLiveDataResponse(BaseModel):
 
 
 class Type(Enum):
-    ADMISSION = 'ADMISSION'
-    PACKAGE = 'PACKAGE'
-    ATTRACTION = 'ATTRACTION'
+    ADMISSION = "ADMISSION"
+    PACKAGE = "PACKAGE"
+    ATTRACTION = "ATTRACTION"
 
 
 class SchedulePriceObject(BaseModel):
@@ -171,11 +171,11 @@ class SchedulePriceObject(BaseModel):
 
 
 class Type1(Enum):
-    OPERATING = 'OPERATING'
-    TICKETED_EVENT = 'TICKETED_EVENT'
-    PRIVATE_EVENT = 'PRIVATE_EVENT'
-    EXTRA_HOURS = 'EXTRA_HOURS'
-    INFO = 'INFO'
+    OPERATING = "OPERATING"
+    TICKETED_EVENT = "TICKETED_EVENT"
+    PRIVATE_EVENT = "PRIVATE_EVENT"
+    EXTRA_HOURS = "EXTRA_HOURS"
+    INFO = "INFO"
 
 
 class ScheduleEntry(BaseModel):