Source code for utility_modules.import_zip

#!/usr/bin/env python

import pandas as pd
import zipfile
import os
import pdb
try:
    from StringIO import StringIO
except ImportError:
    from io import StringIO


[docs]def unzip_to_df(fn): 
    with zipfile.ZipFile(fn, 'r') as f:
        filename = f.namelist()[0]
        x = f.read(filename)
    return  pd.read_csv(StringIO(x))



[docs]def zipdir(path,ziph):
	for root, dirs, files in os.walk(path):
		for file in files:
			ziph.write(os.path.join(root,file))