import os 
def getFileListInCurDir():
 """ get file list """
 path = os.getcwd()
 return os.listdir(path)

def extractImageFileList(fileList, replacePercentSign = True):
 """ get image list """
 exts = ('.jpg', '.gif', '.bmp', '.png')
 imageFileList = []

 for ext in exts:
  for fileName in fileList:
   if fileName.find(ext) != -1:
    if replacePercentSign and fileName.find('%') != -1:
     newFileName = fileName.replace('%', '$')
     os.rename(fileName, newFileName)
     fileName = newFileName
     
    imageFileList.append(fileName)
 
 return imageFileList

class HtmlGalleryFile:
 def __init__(self, fileNameOnly, fileCount, align):
  self.fileName = fileNameOnly + repr(fileCount) + '.html'
  self.file = open(self.fileName, 'w')
  
  self.writeStartTag(align)
  
 def close(self, notifyDone = True):
  self.writeEndTag()
  
  self.file.close()
  
  if notifyDone:
   print '%s successfully done' % self.fileName
 
 def writeStartTag(self, align): 
  self.file.write('')
  self.file.write('
' % align)
  
 def writeEndTag(self):
  self.file.write('
')
  self.file.write('')
 
 def write(self, tag):
  self.file.write(tag)
  
 def getFile(self):
  return self.file
 
def makeHtmlImageGallery(fileNameOnly, imageFileList, maxCountPerHtml = -1, align = 'center'):
 fileCount = 0
 imageCount = 0
 
 html = HtmlGalleryFile(fileNameOnly, fileCount, align)
 
 for fileName in imageFileList:
  if maxCountPerHtml != -1 and imageCount >= maxCountPerHtml:
   imageCount = 0
   fileCount += 1
   
   html.close()
   html = HtmlGalleryFile(fileNameOnly, fileCount, align)
   
  s = '
' % fileName
  html.write(s)
  
  imageCount += 1

 html.close()
 
def getOptions():
 options = {}
 
 fileNameOnly = raw_input('html file name(without extension, default: image): ')
 fileNameOnly = fileNameOnly.replace('\r', '')
 if fileNameOnly == '':
  fileNameOnly = 'image'

 maxCountPerHtml = raw_input('max image count per html (default: unlimited): ')
 maxCountPerHtml = maxCountPerHtml.replace('\r', '')
 if maxCountPerHtml == '':
  maxCountPerHtml = -1
 else:
  maxCountPerHtml = int(maxCountPerHtml)
  
 replacePercentSign = True
 align = 'center'
 
 options['fileNameOnly'] = fileNameOnly
 options['maxCountPerHtml'] = maxCountPerHtml
 options['replacePercentSign'] = replacePercentSign
 options['align'] = align
 
 return options

if __name__ == "__main__":
 options = getOptions()

 fileList = getFileListInCurDir()
 print 'getting file list done.'

 imageList = extractImageFileList(fileList, options['replacePercentSign'])
 print 'extract image file list done. %d image(s) found.' % len(imageList)
    
 makeHtmlImageGallery(options['fileNameOnly'], imageList, options['maxCountPerHtml'], options['align'])
 print 'all progress done.'