import zipfile
from os import listdir, remove, system, popen
from os.path import isfile, join
import tarfile
import gzip
import shutil

files = [f for f in listdir('/Users/bent/Desktop/flag') if isfile(join('/Users/bent/Desktop/flag', f))]
for f in files:
   remove('/Users/bent/Desktop/flag/' + f)

zip_ref = zipfile.ZipFile('/Users/bent/Desktop/Archive.zip', 'r')
zip_ref.extractall('/Users/bent/Desktop/flag')
zip_ref.close()

files = [f for f in listdir('/Users/bent/Desktop/flag') if isfile(join('/Users/bent/Desktop/flag', f))]
i = 1
while len(files) == 1:
   print(i)
   print(files[0])

   filetype = ""

   if files[0] == "flag":
      filetype = popen('file /Users/bent/Desktop/flag/' + files[0]).read()
      print(filetype)

   if 'tar' in filetype:
      tar_ref = tarfile.open('/Users/bent/Desktop/flag/' + files[0])
      remove('/Users/bent/Desktop/flag/' + files[0])
      tar_ref.extractall('/Users/bent/Desktop/flag')
      tar_ref.close()
   elif 'gzip' in filetype:
      with gzip.open('/Users/bent/Desktop/flag/' + files[0],"rb") as f_in, open('/Users/bent/Desktop/flag/flag',"wb") as f_out:
         remove('/Users/bent/Desktop/flag/' + files[0])
         shutil.copyfileobj(f_in, f_out)
   else:
      zip_ref = zipfile.ZipFile('/Users/bent/Desktop/flag/' + files[0], 'r')
      remove('/Users/bent/Desktop/flag/' + files[0])
      zip_ref.extractall('/Users/bent/Desktop/flag')
      zip_ref.close()
   files = [f for f in listdir('/Users/bent/Desktop/flag') if isfile(join('/Users/bent/Desktop/flag', f))]
   i = i + 1