chore: remove data profiling and file response schemas

Desbordante · Sep 25, 2024 · 82f3a16 · 82f3a16
1 parent d36ab13
commit 82f3a16
Show file tree

Hide file tree

Showing 6 changed files with 4 additions and 15 deletions.
diff --git a/internal/dto/repository/file/__init__.py b/internal/dto/repository/file/__init__.py
@@ -1,6 +1,5 @@
 from internal.dto.repository.file.file import (  # noqa: F401
     File,
-    FileResponseSchema,
     FileFindSchema,
     FileCreateSchema,
     FileUpdateSchema,

diff --git a/internal/dto/repository/file/file.py b/internal/dto/repository/file/file.py
@@ -31,9 +31,6 @@ class FileUpdateSchema(FileBaseSchema, BaseUpdateSchema): ...
 class FileFindSchema(FileBaseSchema, BaseSchema): ...  # it's not a typo
 
 
-FileResponseSchema = None
-
-
 class CSVFileFindSchema(FileFindSchema):
     separator: str
     header: list[int]

diff --git a/internal/dto/worker/task/profiling_task.py b/internal/dto/worker/task/profiling_task.py
@@ -11,6 +11,3 @@ class ProfilingTaskBaseSchema(BaseModel):
 
 
 class ProfilingTaskCreateSchema(ProfilingTaskBaseSchema): ...
-
-
-ProfilingTaskResponseSchema = None
diff --git a/internal/repository/flat/file.py b/internal/repository/flat/file.py
@@ -8,7 +8,7 @@
     CSVFileFindSchema,
     CSVFileResponseSchema,
 )
-from internal.dto.repository.file import File, FileCreateSchema, FileResponseSchema
+from internal.dto.repository.file import File, FileCreateSchema
 from internal.infrastructure.data_storage.flat import FlatContext
 
 CHUNK_SIZE = 1024
@@ -22,7 +22,7 @@ async def create(
         file: File,
         file_info: FileCreateSchema,
         context: FlatContext,
-    ) -> FileResponseSchema:
+    ) -> None:
 
         path_to_file = Path.joinpath(
             context.upload_directory_path, str(file_info.file_name)

diff --git a/internal/usecase/file/save_file.py b/internal/usecase/file/save_file.py
@@ -6,7 +6,6 @@
 from internal.domain.file import File as FileEntity
 from internal.dto.repository.file import (
     FileCreateSchema,
-    FileResponseSchema,
     File,
     FailedFileReadingException,
 )
@@ -22,7 +21,7 @@ class FileRepo(Protocol):
 
     async def create(
         self, file: File, file_info: FileCreateSchema, context: DataStorageContext
-    ) -> FileResponseSchema: ...
+    ) -> None: ...
 
 
 class FileMetadataRepo(Protocol):

diff --git a/tests/usecase/test_save_file.py b/tests/usecase/test_save_file.py
@@ -10,7 +10,6 @@
     File,
     FileMetadataCreateSchema,
     FileCreateSchema,
-    FileResponseSchema,
     FailedFileReadingException,
 )
 from internal.uow import UnitOfWork, DataStorageContext
@@ -106,9 +105,7 @@ async def test_save_file(
         updated_at=updated_at,
     )
 
-    file_response = FileResponseSchema
-
-    file_repo_mock.create.return_value = file_response
+    file_repo_mock.create.return_value = None
     file_metadata_repo_mock.create.return_value = file_metadata_response
 
     upload_file_mock = mocker.Mock(spec=File)
Original file line number	Diff line number	Diff line change
Expand Up		@@ -11,6 +11,3 @@ class ProfilingTaskBaseSchema(BaseModel):


		class ProfilingTaskCreateSchema(ProfilingTaskBaseSchema): ...


		ProfilingTaskResponseSchema = None