glife/tools/check_images2.py

#!/usr/bin/env python

import io
import os
import re
import sys

#path = os.getcwd()
#print path

startpattern = """images/"""
imgFormats = ['jpg','gif','png','mp4']

infile = io.open('glife.txt',mode='r',encoding='utf-16')
lines = infile.readlines()

images = []

for name in os.listdir("locations"):
    path = os.path.join("locations", name)
    if os.path.isdir(path):
        continue
        # skip directories
    else:
        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|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