020: class, objects and instance: 一个简单的例子，压缩文件中内容的替换

import sys
import os
import shutil
import zipfile

class ZipReplace(object):
def __init__(self, file_name, search_string, replace_string):
self.file_name = file_name
self.search_string = search_string
self.replace_string = replace_string
self.temp_directory = "unzipped-{}".format(file_name)

def __full_filename(self, file_name):
return os.path.join(self.temp_directory, file_name)

def zip_find_replace(self):
self.unzip_files()
self.find_replace()
self.zip_files()

def unzip_files(self):
os.mkdir(self.temp_directory)
zip = zipfile.ZipFile(self.file_name)
try:
zip.extractall(self.temp_directory)
finally:
zip.close()

def find_replace(self):
for file_name in os.listdir(self.temp_directory):
with open(self.__full_filename(file_name)) as file:
contents = file.read()

contents = contents.replace(self.search_string, self.replace_string)

with open(self.__full_filename(file_name), 'w') as file:
file.write(contents)

def zip_files(self):
file = zipfile.ZipFile(self.file_name, 'w')

for file_name in os.listdir(self.temp_directory):
file.write(self.__full_filename(file_name), file_name)

shutil.rmtree(self.temp_directory)

zr = ZipReplace("test.zip", 'hello', 'hello world...')
zr.zip_find_replace()