diff --git a/AV_Data_Capture.py b/AV_Data_Capture.py index 9bb0142..5ca95d8 100644 --- a/AV_Data_Capture.py +++ b/AV_Data_Capture.py @@ -4,28 +4,21 @@ import time def movie_lists(): #MP4 - a1 = glob.glob(os.getcwd() + r"\*\**\*.mp4") a2 = glob.glob(os.getcwd() + r"\*.mp4") # AVI - b1 = glob.glob(os.getcwd() + r"\*\**\*.avi") b2 = glob.glob(os.getcwd() + r"\*.avi") # RMVB - c1 = glob.glob(os.getcwd() + r"\*\**\*.rmvb") c2 = glob.glob(os.getcwd() + r"\*.rmvb") # WMV - d1 = glob.glob(os.getcwd() + r"\*\**\*.wmv") d2 = glob.glob(os.getcwd() + r"\*.wmv") # MOV - e1 = glob.glob(os.getcwd() + r"\*\**\*.mov") e2 = glob.glob(os.getcwd() + r"\*.mov") # MKV - f1 = glob.glob(os.getcwd() + r"\*\**\*.mkv") f2 = glob.glob(os.getcwd() + r"\*.mkv") # FLV - g1 = glob.glob(os.getcwd() + r"\*\**\*.flv") g2 = glob.glob(os.getcwd() + r"\*.flv") - total = a1+a2+b1+b2+c1+c2+d1+d2+e1+e2+f1+f2+g1+g2 + total = a2+b2+c2+d2+e2+f2+g2 return total def lists_from_test(custom_nuber): @@ -33,7 +26,20 @@ def lists_from_test(custom_nuber): a.append(custom_nuber) return a -os.chdir(os.getcwd()) -for i in movie_lists(): - os.system('python core.py'+' "'+i+'"') -print("[+]All finished!!!") +def CEF(path): + files = os.listdir(path) # 获取路径下的子文件(夹)列表 + for file in files: + try: #试图删除空目录,非空目录删除会报错 + os.removedirs(path + '/' + file) # 删除这个空文件夹 + print('[+]Deleting empty folder',path + '/' + file) + except: + a='' + +if __name__ =='__main__': + os.chdir(os.getcwd()) + for i in movie_lists(): + os.system('python core.py' + ' "' + i + '"') + print("[!]Cleaning empty folders") + CEF('JAV_output') + print("[+]All finished!!!") + time.sleep(3)