Added a check for THP files

Lord-Giganticus · Oct 29, 2020 · 3800da3 · 3800da3
1 parent 3de1514
commit 3800da3
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/THP2MP4.py b/THP2MP4.py
@@ -7,6 +7,11 @@
 print(os.getcwd())
 input("If this is not the correct folder I swear...")
 
+for name in os.listdir('.'):
+    if name.endswith('.thp') == False:
+        input("There are no thp files here! Press enter to exit and next time choose the right folder.")
+        exit()
+
 THP_Conv = open('THPconv.bat', 'w')
 THP_Conv.write('for %%a in ("*.thp") do ffmpeg -i "%%a" "%%~na.mp4"')
 THP_Conv.close()