Minor Improvement

bearlike · Nov 25, 2018 · 425ff88 · 425ff88
1 parent af27c75
commit 425ff88
Show file tree

Hide file tree

Showing 2 changed files with 32 additions and 27 deletions.
diff --git a/RENAME-Movies.py b/RENAME-Movies.py
@@ -57,6 +57,7 @@ def removeIllegal(str):
     return(str)
 
 def Title():
+    os.system("cls")
     os.system('mode con: cols=75 lines=30')
     print("    _       _                  _          _  ")
     print("   /_\ _  _| |_ ___ _ __  __ _| |_ ___ __| | ")
@@ -77,31 +78,28 @@ def Title():
 
 
 def FormatStr(temp):
-            if ".1080p" in temp:
-                sep = ".1080p"
-            elif ".720p" in temp:
-                sep = ".720p"
-            elif "[" in temp:
-                sep = "["
-            elif "1080p" in temp:
-                sep = "1080p"
-            elif "720p" in temp:
-                sep = "720p"
-            if "TamilRockers" in temp:
-                temp = temp.split(' - ',1)[1]
-            try:
-                rest = temp.split(sep,1)[0]
-            except:
-                pass
-            rest = rest.replace("."," ")
-            rest = rest.replace("(","")
-            rest = rest.replace(")","")
-            return(rest)
-
-
+    if ".1080p" in temp:
+        sep = ".1080p"
+    elif ".720p" in temp:
+        sep = ".720p"
+    elif "[" in temp:
+        sep = "["
+    elif "1080p" in temp:
+        sep = "1080p"
+    elif "720p" in temp:
+        sep = "720p"
+    if "TamilRockers" in temp:
+        temp = temp.split(' - ',1)[1]
+    try:
+        rest = temp.split(sep,1)[0]
+    except:
+        pass
+    rest = rest.replace("."," ")
+    rest = rest.replace("(","")
+    rest = rest.replace(")","")
+    return(rest)
 
 #Driver Code
-os.system("cls")
 Title()
 try:
     os.mkdir("Input")
@@ -125,6 +123,8 @@ def FormatStr(temp):
     extn = file[(len(file)-4) : len(file)]
 
     if(file.endswith(".mp4") or file.endswith(".mkv") or file.endswith(".srt")) :
+        Title()
+        print(str(i)+" File(s) Processed....")
         rest = FormatStr(temp)
         year_str =  '('+ rest[len(rest)-4 : len(rest)] +')'
         rest = rest[0:len(rest)-4]
@@ -163,7 +163,8 @@ def FormatStr(temp):
 
 ###############################################################################
 ##RESULT GENERATION
-print (" ")
+Title()
+print ("All Files Processed...")
 if(FileFlag==1):
     print("Solution: Try again after removing the above file(s) from Output folder")
 if(i>0 or ErrorFlag==1):

diff --git a/RENAME-Series.py b/RENAME-Series.py
@@ -11,6 +11,7 @@
 from similarity.damerau import Damerau
 
 def Title():
+    os.system("cls")
     os.system('mode con: cols=90 lines=30')
     print("    _       _                  _          _  ")
     print("   /_\ _  _| |_ ___ _ __  __ _| |_ ___ __| | ")
@@ -113,7 +114,6 @@ def split_line(text):
     return words
 ###############################################################################
 ##MAIN
-os.system("cls")
 Title()
 try:
     os.mkdir(str(os.getcwd())+"\Input")
@@ -131,12 +131,15 @@ def split_line(text):
 ErrorFlag=0
 FileFlag=0
 files = os.listdir(path)
+print("Reading Files....")
 for file in files:
     CheckFlag=0
     temp = file
     extn = file[(len(file)-4) : len(file)]
 ###############################################################################
     if (file.endswith(".mp4") or file.endswith(".mkv") or file.endswith(".srt")):
+        Title()
+        print(str(i)+" File(s) Processed....")
         rest = temp.split(extn,1)[0]
         if ".1080p" in temp:
             sep = ".1080p"
@@ -206,7 +209,7 @@ def split_line(text):
         path_new = os.getcwd() + "\\Output\\Series\\" + NAME
         path_new_1 = os.getcwd() + "\\Output\\Series\\" + NAME +"\\Season "+ str(int(SEASON))
 
-        ###############################################################################
+###############################################################################
     def TEST():
         ##   TESTING
         ## =============
@@ -250,7 +253,8 @@ def TEST():
 
 ###############################################################################
 ##RESULT GENERATION
-print (" ")
+Title()
+print ("All Files Processed...")
 if(FileFlag==1):
     print("Solution: Try again after removing the above file(s) from Output folder")
 if(i>0 or ErrorFlag==1):