diff --git a/warcprox/main.py b/warcprox/main.py index 76e194a..d5a6e3f 100644 --- a/warcprox/main.py +++ b/warcprox/main.py @@ -78,6 +78,8 @@ def _build_arg_parser(prog=os.path.basename(sys.argv[0])): default='./warcs', help='where to write warcs') arg_parser.add_argument('-z', '--gzip', dest='gzip', action='store_true', help='write gzip-compressed warc records') + arg_parser.add_argument('--no-warc-open-suffix', dest='no_warc_open_suffix', + default=False, action='store_true', help=argparse.SUPPRESS) arg_parser.add_argument('-n', '--prefix', dest='prefix', default='WARCPROX', help='WARC filename prefix') arg_parser.add_argument( diff --git a/warcprox/writer.py b/warcprox/writer.py index cf8d72d..419fd77 100644 --- a/warcprox/writer.py +++ b/warcprox/writer.py @@ -53,6 +53,7 @@ class WarcWriter: self._f = None self._fpath = None self._f_finalname = None + self._f_finalname_suffix = '' if options.no_warc_open_suffix else '.open' self._serial = 0 self._lock = threading.RLock() @@ -91,7 +92,7 @@ class WarcWriter: self.prefix, self.timestamp17(), self._serial, self._randomtoken, '.gz' if self.gzip else '') self._fpath = os.path.sep.join([ - self.directory, self._f_finalname + '.open']) + self.directory, self._f_finalname + self._f_finalname_suffix]) self._f = open(self._fpath, 'wb')