#coding=utf-8
__author__ = 'zhm'
from win32com import client as wc
import os
import time
import random
import MySQLdb
import re
def wordsToHtml(dir):
#批量把文件夹的word文档转换成html文件
 #金山WPS调用，抢先版的用KWPS，正式版WPS
 word = wc.Dispatch('KWPS.Application')
 for path, subdirs, files in os.walk(dir):
 for wordFile in files:
  wordFullName = os.path.join(path, wordFile)
  #print "word:" + wordFullName
  doc = word.Documents.Open(wordFullName)
  wordFile2 = unicode(wordFile, "gbk")
  dotIndex = wordFile2.rfind(".")
  if(dotIndex == -1):
  print '********************ERROR: 未取得后缀名！'
  fileSuffix = wordFile2[(dotIndex + 1) : ]
  if(fileSuffix == "doc" or fileSuffix == "docx"):
  fileName = wordFile2[ : dotIndex]
  htmlName = fileName + ".html"
  htmlFullName = os.path.join(unicode(path, "gbk"), htmlName)
  # htmlFullName = unicode(path, "gbk") + "\\" + htmlName
  print u'生成了html文件：' + htmlFullName
  doc.SaveAs(htmlFullName, 8)
  doc.Close()
 word.Quit()
 print ""
 print "Finished!"
def html_add_to_db(dir):
#将转换成功的html文件批量插入数据库中。
 conn = MySQLdb.connect(
 host='localhost',
 port=3306,
 user='root',
 passwd='root',
 db='test',
 charset='utf8'
 )
 cur = conn.cursor()
 for path, subdirs, files in os.walk(dir):
 for htmlFile in files:
  htmlFullName = os.path.join(path, htmlFile)
  title = os.path.splitext(htmlFile)[0]
  targetDir = 'D:/files/htmls/'
  #D:/files为web服务器配置的静态目录
  sconds = time.time()
  msconds = sconds * 1000
  targetFile = os.path.join(targetDir, str(int(msconds))+str(random.randint(100, 10000)) +'.html')
  htmlFile2 = unicode(htmlFile, "gbk")
  dotIndex = htmlFile2.rfind(".")
  if(dotIndex == -1):
  print '********************ERROR: 未取得后缀名！'
  fileSuffix = htmlFile2[(dotIndex + 1) : ]
  if(fileSuffix == "htm" or fileSuffix == "html"):
  if not os.path.exists(targetDir):
   os.makedirs(targetDir)
  htmlFullName = os.path.join(unicode(path, "gbk"), htmlFullName)
  htFile = open(htmlFullName,'rb')
  #获取网页内容
  htmStrCotent = htFile.read()
  #找出里面的图片
  img=re.compile(r"""<img\s.*?\s?src\s*=\s*['|"]?([^\s'"]+).*?>""",re.I)
  m = img.findall(htmStrCotent)
  for tagContent in m:
   imgSrc = unicode(tagContent, "gbk")
   imgSrcFullName = os.path.join(path, imgSrc)
   #上传图片
   imgTarget = 'D:/files/images/whzx/'
   img_sconds = time.time()
   img_msconds = sconds * 1000
   targetImgFile = os.path.join(imgTarget, str(int(img_msconds))+str(random.randint(100, 10000)) +'.png')
   if not os.path.exists(imgTarget):
   os.makedirs(imgTarget)
   if not os.path.exists(targetImgFile) or(os.path.exists(targetImgFile) and (os.path.getsize(targetImgFile) != os.path.getsize(imgSrcFullName))):
   tmpImgFile = open(imgSrcFullName,'rb')
   tmpWriteImgFile = open(targetImgFile, "wb")
   tmpWriteImgFile.write(tmpImgFile.read())
   tmpImgFile.close()
   tmpWriteImgFile.close()
   htmStrCotent=htmStrCotent.replace(tagContent,targetImgFile.split(":")[1])
  if not os.path.exists(targetFile) or(os.path.exists(targetFile) and (os.path.getsize(targetFile) != os.path.getsize(htmlFullName))):
   #用iframe包装转换好的html文件。
   iframeHtml='''
   <script type="text/javascript" language="javascript">
   function iFrameHeight() {
    var ifm= document.getElementById("iframepage");
    var subWeb = document.frames ? document.frames["iframepage"].document:ifm.contentDocument;
    if(ifm != null && subWeb != null) {
    ifm.height = subWeb.body.scrollHeight;
    }
   }
   </script>
   <iframe src='''+targetFile.split(':')[1]+'''
   marginheight="0" marginwidth="0" frameborder="0" scrolling="no" width="765" height=100% id="iframepage" name="iframepage" onLoad="iFrameHeight()" ></iframe>
   '''
   tmpTargetFile = open(targetFile, "wb")
   tmpTargetFile.write(htmStrCotent)
   tmpTargetFile.close()
   htFile.close()
   try:
   # 执行
   sql = "insert into common_article(title,content) values(%s,%s)"
   param = (unicode(title, "gbk"),iframeHtml)
   cur.execute(sql,param)
   except:
   print "Error: unable to insert data"
 cur.close()
 conn.commit()
 # 关闭数据库连接
 conn.close()
if __name__ == '__main__':
 wordsToHtml('d:/word')
 html_add_to_db('d:/word')