Browse Source

fix check image to check mod_ibiza.txt instead the location

rachels 2 years ago
parent
commit
aadb2d5c89
1 changed files with 20 additions and 23 deletions
  1. 20 23
      check_images.py

+ 20 - 23
check_images.py

@@ -3,37 +3,34 @@
 import io
 import os
 import re
+import sys
 
 #path = os.getcwd()
 #print path
 
-startpattern = """images/"""
-imgFormats = ['jpg','gif']
-
-infile = io.open('mod_Ibiza.txt',mode='r',encoding='utf-16')
-lines = infile.readlines()
+startpattern = """mod/ibiza/"""
+imgFormats = ['jpg','jpeg','gif','png','mp4']
 
 images = []
 
-for name in os.listdir("locations"):
-    ifile = io.open(
-        os.path.join("locations", name),
-        mode='rt',
-        encoding='utf-8'
-    )
-    text = ifile.read()
-    for match in re.finditer(r"images.+?[.](gif|jpg|png)", text, flags=re.U):
-        imgfile = match.group().encode("utf-8")
-        randmatch = re.search(r"'\s*[+]\s*rand\s*[(]\s*(\d+)\s*[,]\s*(\d+)\s*[)]\s*[+]\s*'", imgfile)
-        if randmatch != None:
-            for i in range(int(randmatch.group(1)), 1+int(randmatch.group(2))):
-                images.append(re.sub(r"'\s*[+]\s*rand\s*[(].*?[)]\s*[+]\s*'", str(i), imgfile))
-        else:
-            images.append(imgfile)
-    
-    ifile.close()
-
+ifile = io.open(
+    "mod_ibiza.txt",
+    mode='rt',
+    encoding='utf-16'
+)
+text = ifile.read()
+for match in re.finditer(r"mod/ibiza/.+?[.](gif|jpg|jpeg|png|mp4)", text, flags=re.U):
+    imgfile = match.group().encode("utf-8")
+    randmatch = re.search(r"'\s*[+]\s*rand\s*[(]\s*(\d+)\s*[,]\s*(\d+)\s*[)]\s*[+]\s*'", imgfile)
+    if randmatch != None:
+        for i in range(int(randmatch.group(1)), 1+int(randmatch.group(2))):
+            images.append(re.sub(r"'\s*[+]\s*rand\s*[(].*?[)]\s*[+]\s*'", str(i), imgfile))
+    else:
+        images.append(imgfile)
+        
+ifile.close()
 
 for image in images:
     if not re.search(r"[<$]", image) and not os.path.isfile(image):
         print "Image not found:", image
+