extractor.py

#hitcon
import zipfile, tarfile
import os
import sys
sys.setrecursionlimit(10000)


def unzip(file, path):
    print file
    zip = zipfile.ZipFile(file)
    zipList = zip.namelist()
    for file in zipList:
        if file.endswith('zip'):
            break
    zip.extract(file, path).split('\\')
    for file in zipList:
        if os.path.isdir(file):
            return path+'/'+file
def untar(file, path):
    print file
    tar=tarfile.open(file)
    tar.extractall()
    tarList=tar.getnames()
    for file in tarList:
        if os.path.isdir(file):
            return path+'/'+file   
    
def dirHandle(path):
    os.chdir(path)

    for file in os.listdir(path):
        if os.path.isfile(file):
            if file.endswith('zip'):
                path=unzip(file, path)
            if file.endswith('.tar.bz2') or file.endswith('.tar.gz') or file.endswith('.tar'):
                path=untar(file, path)
            print path.replace('//', '\\')
            dirHandle(path)
'''
def pathTruncate():
    dst='F:\studies\tutorials\CTF stuff\hitcon14'
    for file in os.listdir(path):
        if os.path.isfile(file):
            copyfile(file, dst)
'''          
path=r'F:\studies\tutorials\CTF stuff\hitcon14\\'
dirHandle(path)