|
import zipfile |
|
import os |
|
import io |
|
import time |
|
from PIL import Image |
|
|
|
|
|
class ZipReader(object): |
|
zip_bank = dict() |
|
|
|
def __init__(self): |
|
super(ZipReader, self).__init__() |
|
|
|
@staticmethod |
|
def get_zipfile(path): |
|
zip_bank = ZipReader.zip_bank |
|
if path in zip_bank: |
|
return zip_bank[path] |
|
else: |
|
print("creating new zip_bank") |
|
zfile = zipfile.ZipFile(path, 'r') |
|
zip_bank[path] = zfile |
|
return zip_bank[path] |
|
|
|
@staticmethod |
|
def split_zip_style_path(path): |
|
pos_zip_at = path.index('.zip@') |
|
if pos_zip_at == len(path): |
|
print("character '@' is not found from the given path '%s'" % (path)) |
|
assert 0 |
|
pos_at = pos_zip_at + len('.zip@') - 1 |
|
|
|
zip_path = path[0: pos_at] |
|
folder_path = path[pos_at + 1:] |
|
folder_path = str.strip(folder_path, '/') |
|
return zip_path, folder_path |
|
|
|
@staticmethod |
|
def list_folder(path): |
|
zip_path, folder_path = ZipReader.split_zip_style_path(path) |
|
|
|
zfile = ZipReader.get_zipfile(zip_path) |
|
folder_list = [] |
|
for file_foler_name in zfile.namelist(): |
|
file_foler_name = str.strip(file_foler_name, '/') |
|
if file_foler_name.startswith(folder_path) and \ |
|
len(os.path.splitext(file_foler_name)[-1]) == 0 and \ |
|
file_foler_name != folder_path: |
|
if len(folder_path) == 0: |
|
folder_list.append(file_foler_name) |
|
else: |
|
folder_list.append(file_foler_name[len(folder_path)+1:]) |
|
|
|
return folder_list |
|
|
|
@staticmethod |
|
def list_files(path, extension=['.*']): |
|
zip_path, folder_path = ZipReader.split_zip_style_path(path) |
|
|
|
zfile = ZipReader.get_zipfile(zip_path) |
|
file_lists = [] |
|
for file_foler_name in zfile.namelist(): |
|
file_foler_name = str.strip(file_foler_name, '/') |
|
if file_foler_name.startswith(folder_path) and str.lower(os.path.splitext(file_foler_name)[-1]) in extension: |
|
if len(folder_path) == 0: |
|
file_lists.append(file_foler_name) |
|
else: |
|
file_lists.append(file_foler_name[len(folder_path)+1:]) |
|
|
|
return file_lists |
|
|
|
@staticmethod |
|
def imread(path): |
|
zip_path, path_img = ZipReader.split_zip_style_path(path) |
|
zfile = ZipReader.get_zipfile(zip_path) |
|
data = zfile.read(path_img) |
|
im = Image.open(io.BytesIO(data)) |
|
return im |
|
|
|
@staticmethod |
|
def read(path): |
|
zip_path, path_img = ZipReader.split_zip_style_path(path) |
|
zfile = ZipReader.get_zipfile(zip_path) |
|
data = zfile.read(path_img) |
|
return data |
|
|
