Check kind of type in Python (#909)

Moves the kind of type detection from the query for reflection from
EdgeQL to Python because the computed property use in the query turned
out to be too costly:
```
─────────────────────────── Query ───────────────────────────
analyze
with
    module schema
select ➊  Type {
    ➋  kind := assert_exists(
        'Object' IF Type IS ObjectType ELSE
        ➌  'Scalar' IF Type IS ScalarType ELSE
        ➍  'Array' IF Type IS Array ELSE
        'NamedTuple' IF ➎  Type[IS Tuple].named ?? false ELSE
        ➏  'Tuple' IF Type IS Tuple ELSE
        ➐  'Range' IF Type IS Range ELSE
        ➑  'MultiRange' IF Type IS MultiRange ELSE
        ➒  'Pseudo' IF Type IS PseudoType ELSE
        <str>{},
        message := "unexpected type",
    ),
};

────────────────────────────────────────────── Coarse-grained Query Plan ──────────────────────────────────────────────
     │    Time        Cost  Loops   Rows  Width │ Relations
root │  2282.4   837696.69    1.0  248.0     32 │ schema::ArrayExprAlias, schema::Tuple, schema::ScalarType,
     │                                          │ schema::Range, schema::RangeExprAlias, schema::MultiRange,
     │                                          │ schema::TupleExprAlias, schema::PseudoType, schema::ObjectType,
     │                                          │ schema::MultiRangeExprAlias, schema::Array
```

vs
```
──────────────────────── Query ────────────────────────
analyze
with
    module schema
select ➊  Type {
    ➋  is_object := Type IS ObjectType,
    ➌  is_scalar := Type IS ScalarType,
    ➍  is_array := Type IS Array,
    ➎  is_named_tuple := Type[IS Tuple].named ?? false,
    ➏  is_tuple := Type IS Tuple,
    ➐  is_range := Type IS Range,
    ➑  is_multi_range := Type IS MultiRange,
    ➒  is_pseudo := Type IS PseudoType,
};

────────────────────────────────────────────── Coarse-grained Query Plan ──────────────────────────────────────────────
     │   Time     Cost  Loops   Rows  Width │ Relations
root │  210.6  6316.56    1.0  248.0     32 │ schema::ArrayExprAlias, schema::Tuple, schema::ScalarType, schema::Range,
     │                                      │ schema::RangeExprAlias, schema::MultiRange, schema::TupleExprAlias,
     │                                      │ schema::PseudoType, schema::ObjectType, schema::MultiRangeExprAlias,
     │                                      │ schema::Array
```

Author: nsidnev

gel/_internal/_reflection/__init__.py

@@ -83,6 +83,7 @@
     is_range_type,
     is_scalar_type,
     is_tuple_type,
+    kind_of_type,
 )
 
 from ._modules import (
@@ -147,4 +148,5 @@
     "is_range_type",
     "is_scalar_type",
     "is_tuple_type",
+    "kind_of_type",
 )

gel/_internal/_reflection/_query.py

@@ -164,23 +164,19 @@
     )
 
 SELECT Type {
-    kind := assert_exists(
-        'Object' IF Type IS ObjectType ELSE
-        'Scalar' IF Type IS ScalarType ELSE
-        'Array' IF Type IS Array ELSE
-        'NamedTuple' IF Type[IS Tuple].named ?? false ELSE
-        'Tuple' IF Type IS Tuple ELSE
-        'Range' IF Type IS Range ELSE
-        'MultiRange' IF Type IS MultiRange ELSE
-        'Pseudo' IF Type IS PseudoType ELSE
-        <str>{},
-        message := "unexpected type",
-    ),
-
     id,
     builtin,
     internal,
 
+    is_object := Type IS ObjectType,
+    is_scalar := Type IS ScalarType,
+    is_array := Type IS Array,
+    is_named_tuple := Type[IS Tuple].named ?? false,
+    is_tuple := Type IS Tuple,
+    is_range := Type IS Range,
+    is_multi_range := Type IS MultiRange,
+    is_pseudo := Type IS PseudoType,
+
     name := (
         array_join(array_agg([IS ObjectType].union_of.name), ' | ')
         IF EXISTS [IS ObjectType].union_of

gel/_internal/_reflection/_types.py

@@ -48,10 +48,22 @@ def __str__(self) -> str:
 
 @sobject
 class Type(SchemaObject, abc.ABC):
-    kind: TypeKind
     builtin: bool
     internal: bool
 
+    is_object: bool
+    is_scalar: bool
+    is_array: bool
+    is_named_tuple: bool
+    is_tuple: bool
+    is_range: bool
+    is_multi_range: bool
+    is_pseudo: bool
+
+    @functools.cached_property
+    def kind(self) -> TypeKind:
+        return kind_of_type(self)
+
     def __str__(self) -> str:
         return self.name
 
@@ -221,7 +233,18 @@ def issubclass(self, parent: Self) -> bool:
 
 @struct
 class PseudoType(Type):
-    kind: Literal[TypeKind.Pseudo]
+    is_object: Literal[False]
+    is_scalar: Literal[False]
+    is_array: Literal[False]
+    is_named_tuple: Literal[False]
+    is_tuple: Literal[False]
+    is_range: Literal[False]
+    is_multi_range: Literal[False]
+    is_pseudo: Literal[True]
+
+    @functools.cached_property
+    def kind(self) -> Literal[TypeKind.Pseudo]:
+        return TypeKind.Pseudo
 
     @functools.cached_property
     def generic(self) -> bool:
@@ -246,23 +269,47 @@ def _assignable_from(
 
 @struct
 class ScalarType(InheritingType):
-    kind: Literal[TypeKind.Scalar]
     is_seq: bool
     enum_values: tuple[str, ...] | None = None
     material_id: str | None = None
     cast_type: str | None = None
 
+    is_object: Literal[False]
+    is_scalar: Literal[True]
+    is_array: Literal[False]
+    is_named_tuple: Literal[False]
+    is_tuple: Literal[False]
+    is_range: Literal[False]
+    is_multi_range: Literal[False]
+    is_pseudo: Literal[False]
+
+    @functools.cached_property
+    def kind(self) -> Literal[TypeKind.Scalar]:
+        return TypeKind.Scalar
+
 
 @struct
 class ObjectType(InheritingType):
-    kind: Literal[TypeKind.Object]
     union_of: tuple[TypeRef, ...]
     intersection_of: tuple[TypeRef, ...]
     compound_type: bool
     # N.B: We store Pointers sorted by name, from the
     # introspection query
     pointers: tuple[Pointer, ...]
 
+    is_object: Literal[True]
+    is_scalar: Literal[False]
+    is_array: Literal[False]
+    is_named_tuple: Literal[False]
+    is_tuple: Literal[False]
+    is_range: Literal[False]
+    is_multi_range: Literal[False]
+    is_pseudo: Literal[False]
+
+    @functools.cached_property
+    def kind(self) -> Literal[TypeKind.Object]:
+        return TypeKind.Object
+
     def get_pointer(self, name: str) -> Pointer:
         for ptr in self.pointers:
             if ptr.name == name:
@@ -506,9 +553,21 @@ def _assignable_from(
 
 @struct
 class ArrayType(HomogeneousCollectionType):
-    kind: Literal[TypeKind.Array]
     array_element_id: str
 
+    is_object: Literal[False]
+    is_scalar: Literal[False]
+    is_array: Literal[True]
+    is_named_tuple: Literal[False]
+    is_tuple: Literal[False]
+    is_range: Literal[False]
+    is_multi_range: Literal[False]
+    is_pseudo: Literal[False]
+
+    @functools.cached_property
+    def kind(self) -> Literal[TypeKind.Array]:
+        return TypeKind.Array
+
     @functools.cached_property
     def _element_type_id(self) -> str:
         return self.array_element_id
@@ -520,7 +579,19 @@ def get_id_and_name(self, element_type: Type) -> tuple[str, str]:
 
 @struct
 class RangeType(HomogeneousCollectionType):
-    kind: Literal[TypeKind.Range]
+    is_object: Literal[False]
+    is_scalar: Literal[False]
+    is_array: Literal[False]
+    is_named_tuple: Literal[False]
+    is_tuple: Literal[False]
+    is_range: Literal[True]
+    is_multi_range: Literal[False]
+    is_pseudo: Literal[False]
+
+    @functools.cached_property
+    def kind(self) -> Literal[TypeKind.Range]:
+        return TypeKind.Range
+
     range_element_id: str
 
     @functools.cached_property
@@ -534,7 +605,19 @@ def get_id_and_name(self, element_type: Type) -> tuple[str, str]:
 
 @struct
 class MultiRangeType(HomogeneousCollectionType):
-    kind: Literal[TypeKind.MultiRange]
+    is_object: Literal[False]
+    is_scalar: Literal[False]
+    is_array: Literal[False]
+    is_named_tuple: Literal[False]
+    is_tuple: Literal[False]
+    is_range: Literal[False]
+    is_multi_range: Literal[True]
+    is_pseudo: Literal[False]
+
+    @functools.cached_property
+    def kind(self) -> Literal[TypeKind.MultiRange]:
+        return TypeKind.MultiRange
+
     multirange_element_id: str
 
     @functools.cached_property
@@ -563,7 +646,18 @@ def _element_type_ids(self) -> list[str]:
 
 @struct
 class TupleType(_TupleType):
-    kind: Literal[TypeKind.Tuple]
+    is_object: Literal[False]
+    is_scalar: Literal[False]
+    is_array: Literal[False]
+    is_named_tuple: Literal[False]
+    is_tuple: Literal[True]
+    is_range: Literal[False]
+    is_multi_range: Literal[False]
+    is_pseudo: Literal[False]
+
+    @functools.cached_property
+    def kind(self) -> Literal[TypeKind.Tuple]:
+        return TypeKind.Tuple
 
     def get_id_and_name(
         self, element_types: tuple[Type, ...]
@@ -576,7 +670,18 @@ def get_id_and_name(
 
 @struct
 class NamedTupleType(_TupleType):
-    kind: Literal[TypeKind.NamedTuple]
+    is_object: Literal[False]
+    is_scalar: Literal[False]
+    is_array: Literal[False]
+    is_named_tuple: Literal[True]
+    is_tuple: Literal[False]
+    is_range: Literal[False]
+    is_multi_range: Literal[False]
+    is_pseudo: Literal[False]
+
+    @functools.cached_property
+    def kind(self) -> Literal[TypeKind.NamedTuple]:
+        return TypeKind.NamedTuple
 
     def get_id_and_name(
         self, element_types: tuple[Type, ...]
@@ -661,7 +766,7 @@ def get_nearest_common_ancestors(
     | MultiRangeType
 )
 
-AnyType = PrimitiveType | PseudoType | ObjectType
+AnyType = PrimitiveType | ObjectType | PseudoType
 
 Types = dict[str, Type]
 
@@ -722,6 +827,27 @@ def is_primitive_type(t: Type) -> TypeGuard[PrimitiveType]:
     return not isinstance(t, (ObjectType, PseudoType))
 
 
+def kind_of_type(ty: Type) -> TypeKind:
+    if ty.is_object:
+        return TypeKind.Object
+    elif ty.is_scalar:
+        return TypeKind.Scalar
+    elif ty.is_array:
+        return TypeKind.Array
+    elif ty.is_named_tuple:
+        return TypeKind.NamedTuple
+    elif ty.is_tuple:
+        return TypeKind.Tuple
+    elif ty.is_range:
+        return TypeKind.Range
+    elif ty.is_multi_range:
+        return TypeKind.MultiRange
+    elif ty.is_pseudo:
+        return TypeKind.Pseudo
+    else:
+        raise ValueError("unexpected type")
+
+
 @sobject
 class Pointer(SchemaObject):
     card: Cardinality
@@ -752,7 +878,7 @@ def fetch_types(
     types: list[Type] = db.query(_query.TYPES, builtin=builtin)
     result = {}
     for t in types:
-        cls = _kind_to_class[t.kind]
+        cls = _kind_to_class[kind_of_type(t)]
         replace: dict[str, Any] = {}
         if issubclass(cls, CollectionType):
             replace["name"] = _edgeql.unmangle_unqual_name(t.name)