This commit is contained in:
andreabravetti 2012-07-28 00:41:01 -07:00
commit d23e4b1ae3

View file

@ -166,6 +166,22 @@ def sanitize_open(filename, open_mode):
It returns the tuple (stream, definitive_file_name). It returns the tuple (stream, definitive_file_name).
""" """
# do not overwrite existing files
if filename != u'-':
if filename.endswith(u'.part'):
_addpart = u'.part'
_name, _ext = os.path.splitext(filename[:-len(u'.part')])
filename = filename[:-len(u'.part')]
else:
_addpart = u''
_name, _ext = os.path.splitext(filename)
_count = 0
while os.path.isfile(filename):
_count += 1
filename = "%s-%s%s" % (_name, _count, _ext)
filename = "%s%s" % (filename,_addpart)
try: try:
if filename == u'-': if filename == u'-':
if sys.platform == 'win32': if sys.platform == 'win32':