Compare commits
6 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
| a920352407 | |||
| bd38dfacc7 | |||
| 281483f0fc | |||
| 932838de8f | |||
| a0b39e4c64 | |||
| 759cd0ef58 |
@@ -206,6 +206,7 @@ class AutomationContext:
|
||||
query = gql(
|
||||
"""
|
||||
mutation AutomateFunctionRunStatusReport(
|
||||
$projectId: String!
|
||||
$functionRunId: String!
|
||||
$status: AutomateRunStatus!
|
||||
$statusMessage: String
|
||||
@@ -213,6 +214,7 @@ class AutomationContext:
|
||||
$contextView: String
|
||||
){
|
||||
automateFunctionRunStatusReport(input: {
|
||||
projectId: $projectId
|
||||
functionRunId: $functionRunId
|
||||
status: $status
|
||||
statusMessage: $statusMessage
|
||||
@@ -236,6 +238,7 @@ class AutomationContext:
|
||||
object_results = None
|
||||
|
||||
params = {
|
||||
"projectId": self.automation_run_data.project_id,
|
||||
"functionRunId": self.automation_run_data.function_run_id,
|
||||
"status": self.run_status.value,
|
||||
"statusMessage": self._automation_result.status_message,
|
||||
@@ -355,6 +358,24 @@ class AutomationContext:
|
||||
visual_overrides,
|
||||
)
|
||||
|
||||
def attach_success_to_objects(
|
||||
self,
|
||||
category: str,
|
||||
object_ids: Union[str, List[str]],
|
||||
message: Optional[str] = None,
|
||||
metadata: Optional[Dict[str, Any]] = None,
|
||||
visual_overrides: Optional[Dict[str, Any]] = None,
|
||||
) -> None:
|
||||
"""Add a new success case to the run results."""
|
||||
self.attach_result_to_objects(
|
||||
ObjectResultLevel.SUCCESS,
|
||||
category,
|
||||
object_ids,
|
||||
message,
|
||||
metadata,
|
||||
visual_overrides,
|
||||
)
|
||||
|
||||
def attach_info_to_objects(
|
||||
self,
|
||||
category: str,
|
||||
|
||||
@@ -69,6 +69,7 @@ class AutomationStatus(str, Enum):
|
||||
class ObjectResultLevel(str, Enum):
|
||||
"""Possible status message levels for object reports."""
|
||||
|
||||
SUCCESS = "SUCCESS"
|
||||
INFO = "INFO"
|
||||
WARNING = "WARNING"
|
||||
ERROR = "ERROR"
|
||||
|
||||
@@ -49,4 +49,4 @@ class ProjectUpdateRoleInput(BaseModel):
|
||||
|
||||
class UserProjectsFilter(BaseModel):
|
||||
search: str
|
||||
onlyWithRole: Optional[Sequence[str]] = None
|
||||
onlyWithRoles: Optional[Sequence[str]] = None
|
||||
|
||||
@@ -5,16 +5,19 @@ from pydantic import BaseModel
|
||||
|
||||
class UpdateVersionInput(BaseModel):
|
||||
versionId: str
|
||||
projectId: str
|
||||
message: Optional[str]
|
||||
|
||||
|
||||
class MoveVersionsInput(BaseModel):
|
||||
targetModelName: str
|
||||
versionIds: Sequence[str]
|
||||
projectId: str
|
||||
|
||||
|
||||
class DeleteVersionsInput(BaseModel):
|
||||
versionIds: Sequence[str]
|
||||
projectId: str
|
||||
|
||||
|
||||
class CreateVersionInput(BaseModel):
|
||||
|
||||
@@ -303,19 +303,9 @@ class RevitParameter(Base, speckle_type="Objects.BuiltElements.Revit.Parameter")
|
||||
isTypeParameter: bool = False
|
||||
|
||||
|
||||
@deprecated(version="2.20", reason="Namespace changed, collectionType deprecated")
|
||||
class Collection(
|
||||
Base, speckle_type="Speckle.Core.Models.Collection", detachable={"elements"}
|
||||
):
|
||||
name: Optional[str] = None
|
||||
collectionType: Optional[str] = None
|
||||
elements: Optional[List[Base]] = None
|
||||
|
||||
|
||||
class Collection( # noqa: F811
|
||||
Base,
|
||||
speckle_type="Speckle.Core.Models.Collections.Collection",
|
||||
detachable={"elements"},
|
||||
):
|
||||
name: Optional[str] = None
|
||||
elements: Optional[List[Base]]
|
||||
|
||||
@@ -103,9 +103,13 @@ class TestVersionResource:
|
||||
assert result.versions.totalCount == 1
|
||||
assert result.versions.items[0].id == test_version.id
|
||||
|
||||
def test_version_update(self, client: SpeckleClient, test_version: Version):
|
||||
def test_version_update(
|
||||
self, client: SpeckleClient, test_version: Version, test_project: Project
|
||||
):
|
||||
new_message = "MY new version message"
|
||||
input = UpdateVersionInput(versionId=test_version.id, message=new_message)
|
||||
input = UpdateVersionInput(
|
||||
versionId=test_version.id, projectId=test_project.id, message=new_message
|
||||
)
|
||||
updated_version = client.version.update(input)
|
||||
|
||||
assert isinstance(updated_version, Version)
|
||||
@@ -121,7 +125,9 @@ class TestVersionResource:
|
||||
test_model_2: Model,
|
||||
):
|
||||
input = MoveVersionsInput(
|
||||
targetModelName=test_model_2.name, versionIds=[test_version.id]
|
||||
targetModelName=test_model_2.name,
|
||||
versionIds=[test_version.id],
|
||||
projectId=test_project.id,
|
||||
)
|
||||
moved_model_id = client.version.move_to_model(input)
|
||||
|
||||
@@ -137,7 +143,9 @@ class TestVersionResource:
|
||||
def test_version_delete(
|
||||
self, client: SpeckleClient, test_version: Version, test_project: Project
|
||||
):
|
||||
input = DeleteVersionsInput(versionIds=[test_version.id])
|
||||
input = DeleteVersionsInput(
|
||||
versionIds=[test_version.id], projectId=test_project.id
|
||||
)
|
||||
|
||||
response = client.version.delete(input)
|
||||
assert response is True
|
||||
|
||||
Reference in New Issue
Block a user