Skip to content
This repository has been archived by the owner on Feb 25, 2018. It is now read-only.

Commit

Permalink
be sure to specify a max_size to SpooledTemporaryFile
Browse files Browse the repository at this point in the history
  • Loading branch information
rfk committed Mar 2, 2010
1 parent ee033d1 commit 961f31c
Show file tree
Hide file tree
Showing 2 changed files with 10 additions and 3 deletions.
5 changes: 5 additions & 0 deletions ChangeLog
Original file line number Diff line number Diff line change
@@ -1,4 +1,9 @@

Version 0.3.7

* Buffer: explicitly pass a max_size to SpooledTemporaryFile, with
a sensible default value.

Version 0.3.6

* More robust __del__ method for FileWrapper subclasses
Expand Down
8 changes: 5 additions & 3 deletions filelike/wrappers/buffer.py
Original file line number Diff line number Diff line change
Expand Up @@ -36,9 +36,11 @@
from filelike.wrappers import FileWrapper

try:
from tempfile import SpooledTemporaryFile as TemporaryFile
from tempfile import SpooledTemporaryFile
except ImportError:
from tempfile import TemporaryFile
def SpooledTemporaryFile(max_size=None,*args,**kwds):
return TemporaryFile(*args,**kwds)


class Buffer(FileWrapper):
Expand All @@ -50,9 +52,9 @@ class Buffer(FileWrapper):
back to the file on close.
"""

def __init__(self,fileobj,mode=None):
def __init__(self,fileobj,mode=None,max_size_in_memory=1024*8):
"""Buffered file wrapper constructor."""
self._buffer = TemporaryFile()
self._buffer = SpooledTemporaryFile(max_size=max_size_in_memory)
self._in_eof = False
self._in_pos = 0
super(Buffer,self).__init__(fileobj,mode)
Expand Down

0 comments on commit 961f31c

Please sign in to comment.