Coverage for src/app/controllers/add_file_controller.py: 58%

1from dto.file_dto import FileDTO 

2from usecases.add_file_useCase import AddFileUseCase 

3from models.file_model import FileModel 

4 

5 

6class AddFileController: 

7 """ 

8 Controller for adding a file to the database. 

9 """ 

10 

11 def __init__(self, add_file_usecase: AddFileUseCase): 

12 

13 self.add_file_usecase = add_file_usecase 

14 

15 def load_file(self, file: FileDTO): 

16 """ 

17 Add a file to the database. 

18 Args: 

19 file (FileDTO): The file data transfer object containing file details. 

20 

21 Returns: 

22 None 

23 """ 

24 try: 

25 file_model = FileModel(file.get_file_name(), file.get_file_content()) 

26 self.add_file_usecase.load_file(file_model) 

27 except Exception as e: 

28 raise e