From 2a2196c297b370fc262ea3c2fb819a6177fce9c9 Mon Sep 17 00:00:00 2001
From: Damir Ibragimov <ghostklart@yandex.ru>
Date: Tue, 22 Oct 2024 12:11:46 +0000
Subject: [PATCH] Update undelete_for_hse.py

---
 undelete_for_hse.py | 33 ++++++++++++++++-----------------
 1 file changed, 16 insertions(+), 17 deletions(-)

diff --git a/undelete_for_hse.py b/undelete_for_hse.py
index f021ba0..62b97bf 100644
--- a/undelete_for_hse.py
+++ b/undelete_for_hse.py
@@ -2,8 +2,9 @@ import os
 import traceback
 from obs import ObsClient, Versions
 
-bucket = input(f"""Insert bucket name: """)
-prefiks = input(f"""Insert prefix for bucket: """)
+bucket = os.getenv(
+    "BUCKET"
+)
 
 ak = os.getenv(
 	"AKID"
@@ -44,24 +45,22 @@ try:
 		)
 
 		if listing['status'] < 300:
+            for dobject in listing['body']['versions']:
+                if dobject['versionId'] != None and dobject['isLatest'] != True:
+                    actualObject = {
+                        "key": dobject['key'],
+                        "version": dobject['versionId']
+                    }
+                    actualVersions.append(actualObject)
+
 			for file_marker in listing['body']['markers']:
 				file_key = file_marker['key']
 				file_version = file_marker['versionId']
-				
-                # creating lists
-
-				if file_marker['isLatest'] = True:
-					actualFile = {
-                        "key": file_key,
-                        "version_id": file_version
-                    }
-                    actualVersions.append(actualFile)
-				else:
-                    deleteMarkerFile = {
-                        "key": file_key,
-                        "version_id": file_version
-                    }
-                    deleteVersions.append(deleteMarkerFile)
+                deleteMarkerObject = {
+                    "key": file_key,
+                    "version": file_version
+                }
+                deleteVersions.append(deleteMarkerObject)
 
             # going to next file page