Support Blueprint boolean constraints (#619)

Author: razor-x

codegen/layouts/resource.hbs

@@ -1,4 +1,4 @@
-from typing import Any, Dict, List, Optional, Union
+from typing import Any, Dict, List, Literal, Optional, Union
 from dataclasses import dataclass
 from ..deep_attr_dict import DeepAttrDict
 from ..resource_mapping import ResourceMapping

codegen/layouts/route.hbs

@@ -1,4 +1,4 @@
-from typing import Optional, Any, List, Dict, Union
+from typing import Optional, Any, List, Dict, Literal, Union
 import abc
 from ..client import SeamHttpClient
 from ..route import route_metadata

codegen/lib/layouts/resources.ts

@@ -107,6 +107,19 @@ const mergeOccurrences = (occurrences: Property[], path: string): Property => {
     )
   }
 
+  if (first.format === 'boolean') {
+    const booleans = occurrences as Array<
+      Extract<Property, { format: 'boolean' }>
+    >
+    const values = booleans.some(({ values }) => values == null)
+      ? undefined
+      : [...new Set(booleans.flatMap(({ values }) => values ?? []))]
+    const merged = { ...first, ...docs }
+    if (values == null) delete merged.values
+    else merged.values = values
+    return merged
+  }
+
   if (first.format === 'object') {
     return {
       ...first,

codegen/lib/python-type.ts

@@ -18,6 +18,10 @@ export const mapParameterToPythonType = (parameter: Parameter): string => {
     return parameter.isInt ? 'int' : 'float'
   }
 
+  if (parameter.format === 'boolean') {
+    return mapBooleanToPythonType(parameter.values)
+  }
+
   return mapScalarFormatToPythonType(parameter.format)
 }
 
@@ -51,6 +55,10 @@ export const mapRequiredPropertyToPythonType = (
     return property.isInt ? 'int' : 'float'
   }
 
+  if (property.format === 'boolean') {
+    return mapBooleanToPythonType(property.values)
+  }
+
   // Batch resource properties are lists of the named resource on the wire,
   // though the blueprint types them as records.
   if (property.format === 'record' && 'resourceType' in property) {
@@ -64,6 +72,11 @@ export const mapRequiredPropertyToPythonType = (
   return mapScalarFormatToPythonType(property.format)
 }
 
+const mapBooleanToPythonType = (values?: boolean[]): string =>
+  values == null || values.length === 0
+    ? 'bool'
+    : `Literal[${values.map((value) => (value ? 'True' : 'False')).join(', ')}]`
+
 const mapScalarFormatToPythonType = (format: ScalarFormat): string => {
   switch (format) {
     case 'string':

package-lock.json

@@ -28,7 +28,7 @@
     }
   },
   "devDependencies": {
-    "@seamapi/blueprint": "^1.5.1",
+    "@seamapi/blueprint": "^1.6.0",
     "@seamapi/fake-seam-connect": "1.86.0",
     "@seamapi/smith": "^1.1.0",
     "@seamapi/types": "1.1001.0",