def load_zip(self):
if not os.path.exists(self.zipfile):
logging.error('[error] Cannot find page.zip.')
return False
#prepare directory
try:
if os.path.isdir(self.tmp_dir):
logging.info('Removing ' + self.tmp_dir + ' folder...')
shutil.rmtree(self.tmp_dir)
except Exception as e:
logging.error('[error] Cannot delete __tmp__ folder.')
return False
try:
os.makedirs(self.tmp_dir)
except Exception as e:
logging.error('[error] Cannot create __tmp__ folder.')
return False
try:
logging.info('Extracting ' + self.zipfile + ' file...')
Util.unzip(self.zipfile, self.tmp_dir)
except Exception as e:
logging.error('[error] Cannot extract page.zip .')
return False
logging.info('Loading xml...')
path = os.path.join(self.tmp_dir, self.filename)
self.soup = BeautifulSoup(open(path).read())
#cleanup
try:
if os.path.isdir(self.tmp_dir):
shutil.rmtree(self.tmp_dir)
except Exception as e:
logging.error('[error] Cannot delete __tmp__ folder.')
return False
return True
