From d38360be6b7fcd5b700915950a0c18ca5cd81fb3 Mon Sep 17 00:00:00 2001 From: Jim Miller Date: Thu, 19 May 2011 17:30:41 -0500 Subject: [PATCH] Fix zip output file name. --- app.yaml | 2 +- fanficdownloader/writers/base_writer.py | 22 +++++++++++----------- 2 files changed, 12 insertions(+), 12 deletions(-) diff --git a/app.yaml b/app.yaml index f8c0d6b..de13068 100644 --- a/app.yaml +++ b/app.yaml @@ -1,5 +1,5 @@ # fanfictionloader -application: ffd-retief +application: fanfictionloader version: 4-0-0 runtime: python api_version: 1 diff --git a/fanficdownloader/writers/base_writer.py b/fanficdownloader/writers/base_writer.py index 6c3a68e..085970d 100644 --- a/fanficdownloader/writers/base_writer.py +++ b/fanficdownloader/writers/base_writer.py @@ -101,12 +101,18 @@ class BaseStoryWriter(Configurable): self.story.setMetadata('formatext',self.getFormatExt()) def getOutputFileName(self): - return self.getFileName(self.getConfig('output_filename')) + if self.getConfig('zip_output'): + return self.getZipFileName() + else: + return self.getBaseFileName() + def getBaseFileName(self): + return self.formatFileName(self.getConfig('output_filename')) + def getZipFileName(self): - return self.getFileName(self.getConfig('zip_filename'),extension=".zip") + return self.formatFileName(self.getConfig('zip_filename'),extension=".zip") - def getFileName(self,template,extension="${formatext}"): + def formatFileName(self,template,extension="${formatext}"): values = self.story.metadata # fall back default: if not template: @@ -179,13 +185,7 @@ class BaseStoryWriter(Configurable): for tag in self.getConfigList("extratags"): self.story.addToList("extratags",tag) - zipfilename=self.getZipFileName() - filename=self.getOutputFileName() - - if self.getConfig('zip_output'): - outfilename=zipfilename - else: - outfilename=filename + outfilename=self.getOutputFileName() if not outstream: close=True @@ -227,7 +227,7 @@ class BaseStoryWriter(Configurable): out = StringIO.StringIO() self.writeStoryImpl(out) zipout = ZipFile(outstream, 'w', compression=ZIP_DEFLATED) - zipout.writestr(filename,out.getvalue()) + zipout.writestr(self.getBaseFileName(),out.getvalue()) # declares all the files created by Windows. otherwise, when # it runs in appengine, windows unzips the files as 000 perms. for zf in zipout.filelist: