From 65806ccbe38e8e6aa681f3ef487d1e431f3ac5d9 Mon Sep 17 00:00:00 2001
From: Karma Riuk <riukkarma@gmail.com>
Date: Wed, 14 May 2025 09:18:11 +0200
Subject: [PATCH] now the metadata knows it's archive name

---
 dataset.py | 8 ++++++++
 1 file changed, 8 insertions(+)

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