Skip to content

Commit

Permalink
feat(area): rename area_name to area_id
Browse files Browse the repository at this point in the history
  • Loading branch information
TheoPascoli committed Jan 30, 2025
1 parent 6ba034b commit ea1f96c
Show file tree
Hide file tree
Showing 4 changed files with 10 additions and 10 deletions.
2 changes: 1 addition & 1 deletion antarest/study/business/area_management.py
Original file line number Diff line number Diff line change
Expand Up @@ -684,7 +684,7 @@ def update_area_ui(self, study: Study, area_id: str, area_ui: UpdateAreaUi, laye
file_study = self.storage_service.get_storage(study).get_raw(study)

command = MoveArea(
area_name=area_id,
area_id=area_id,
new_area_parameters=area_ui.model_dump(),
layer=layer,
command_context=self.storage_service.variant_study_service.command_factory.command_context,
Expand Down
14 changes: 7 additions & 7 deletions antarest/study/storage/variantstudy/model/command/move_area.py
Original file line number Diff line number Diff line change
Expand Up @@ -36,7 +36,7 @@ class MoveArea(ICommand):
# Command parameters
# ==================

area_name: str
area_id: str
new_area_parameters: t.Dict[str, t.Any]
layer: str

Expand All @@ -45,14 +45,14 @@ def _apply_config(self, study_data: FileStudyTreeConfig) -> t.Tuple[CommandOutpu
return (
CommandOutput(
status=True,
message=f"area '{self.area_name}' updated",
message=f"area '{self.area_id}' updated",
),
{},
)

@override
def _apply(self, study_data: FileStudy, listener: t.Optional[ICommandListener] = None) -> CommandOutput:
current_area = study_data.tree.get(["input", "areas", self.area_name, "ui"])
current_area = study_data.tree.get(["input", "areas", self.area_id, "ui"])

if self.layer == "0":
ui = current_area["ui"]
Expand All @@ -63,7 +63,7 @@ def _apply(self, study_data: FileStudy, listener: t.Optional[ICommandListener] =
current_area["layerY"][self.layer] = self.new_area_parameters["y"]
current_area["layerColor"][self.layer] = ",".join(map(str, self.new_area_parameters["color_rgb"]))

study_data.tree.save(current_area, ["input", "areas", self.area_name, "ui"])
study_data.tree.save(current_area, ["input", "areas", self.area_id, "ui"])

output, _ = self._apply_config(study_data.config)

Expand All @@ -73,19 +73,19 @@ def _apply(self, study_data: FileStudy, listener: t.Optional[ICommandListener] =
def to_dto(self) -> CommandDTO:
return CommandDTO(
action=CommandName.MOVE_AREA.value,
args={"area_name": self.area_name, "new_area_parameters": self.new_area_parameters, "layer": self.layer},
args={"area_id": self.area_id, "new_area_parameters": self.new_area_parameters, "layer": self.layer},
study_version=self.study_version,
)

@override
def match_signature(self) -> str:
return str(self.command_name.value + MATCH_SIGNATURE_SEPARATOR + self.area_name)
return str(self.command_name.value + MATCH_SIGNATURE_SEPARATOR + self.area_id)

@override
def match(self, other: ICommand, equal: bool = False) -> bool:
if not isinstance(other, MoveArea):
return False
return self.area_name == other.area_name
return self.area_id == other.area_id

@override
def _create_diff(self, other: "ICommand") -> t.List["ICommand"]:
Expand Down
2 changes: 1 addition & 1 deletion tests/storage/business/test_arealink_manager.py
Original file line number Diff line number Diff line change
Expand Up @@ -185,7 +185,7 @@ def test_area_crud(empty_study: FileStudy, matrix_service: SimpleMatrixService):
id=None,
action=CommandName.MOVE_AREA.value,
args={
"area_name": "test",
"area_id": "test",
"new_area_parameters": {
"x": 100,
"y": 200,
Expand Down
2 changes: 1 addition & 1 deletion tests/variantstudy/test_command_factory.py
Original file line number Diff line number Diff line change
Expand Up @@ -41,7 +41,7 @@
CommandDTO(action=CommandName.REMOVE_AREA.value, args=[{"id": "id"}], study_version=STUDY_VERSION_8_8),
CommandDTO(
action=CommandName.MOVE_AREA.value,
args={"area_name": "id", "new_area_parameters": {}, "layer": "0"},
args={"area_id": "id", "new_area_parameters": {}, "layer": "0"},
study_version=STUDY_VERSION_8_8,
),
CommandDTO(
Expand Down

0 comments on commit ea1f96c

Please sign in to comment.