Merged PR 378: Add create file/folder and file access requests
## Summary Added requests in file system that allows: - file to be created - folder to be created - file to be "accessed" ## Test process https://dev.azure.com/ma-dev-uk/PrimAITE/_git/PrimAITE/pullrequest/378?_a=files&path=/tests/integration_tests/test_simulation/test_request_response.py ## Checklist - [X] PR is linked to a **work item** - [X] **acceptance criteria** of linked ticket are met - [X] performed **self-review** of the code - [X] written **tests** for any new functionality added with this PR - [ ] updated the **documentation** if this PR changes or adds functionality - [ ] written/updated **design docs** if this PR implements new functionality - [ ] updated the **change log** - [X] ran **pre-commit** checks for code style - [ ] attended to any **TO-DOs** left in the code Related work items: #2606
This commit is contained in:
@@ -1,7 +1,7 @@
|
||||
from __future__ import annotations
|
||||
|
||||
from pathlib import Path
|
||||
from typing import Dict, Optional
|
||||
from typing import Any, Dict, List, Optional
|
||||
|
||||
from prettytable import MARKDOWN, PrettyTable
|
||||
|
||||
@@ -63,6 +63,65 @@ class FileSystem(SimComponent):
|
||||
request_type=RequestType(func=self._delete_manager),
|
||||
)
|
||||
|
||||
self._create_manager = RequestManager()
|
||||
|
||||
def _create_file_action(request: List[Any], context: Any) -> RequestResponse:
|
||||
file = self.create_file(folder_name=request[0], file_name=request[1])
|
||||
if not file:
|
||||
return RequestResponse.from_bool(False)
|
||||
return RequestResponse(
|
||||
status="success",
|
||||
data={
|
||||
"file_name": file.name,
|
||||
"folder_name": file.folder_name,
|
||||
"file_type": file.file_type,
|
||||
"file_size": file.size,
|
||||
},
|
||||
)
|
||||
|
||||
self._create_manager.add_request(
|
||||
name="file",
|
||||
request_type=RequestType(func=_create_file_action),
|
||||
)
|
||||
|
||||
def _create_folder_action(request: List[Any], context: Any) -> RequestResponse:
|
||||
folder = self.create_folder(folder_name=request[0])
|
||||
if not folder:
|
||||
return RequestResponse.from_bool(False)
|
||||
return RequestResponse(status="success", data={"folder_name": folder.name})
|
||||
|
||||
self._create_manager.add_request(
|
||||
name="folder",
|
||||
request_type=RequestType(func=_create_folder_action),
|
||||
)
|
||||
rm.add_request(
|
||||
name="create",
|
||||
request_type=RequestType(func=self._create_manager),
|
||||
)
|
||||
|
||||
def _access_file_action(request: List[Any], context: Any) -> RequestResponse:
|
||||
file = self.get_file(folder_name=request[0], file_name=request[1])
|
||||
if not file:
|
||||
return RequestResponse.from_bool(False)
|
||||
|
||||
if self.access_file(folder_name=request[0], file_name=request[1]):
|
||||
return RequestResponse(
|
||||
status="success",
|
||||
data={
|
||||
"file_name": file.name,
|
||||
"folder_name": file.folder_name,
|
||||
"file_type": file.file_type,
|
||||
"file_size": file.size,
|
||||
"file_status": file.health_status,
|
||||
},
|
||||
)
|
||||
return RequestResponse.from_bool(False)
|
||||
|
||||
rm.add_request(
|
||||
name="access",
|
||||
request_type=RequestType(func=_access_file_action),
|
||||
)
|
||||
|
||||
self._restore_manager = RequestManager()
|
||||
self._restore_manager.add_request(
|
||||
name="file",
|
||||
@@ -494,3 +553,28 @@ class FileSystem(SimComponent):
|
||||
return False
|
||||
|
||||
return folder.restore_file(file_name=file_name)
|
||||
|
||||
def access_file(self, folder_name: str, file_name: str) -> bool:
|
||||
"""
|
||||
Access a file.
|
||||
|
||||
Used by agents to simulate a file being accessed.
|
||||
|
||||
:param: folder_name: name of the folder where the file is stored
|
||||
:type: folder_name: str
|
||||
|
||||
:param: file_name: name of the file to access
|
||||
:type: file_name: str
|
||||
"""
|
||||
folder = self.get_folder(folder_name=folder_name)
|
||||
|
||||
if folder:
|
||||
file = folder.get_file(file_name=file_name)
|
||||
|
||||
if file:
|
||||
file.num_access += 1
|
||||
return True
|
||||
else:
|
||||
self.sys_log.error(f"Unable to access file that does not exist. (file name: {file_name})")
|
||||
|
||||
return False
|
||||
|
||||
Reference in New Issue
Block a user