added file size to namespace archive

pepkit · Sep 9, 2024 · 8b43a07 · 8b43a07
1 parent 1efb3e4
commit 8b43a07
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 0 deletions.
diff --git a/pepdbagent/db_utils.py b/pepdbagent/db_utils.py
@@ -373,6 +373,7 @@ class TarNamespace(Base):
     file_path: Mapped[str] = mapped_column(nullable=False)
     creation_date: Mapped[datetime.datetime] = mapped_column(default=deliver_update_date)
     number_of_projects: Mapped[int] = mapped_column(default=0)
+    file_size: Mapped[int] = mapped_column(nullable=False)
 
 
 class BaseEngine:

diff --git a/pepdbagent/models.py b/pepdbagent/models.py
@@ -305,6 +305,7 @@ class TarNamespaceModel(BaseModel):
     file_path: str
     creation_date: datetime.datetime = None
     number_of_projects: int = 0
+    file_size: int = 0
 
 
 class TarNamespaceModelReturn(BaseModel):

diff --git a/pepdbagent/modules/namespace.py b/pepdbagent/modules/namespace.py
@@ -317,6 +317,7 @@ def upload_tar_info(self, tar_info: TarNamespaceModel) -> None:
                 namespace=tar_info.namespace,
                 creation_date=tar_info.creation_date,
                 number_of_projects=tar_info.number_of_projects,
+                file_size=tar_info.file_size,
             )
             session.add(new_tar)
             session.commit()
@@ -348,6 +349,7 @@ def get_tar_info(self, namespace: str) -> TarNamespaceModelReturn:
                         file_path=result.file_path,
                         creation_date=result.creation_date,
                         number_of_projects=result.number_of_projects,
+                        file_size=result.file_size,
                     )
                 )