LBRYCommunity/lbry-sdk
1
0
Fork 1
Code Issues Pull requests Projects Releases 282 Packages Wiki Activity

add -x to filename before extension for duplicate downloads

Browse source
This commit is contained in:
Jack 2016-04-22 02:45:41 -04:00
parent fecd29a967
commit bac7f097a9
1 changed files with 12 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

14
lbrynet/lbryfile/client/LBRYFileDownloader.py
View file

@ -211,10 +211,20 @@ class LBRYFileSaver(LBRYFileDownloader):
file_name = "_" file_name = "_"
if os.path.exists(os.path.join(self.download_directory, file_name)): if os.path.exists(os.path.join(self.download_directory, file_name)):
ext_num = 1 ext_num = 1
def _get_file_name(ext):
if len(file_name.split(".")):
fn = ''.join(file_name.split(".")[:-1])
file_ext = ''.join(file_name.split(".")[-1])
return fn + "-" + str(ext) + "." + file_ext
else:
return file_name + "_" + str(ext)
while os.path.exists(os.path.join(self.download_directory, while os.path.exists(os.path.join(self.download_directory,
file_name + "_" + str(ext_num))): _get_file_name(ext_num))):
ext_num += 1 ext_num += 1
file_name = file_name + "_" + str(ext_num)
file_name = _get_file_name(ext_num)
try: try:
self.file_handle = open(os.path.join(self.download_directory, file_name), 'wb') self.file_handle = open(os.path.join(self.download_directory, file_name), 'wb')
self.file_written_to = os.path.join(self.download_directory, file_name) self.file_written_to = os.path.join(self.download_directory, file_name)