VienDesuPorting/VNTools
4
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

vnrecode: duplicates check for more than two files

Browse source
This commit is contained in:
OleSTEEP 2024-10-19 02:20:58 +03:00
parent df20bd3636
commit 407ab98000
2 changed files with 15 additions and 11 deletions
Download patch file Download diff file Expand all files Collapse all files

2
vnrecode/compress.py
View file

@ -192,6 +192,6 @@ class Compress:
case "unknown": case "unknown":
out_file = self.unknown(source, output) out_file = self.unknown(source, output)
self.__utils.out_rename(out_file, source.name) self.__utils.out_rename(out_file, source)
self.__printer.bar.update() self.__printer.bar.update()
self.__printer.bar.next() self.__printer.bar.next()

24
vnrecode/utils.py
View file

@ -17,7 +17,7 @@ class Utils:
self.__errors = 0 self.__errors = 0
self.__params = params_inst self.__params = params_inst
self.__printer = printer_inst self.__printer = printer_inst
self.__duplicates = [] self.__duplicates = {}
@staticmethod @staticmethod
def sys_pause(): def sys_pause():
@ -102,9 +102,13 @@ class Utils:
:return: Duplicate path name with (vncopy) on end :return: Duplicate path name with (vncopy) on end
""" """
if path.is_file() and path.exists(): if path.is_file() and path.exists():
new_path = Path(path.stem + "(vncopy)" + path.suffix) orig_name = path.name.replace("(vncopy)", "")
self.__duplicates.append(new_path) new_path = Path(path.parent, path.stem + "(vncopy)" + path.suffix)
return new_path try: self.__duplicates[orig_name]
except KeyError: self.__duplicates[orig_name] = []
if not new_path.name in self.__duplicates[orig_name]:
self.__duplicates[orig_name].append(new_path.name)
return self.catch_duplicates(new_path)
return path return path
def print_duplicates(self): def print_duplicates(self):
@ -112,13 +116,13 @@ class Utils:
Method prints message about all duplicates generated during recode process Method prints message about all duplicates generated during recode process
:return: None :return: None
""" """
for filename in self.__duplicates: for filename in self.__duplicates.keys():
self.__printer.warning( self.__printer.warning(
f'Duplicate file has been found! Check manually this files - "{filename.name}", ' f'Duplicate file has been found! Check manually this files - "{filename}", ' +
f'"{filename.stem + "(vncopy)" + filename.suffix}"' ', '.join(self.__duplicates[filename])
) )
def out_rename(self, out_path: Path, target: str): def out_rename(self, out_path: Path, target: Path):
""" """
Method removes md5 hash from file name and changes file extension in dependence of mimic mode Method removes md5 hash from file name and changes file extension in dependence of mimic mode
:param out_path: Recoded file Path :param out_path: Recoded file Path
@ -126,7 +130,7 @@ class Utils:
:return: None :return: None
""" """
if not self.__params.mimic_mode: if not self.__params.mimic_mode:
dest_name = self.catch_duplicates(Path(out_path.parent, target)) dest_name = self.catch_duplicates(Path(out_path.parent, target.stem+out_path.suffix))
os.rename(out_path, dest_name) os.rename(out_path, dest_name)
else: else:
os.rename(out_path, Path(out_path.parent, target)) os.rename(out_path, Path(out_path.parent, target.name))