local changes

Author: charlottekostelic

overload_web/application/commands/file_io.py

@@ -1,4 +1,5 @@
 import logging
+from typing import Any
 
 from overload_web.application import ports
 from overload_web.domain.models import files
@@ -57,3 +58,11 @@ def execute(file: bytes, file_name: str, dir: str, writer: ports.FileWriter) ->
         """
         out_file = writer.write(file=file, file_name=file_name, dir=dir)
         return out_file
+
+
+class SaveIncomingFile:
+    @staticmethod
+    def execute(
+        repo: ports.SqlRepositoryProtocol, file: files.VendorFile
+    ) -> dict[str, Any]:
+        return repo.save(file)

overload_web/infrastructure/file_io.py

@@ -119,3 +119,39 @@ def write(self, file: bytes, file_name: str, dir: str) -> str:
         out_file = self.client.put_file(file=converted_file, remote=True, dir=dir)
         logger.info(f"Writing file to directory: {dir}/{out_file}")
         return getattr(out_file, "file_name", file_name)
+
+
+# class IncomingFileBatch(SQLModel, table=True):
+#     __tablename__ = "incoming_file_batches"
+
+#     id: int = Field(default=None, primary_key=True, index=True)
+#     files: list["IncomingFileModel"] = Relationship(
+#         back_populates="batch", sa_relationship_kwargs={"lazy": "selectin"}
+#     )
+
+
+# class IncomingFileModel(SQLModel, table=True):
+#     __tablename__ = "incoming_files"
+#     id: int = Field(default=None, primary_key=True, index=True)
+#     content: bytes = Field(nullable=False)
+#     file_name: str = Field(nullable=False)
+#     batch_id: int = Field(
+#         default=None, foreign_key="incoming_file_batches.id", exclude=True
+#     )
+#     batch: IncomingFileBatch = Relationship(back_populates="files")
+
+
+# class IncomingFileRepository:
+#     def __init__(self, session: Session):
+#         self.session = session
+
+#     def get(self, id: str | int) -> dict[str, Any] | None:
+#         batch = self.session.get(IncomingFileBatch, id)
+#         return batch.model_dump() if batch else None
+
+#     def save(self, obj: IncomingFileBatch) -> dict[str, Any]:
+#         valid_obj = IncomingFileBatch.model_validate(obj, from_attributes=True)
+#         self.session.add(valid_obj)
+#         self.session.commit()
+#         self.session.refresh(valid_obj)
+#         return valid_obj.model_dump()

overload_web/presentation/deps.py

@@ -75,6 +75,13 @@ def pvf_batch_db(
     yield batch_db.PVFBatchRepository(session=session)
 
 
+# def incoming_file_db(
+#     session: Annotated[Any, Depends(get_session)],
+# ) -> Generator[file_io.IncomingFileRepository, None, None]:
+#     """Create an PVFBatch repository."""
+#     yield file_io.IncomingFileRepository(session=session)
+
+
 @runtime_checkable
 class FileProtocol(Protocol):
     """A protocol representing a FastAPI `UploadFile` object."""