infrahub-server 1.6.3__py3-none-any.whl → 1.7.0b0__py3-none-any.whl

infrahub/core/graph/__init__.py

@@ -1 +1 @@
-GRAPH_VERSION = 48
+GRAPH_VERSION = 50

infrahub/core/graph/constraints.py

@@ -176,12 +176,12 @@ class ConstraintManagerBase:
         return self.nodes + self.rels
 
     @classmethod
-    def from_graph_schema(cls, db: InfrahubDatabase, schema: dict[str, dict[str, BaseModel]]) -> Self:
+    def from_graph_schema(cls, db: InfrahubDatabase, schema: dict[str, dict[str, type[BaseModel]]]) -> Self:
         manager = cls(db=db)
 
         # Process the nodes first
         for schema_item in schema["nodes"].values():
-            properties_class: BaseModel = schema_item.model_fields["properties"].annotation  # type: ignore[assignment]
+            properties_class: type[BaseModel] = schema_item.model_fields["properties"].annotation  # type: ignore[assignment]
             default_label = str(schema_item.model_fields["default_label"].default)
             for field_name, field in properties_class.model_fields.items():
                 clean_field_name = field.alias or field_name

infrahub/core/graph/schema.py

@@ -95,9 +95,6 @@ class GraphRelationshipRelationships(BaseModel):
     HAS_OWNER: GraphRelationship = Field(
         GraphRelationship(peer="Node", direction=GraphRelDirection.OUTBOUND), description=""
     )
-    IS_VISIBLE: GraphRelationship = Field(
-        GraphRelationship(peer="Boolean", direction=GraphRelDirection.OUTBOUND), description=""
-    )
     IS_PROTECTED: GraphRelationship = Field(
         GraphRelationship(peer="Boolean", direction=GraphRelDirection.OUTBOUND), description=""
     )
@@ -133,9 +130,6 @@ class GraphAttributeRelationships(BaseModel):
     HAS_OWNER: GraphRelationship = Field(
         GraphRelationship(peer="Node", direction=GraphRelDirection.OUTBOUND), description=""
     )
-    IS_VISIBLE: GraphRelationship = Field(
-        GraphRelationship(peer="Boolean", direction=GraphRelDirection.OUTBOUND), description=""
-    )
     IS_PROTECTED: GraphRelationship = Field(
         GraphRelationship(peer="Boolean", direction=GraphRelDirection.OUTBOUND), description=""
     )
@@ -208,9 +202,6 @@ class GraphBooleanProperties(BaseModel):
 
 
 class GraphBooleanRelationships(BaseModel):
-    IS_VISIBLE: GraphRelationship = Field(
-        GraphRelationship(peer="Attribute|Relationship", direction=GraphRelDirection.INBOUND), description=""
-    )
     IS_PROTECTED: GraphRelationship = Field(
         GraphRelationship(peer="Attribute|Relationship", direction=GraphRelDirection.INBOUND), description=""
     )
@@ -243,7 +234,7 @@ class GraphRelationshipDefault(BaseModel):
     hierarchy: Optional[str] = Field(None, description="Name of the hierarchy this relationship is part of")
 
 
-def get_graph_schema() -> dict[str, dict[str, Any]]:
+def get_graph_schema() -> dict[str, dict[str, type[Any]]]:
     """Generate the graph schema dictionary containing nodes and relationships."""
     graph_node_list: list[type[GraphVertex]] = [
         GraphNodeNode,
@@ -265,10 +256,9 @@ def get_graph_schema() -> dict[str, dict[str, Any]]:
             "IS_RELATED": GraphRelationshipDefault,
             "HAS_SOURCE": GraphRelationshipDefault,
             "HAS_OWNER": GraphRelationshipDefault,
-            "IS_VISIBLE": GraphRelationshipDefault,
             "IS_PROTECTED": GraphRelationshipDefault,
         },
     }
 
 
-GRAPH_SCHEMA: dict[str, dict[str, Any]] = get_graph_schema()
+GRAPH_SCHEMA: dict[str, dict[str, type[Any]]] = get_graph_schema()