diff --git a/ncmdump.py b/ncmdump.py index 3a1b407..093182d 100644 --- a/ncmdump.py +++ b/ncmdump.py @@ -78,7 +78,7 @@ def dump(file_path): return file_name -class dumper(threading.Thread): +class Dumper(threading.Thread): def __init__(self, fpath, basket): super().__init__() self._fpath = fpath @@ -119,9 +119,7 @@ def printBar(now:int, total:int, threads:int): print(bar_text, end='', flush=True) -if __name__ == '__main__': - if not os.path.exists('./unlock'): - os.mkdir('./unlock') +def main(): dirs = os.listdir() musicFiles = dict() for i in dirs: @@ -129,7 +127,13 @@ if __name__ == '__main__': if os.path.splitext(i)[1] == '.ncm': musicFiles[os.path.abspath(i)] = 0 totalCount = len(musicFiles) + if totalCount == 0: + print('\033[33m未找到需要解密的文件 nothing to do...\033[0m') + input('按下回车继续...') + return process = [0, 0, 0] + if not os.path.exists('./unlock'): + os.mkdir('./unlock') while True: count = 0 tmp = None @@ -139,8 +143,7 @@ if __name__ == '__main__': if count < 5 and musicFiles[key] == 0: upperPrint('解密中...\t{}'.format(key)) musicFiles[key] = 1 - tmp = dumper(key, musicFiles) - tmp.start() + Dumper(key, musicFiles).start() count += 1 fcount = 0 for key in musicFiles: @@ -154,3 +157,7 @@ if __name__ == '__main__': print('\n\033[32m解密完成,存放在unlock目录中\033[0m') input('按下回车继续...') + + +if __name__ == '__main__': + main()