Hacktivitycon CTF Tootsie Pop

tootsie_pop.py

#it stops once because of python recursion limits but can be manually restarted
#it extracts files and then moves them to one directory so the filename doesn't get too long(which is not supported in some tools like bzip)
from xtract import xtract

import os
import glob

# def names():
def delempty(dirr):
    for i in [f[0] for f in os.walk(dirr) if os.path.isdir(f[0])]:
        try:
            os.rmdir(i)
        except OSError:
            pass

def ext(inp):
    delempty('.')
    for i in inp:
        if i =="pop.zip":
            try: xtract('pop.zip')
            except: pass
            out='pop'
        else:
            out = xtract(i)

        if i != "pop.zip":
            os.remove(i)

        if os.path.isdir(out):
            th = [
                i for i in glob.glob(out + "/**", recursive=True) if os.path.isfile(i)
            ]
            ext(th)
        else:
            os.rename(out, "pop/" + os.path.basename(out))
            ext(["pop/" + os.path.basename(out)])


ext(["pop.zip"])