import sys
import os
 
def writePlist(fileName, content):    
    # Create a file object:
    # in "write" mode
    FILE = open(fileName,"w")
    header = "<?xml version=\"1.0\" encoding=\"UTF-8\"?>\n"
    header += "<!DOCTYPE plist PUBLIC \"-//Apple//DTD PLIST 1.0//EN\" \"http://www.apple.com/DTDs/PropertyList-1.0.dtd\">\n"
    header += "<plist version=\"1.0\">\n<array>\n"
    content = header + content + "</array>\n</plist>"
 
    # Write all the lines at once:
    FILE.writelines(content)
 
    FILE.close()
 
# If no database is specified, drop out
if len(sys.argv) < 2:
    sys.exit('Usage: %s directoryOfImages' % sys.argv[0])
 
# Get the path to the directory of images
imageDir = sys.argv[1] + '/'    
 
 
if not os.path.exists(imageDir):    
    sys.exit('Directory: %s does not exist' % imageDir)    
 
fileName=""
fileContents=""
 
listing = os.listdir(imageDir)
for infile in listing:
    fileParts = infile.split("_")
    fileStub = fileParts[0]    
    if fileStub != fileName:
        if fileName != "":
            writePlist(fileName + ".plist", fileContents)              
        fileName = fileStub
        fileContents = "\t<string>" + infile + "</string>\n"
    fileContents+= "\t<string>" + infile + "</string>\n"
 
#write the last file
writePlist(fileName + ".plist", fileContents)