Update undelete_for_hse.py

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