diff --git a/dataset.py b/dataset.py
index 9e9d8e8..754aa86 100644
--- a/dataset.py
+++ b/dataset.py
@@ -25,6 +25,9 @@ class Selection:
     diff_after_address_change: Optional[bool]
     is_code_related: bool
 
+class ArchiveState(Enum):
+    BASE = "base"
+    MERGED = "merged"
 
 @dataclass
 class Metadata:
@@ -40,6 +43,11 @@ class Metadata:
     last_cmd_error_msg: str = ""
     selection: Optional[Selection] = None
 
+    def archive_name(self, state: ArchiveState, only_id:bool=False):
+        if only_id:
+            return f"{self.id}_{state.value}.tar.gz"
+        return f"{self.repo.replace('/', '_')}_{self.pr_number}_{state.value}.tar.gz"
+
 @dataclass
 class DatasetEntry:
     metadata: Metadata